GithubBackup/httprunner
1
0
Fork 0
mirror of https://github.com/httprunner/httprunner.git synced 2026-05-13 05:20:06 +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 3850427c68
commit c29d770440
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
tests/test_cli.py
View File

@@ -11,12 +11,14 @@ class TestCli(unittest.TestCase):
self.captured_output = io.StringIO()
sys.stdout = self.captured_output
def tearDown(self):
sys.stdout = sys.__stdout__ # Reset redirect.
def test_show_version(self):
sys.argv = ["hrun", "-V"]
with self.assertRaises(SystemExit) as cm:
main()
sys.stdout = sys.__stdout__ # Reset redirect.
self.assertEqual(cm.exception.code, 0)
@@ -28,7 +30,6 @@ class TestCli(unittest.TestCase):
with self.assertRaises(SystemExit) as cm:
main()
sys.stdout = sys.__stdout__ # Reset redirect.
self.assertEqual(cm.exception.code, 0)