From 4284fe94150260913f743dd8fd353fb170e64f84 Mon Sep 17 00:00:00 2001 From: Alan Hamlett Date: Tue, 11 Aug 2015 19:06:01 -0700 Subject: [PATCH] look for correct stdout --- tests/test_wakatime_base.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/tests/test_wakatime_base.py b/tests/test_wakatime_base.py index ea09ee8..5647e2b 100644 --- a/tests/test_wakatime_base.py +++ b/tests/test_wakatime_base.py @@ -29,10 +29,11 @@ class BaseTestCase(utils.TestCase): response = Response() response.status_code = 201 mock_requests.return_value = response - args = ['', '--file', 'tests/samples/emptyfile.txt', '--key', '123'] + args = ['', '--file', 'tests/samples/emptyfile.txt', '--key', '123', '--config', 'foo'] retval = main(args) self.assertEquals(retval, 0) - self.assertEquals(sys.stdout.getvalue(), '') + expected_stdout = u("Error: Could not read from config file foo\n") + self.assertEquals(sys.stdout.getvalue(), expected_stdout) self.assertEquals(sys.stderr.getvalue(), '') def test_missing_config_file(self, mock_requests): @@ -58,5 +59,5 @@ class BaseTestCase(utils.TestCase): args = ['', '--file', 'tests/samples/emptyfile.txt', '--config', 'tests/samples/bad_config.cfg'] retval = main(args) self.assertEquals(retval, 103) - self.assertTrue('ParsingError' in sys.stdout.getvalue()) + self.assertIn('ParsingError', sys.stdout.getvalue()) self.assertEquals(sys.stderr.getvalue(), '')