Aplly suggestions by @victor-torres

This commit is contained in:
Eugenio Lacuesta 2019-08-29 10:51:16 -03:00
parent b6b76df057
commit 97a7d775f7
No known key found for this signature in database
GPG Key ID: DA3EF2D0913E9810
1 changed files with 6 additions and 6 deletions

View File

@ -90,9 +90,9 @@ class ContractsManager(object):
cb = request.callback
@wraps(cb)
def cb_wrapper(response):
def cb_wrapper(response, **cb_kwargs):
try:
output = cb(response, **request.cb_kwargs)
output = cb(response, **cb_kwargs)
output = list(iterate_spider_output(output))
except Exception:
case = _create_testcase(method, 'callback')
@ -121,7 +121,7 @@ class Contract(object):
cb = request.callback
@wraps(cb)
def wrapper(response):
def wrapper(response, **cb_kwargs):
try:
results.startTest(self.testcase_pre)
self.pre_process(response)
@ -133,7 +133,7 @@ class Contract(object):
else:
results.addSuccess(self.testcase_pre)
finally:
return list(iterate_spider_output(cb(response, **request.cb_kwargs)))
return list(iterate_spider_output(cb(response, **cb_kwargs)))
request.callback = wrapper
@ -144,8 +144,8 @@ class Contract(object):
cb = request.callback
@wraps(cb)
def wrapper(response):
output = list(iterate_spider_output(cb(response, **request.cb_kwargs)))
def wrapper(response, **cb_kwargs):
output = list(iterate_spider_output(cb(response, **cb_kwargs)))
try:
results.startTest(self.testcase_post)
self.post_process(output)