From e0ec37c5edbe0364b1d88668cc7719e25e6f18cc Mon Sep 17 00:00:00 2001 From: Bertrand Janin Date: Fri, 25 Dec 2015 18:55:53 -0500 Subject: [PATCH] mock can't override flush, so mock stdout instead --- flake8/tests/test_integration.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/flake8/tests/test_integration.py b/flake8/tests/test_integration.py index d1417c6..5b6efc0 100644 --- a/flake8/tests/test_integration.py +++ b/flake8/tests/test_integration.py @@ -40,17 +40,17 @@ class IntegrationTestCase(unittest.TestCase): def _job_tester(self, jobs): # mock stdout.flush so we can count the number of jobs created - with mock.patch('sys.stdout.flush') as mocked: + with mock.patch('sys.stdout') as mocked: guide, report = self.check_files(arglist=['--jobs=%s' % jobs]) if is_windows(): # The code path where guide.options.jobs gets converted to an # int is not run on windows. So, do the int conversion here. self.assertEqual(int(guide.options.jobs), jobs) # On windows, call count is always zero. - self.assertEqual(mocked.call_count, 0) + self.assertEqual(mocked.flush.call_count, 0) else: self.assertEqual(guide.options.jobs, jobs) - self.assertEqual(mocked.call_count, jobs) + self.assertEqual(mocked.flush.call_count, jobs) def test_jobs(self): self._job_tester(2)