refactor hook mechanism:

1, remove EventHook;
2, setup_hooks: could reference request dict;
3, teardown_hooks: could reference Response object.
This commit is contained in:
debugtalk
2018-05-10 13:40:47 +08:00
parent ce995a798f
commit dcc1e70181
11 changed files with 162 additions and 139 deletions

View File

@@ -68,8 +68,16 @@ def gen_random_string(str_len):
random_string = ''.join(random_char_list)
return random_string
def setup_hook_add_kwargs(method, url, kwargs):
kwargs["key"] = "value"
def setup_hook_add_kwargs(request):
request["key"] = "value"
def setup_hook_remove_kwargs(method, url, kwargs):
kwargs.pop("key")
def setup_hook_remove_kwargs(request):
request.pop("key")
def teardown_hook_sleep_N_secs(response, n_secs):
""" sleep n seconds after request
"""
if response.status_code == 200:
time.sleep(0.1)
else:
time.sleep(n_secs)