GithubBackup/httprunner
1
0
Fork 0
mirror of https://github.com/httprunner/httprunner.git synced 2026-08-28 11:36:56 +08:00
Code Issues Packages Projects Releases Wiki Activity

test: Reset redirect in teardown

Browse Source
This commit is contained in:
debugtalk
2019-12-25 22:36:17 +08:00
parent 3caba271b0
commit afccf1e5f8
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/test_cli.py
+3 -2
View File
@@ -11,12 +11,14 @@ class TestCli(unittest.TestCase):
self.captured_output = io.StringIO() self.captured_output = io.StringIO()
sys.stdout = self.captured_output sys.stdout = self.captured_output
def tearDown(self):
sys.stdout = sys.__stdout__ # Reset redirect.
def test_show_version(self): def test_show_version(self):
sys.argv = ["hrun", "-V"] sys.argv = ["hrun", "-V"]
with self.assertRaises(SystemExit) as cm: with self.assertRaises(SystemExit) as cm:
main() main()
sys.stdout = sys.__stdout__ # Reset redirect.
self.assertEqual(cm.exception.code, 0) self.assertEqual(cm.exception.code, 0)
@@ -28,7 +30,6 @@ class TestCli(unittest.TestCase):
with self.assertRaises(SystemExit) as cm: with self.assertRaises(SystemExit) as cm:
main() main()
sys.stdout = sys.__stdout__ # Reset redirect.
self.assertEqual(cm.exception.code, 0) self.assertEqual(cm.exception.code, 0)