Fix linter errors

This commit is contained in:
Luiz Silva 2019-11-08 15:09:58 -03:00
parent 42fcdbd98c
commit 0f532b03ce
2 changed files with 8 additions and 3 deletions

View File

@ -16,7 +16,11 @@ class StickyMetaParamsMiddleware(object):
self.keys_to_sticky = keys_to_sticky
def process_spider_output(self, response, result, spider):
sticky_meta = {k: response.meta[k] for k in self.keys_to_sticky if k in response.meta}
sticky_meta = {
k: response.meta[k]
for k in self.keys_to_sticky
if k in response.meta
}
for r in result:
if not isinstance(r, Request):
yield r

View File

@ -47,8 +47,9 @@ class TestStickyMetaParamsMiddleware(TestCase):
response = Response(self.test_url, request=request)
result = [
Request(self.test_url),
MockItem(name='dummy') # Add a item just to increase the test coverage
MockItem(name='dummy')
]
for result in middleware.process_spider_output(response, result, spider):
results = middleware.process_spider_output(response, result, spider)
for result in results:
if isinstance(result, Request):
self.assertEqual(result.meta, {'param2': 'Stickied!'})