Make failing tests fail

getnikola · Sep 21, 2015 · 53401cf · 53401cf
1 parent 140ebdf
commit 53401cf
Showing 1 changed file with 2 additions and 9 deletions.
diff --git a/tests/test_integration.py b/tests/test_integration.py
@@ -307,18 +307,11 @@ class TestCheck(DemoBuildTest):
 
     def test_check_links(self):
         with cd(self.target_dir):
-            try:
-                __main__.main(['check', '-l'])
-            except SystemExit as e:
-                self.assertEqual(e.code, 0)
+            self.assertIsNone(__main__.main(['check', '-l']))
 
     def test_check_files(self):
         with cd(self.target_dir):
-            try:
-                __main__.main(['check', '-f'])
-            except SystemExit as e:
-                self.assertEqual(e.code, 0)
-
+            self.assertIsNone(__main__.main(['check', '-f']))
 
 class TestCheckAbsoluteSubFolder(TestCheck):
     """Validate links in a site which is: