pre-commit/flake8
1
0
Fork 0
mirror of https://github.com/PyCQA/flake8.git synced 2026-04-11 23:34:17 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

mock can't override flush, so mock stdout instead

Browse source
This commit is contained in:
Bertrand Janin 2015-12-25 18:55:53 -05:00
parent bfe978992a
commit e0ec37c5ed
1 changed files with 3 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

6
flake8/tests/test_integration.py
View file

@ -40,17 +40,17 @@ class IntegrationTestCase(unittest.TestCase):
def _job_tester(self, jobs): def _job_tester(self, jobs):
# mock stdout.flush so we can count the number of jobs created # 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]) guide, report = self.check_files(arglist=['--jobs=%s' % jobs])
if is_windows(): if is_windows():
# The code path where guide.options.jobs gets converted to an # The code path where guide.options.jobs gets converted to an
# int is not run on windows. So, do the int conversion here. # int is not run on windows. So, do the int conversion here.
self.assertEqual(int(guide.options.jobs), jobs) self.assertEqual(int(guide.options.jobs), jobs)
# On windows, call count is always zero. # On windows, call count is always zero.
self.assertEqual(mocked.call_count, 0) self.assertEqual(mocked.flush.call_count, 0)
else: else:
self.assertEqual(guide.options.jobs, jobs) self.assertEqual(guide.options.jobs, jobs)
self.assertEqual(mocked.call_count, jobs) self.assertEqual(mocked.flush.call_count, jobs)
def test_jobs(self): def test_jobs(self):
self._job_tester(2) self._job_tester(2)