forked from allure-framework/allure-python
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathhooks.py
More file actions
68 lines (51 loc) · 1.93 KB
/
hooks.py
File metadata and controls
68 lines (51 loc) · 1.93 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
import inspect
import threading
import allure_commons
from allure_commons.logger import AllureFileLogger
from allure_behave.listener import AllureListener
from behave.configuration import Configuration
HOOKS = [
"before_feature",
"after_feature",
"before_scenario",
"after_scenario",
"before_step",
"after_step"
]
_storage = threading.local()
def wrapper(original, allured):
def hook(*args, **kwargs):
allured(*args, **kwargs)
return original(*args, **kwargs)
return hook
def allure_report(result_dir="allure_results"):
allure_hooks = AllureHooks(result_dir)
frame = inspect.currentframe()
try:
for hook_name in HOOKS:
if hook_name in frame.f_back.f_locals:
frame.f_back.f_locals[hook_name] = wrapper(frame.f_back.f_locals[hook_name],
getattr(allure_hooks, hook_name))
else:
frame.f_back.f_locals[hook_name] = getattr(allure_hooks, hook_name)
finally:
del frame
class AllureHooks(object):
def __init__(self, result_dir):
self.listener = AllureListener(Configuration())
if not hasattr(_storage, 'file_logger'):
_storage.file_logger = AllureFileLogger(result_dir)
allure_commons.plugin_manager.register(_storage.file_logger)
allure_commons.plugin_manager.register(self.listener)
def before_feature(self, context, feature):
self.listener.start_feature()
def after_feature(self, context, feature):
self.listener.stop_feature()
def before_scenario(self, context, scenario):
self.listener.start_scenario(scenario)
def after_scenario(self, context, scenario):
self.listener.stop_scenario(scenario)
def before_step(self, context, step):
self.listener.start_behave_step(step)
def after_step(self, context, step):
self.listener.stop_behave_step(step)