From d3b8f9a2840824bff26ff8a691e1eff201fbbf07 Mon Sep 17 00:00:00 2001 From: Ian Cordasco Date: Thu, 4 Jul 2013 23:41:46 -0400 Subject: [PATCH] A slow start towards rewriting the tests --- flake8/tests/test_engine.py | 17 +++++++++++++++-- 1 file changed, 15 insertions(+), 2 deletions(-) diff --git a/flake8/tests/test_engine.py b/flake8/tests/test_engine.py index 6859ff0..ba04bce 100644 --- a/flake8/tests/test_engine.py +++ b/flake8/tests/test_engine.py @@ -1,8 +1,21 @@ from flake8 import engine import unittest +import mock class TestEngine(unittest.TestCase): def test_get_style_guide(self): - g = engine.get_style_guide() - self.assertTrue(isinstance(g, engine.StyleGuide)) + with mock.patch('flake8.engine._register_extensions') as reg_ext: + reg_ext.return_value = ([], [], []) + g = engine.get_style_guide() + self.assertTrue(isinstance(g, engine.StyleGuide)) + reg_ext.assert_called_once_with() + + def test_get_style_guide_kwargs(self): + m = mock.Mock() + with mock.patch('flake8.engine.StyleGuide') as StyleGuide: + with mock.patch('flake8.engine.get_parser') as get_parser: + get_parser.return_value = (m, []) + engine.get_style_guide(foo='bar') + get_parser.assert_called_once_with() + StyleGuide.assert_called_once_with(**{'parser': m, 'foo': 'bar'})