From b7b00fb95669196167a5e5f76b1f3b8e2feb7b84 Mon Sep 17 00:00:00 2001 From: Jakob de Maeyer Date: Mon, 9 Nov 2015 01:18:41 +0100 Subject: [PATCH] PEP8ify add-ons and tests --- scrapy/addons/__init__.py | 6 ++++-- tests/test_addons/__init__.py | 18 ++++++++++-------- 2 files changed, 14 insertions(+), 10 deletions(-) diff --git a/scrapy/addons/__init__.py b/scrapy/addons/__init__.py index 48de99c1a..169c79eac 100644 --- a/scrapy/addons/__init__.py +++ b/scrapy/addons/__init__.py @@ -323,6 +323,7 @@ class AddonManager(Mapping): """ # Collect all active add-ons and the components they provide ws = WorkingSet('') + def add_dist(project_name, version, **kwargs): if project_name in ws.entry_keys.get('scrapy', []): raise ImportError("Component {} provided by multiple add-ons" @@ -354,8 +355,9 @@ class AddonManager(Mapping): # our own exception or is it helpful enough? if ws.find(req) is None: raise ImportError( - "Add-ons {} require or modify missing component {}" - "".format(required[reqstr]+modified[reqstr], reqstr)) + "Add-ons {} require or modify missing component {}" + "".format(required[reqstr]+modified[reqstr], reqstr) + ) mod_and_req = set(required.keys()).intersection(modified.keys()) for conflict in mod_and_req: diff --git a/tests/test_addons/__init__.py b/tests/test_addons/__init__.py index b135dd04d..32ee25ad8 100644 --- a/tests/test_addons/__init__.py +++ b/tests/test_addons/__init__.py @@ -24,6 +24,7 @@ class AddonTest(unittest.TestCase): def setUp(self): self.rawaddon = Addon() + class AddonWithAttributes(Addon): name = 'Test' version = '1.0' @@ -132,7 +133,7 @@ class AddonManagerTest(unittest.TestCase): def test_add_verifies(self): brokenaddon = self.manager.get_addon( - 'tests.test_addons.addons.BrokenAddon') + 'tests.test_addons.addons.BrokenAddon') self.assertRaises(zope.interface.exceptions.BrokenImplementation, self.manager.add, brokenaddon) @@ -145,11 +146,13 @@ class AddonManagerTest(unittest.TestCase): def test_remove(self): manager = AddonManager() + def test_gets_removed(removearg): manager.add(addonmod) self.assertIn('AddonModule', manager) manager.remove(removearg) self.assertNotIn('AddonModule', manager) + test_gets_removed('AddonModule') test_gets_removed(addonmod) test_gets_removed('tests.test_addons.addonmod') @@ -158,8 +161,7 @@ class AddonManagerTest(unittest.TestCase): self.assertRaises(KeyError, manager.remove, addons.GoodAddon()) def test_get_addon(self): - goodaddon = self.manager.get_addon( - 'tests.test_addons.addons.GoodAddon') + goodaddon = self.manager.get_addon('tests.test_addons.addons.GoodAddon') self.assertIs(goodaddon, addons.GoodAddon) loaded_addonmod = self.manager.get_addon(self.ADDONMODPATH) @@ -194,8 +196,7 @@ class AddonManagerTest(unittest.TestCase): getattr(manager, func)(*args, **kwargs) six.assertCountEqual(self, manager, ['GoodAddon', 'AddonModule']) self.assertIsInstance(manager['GoodAddon'], addons.GoodAddon) - six.assertCountEqual(self, manager.configs['GoodAddon'], - ['key']) + six.assertCountEqual(self, manager.configs['GoodAddon'], ['key']) self.assertEqual(manager.configs['GoodAddon']['key'], 'val2') # XXX: Check module equality, see above self.assertEqual(manager['AddonModule'].name, addonmod.name) @@ -284,17 +285,17 @@ class AddonManagerTest(unittest.TestCase): ua_first.assert_called_once_with(manager.configs['FirstAddon'], manager) ua_second.assert_called_once_with(manager.configs['SecondAddon'], - manager) + manager) manager.update_settings(settings) us_first.assert_called_once_with(manager.configs['FirstAddon'], settings) us_second.assert_called_once_with(manager.configs['SecondAddon'], - settings) + settings) manager.check_configuration(crawler) cc_first.assert_called_once_with(manager.configs['FirstAddon'], crawler) cc_second.assert_called_once_with(manager.configs['SecondAddon'], - crawler) + crawler) self.assertEqual(ua_first.call_count, 1) self.assertEqual(ua_second.call_count, 1) self.assertEqual(us_first.call_count, 1) @@ -321,6 +322,7 @@ class AddonManagerTest(unittest.TestCase): class FirstAddon(addons.GoodAddon): name = 'FirstAddon' + def update_addons(self, config, addons): addons.add(AddedAddon())