67 lines
2.0 KiB
Python
67 lines
2.0 KiB
Python
from unittest.mock import MagicMock
|
|
|
|
from behave import given, then
|
|
|
|
from openfeature.exception import ErrorCode
|
|
from openfeature.flag_evaluation import Reason
|
|
from openfeature.hook import Hook
|
|
|
|
|
|
@given("a client with added hook")
|
|
def step_impl_add_hook(context):
|
|
hook = MagicMock(spec=Hook)
|
|
hook.before = MagicMock()
|
|
hook.after = MagicMock()
|
|
hook.error = MagicMock()
|
|
hook.finally_after = MagicMock()
|
|
context.hook = hook
|
|
context.client.add_hooks([hook])
|
|
|
|
|
|
@then('the "{hook_name}" hook should have been executed')
|
|
def step_impl_should_called(context, hook_name):
|
|
hook = get_hook_from_name(context, hook_name)
|
|
assert hook.called
|
|
|
|
|
|
@then('the "{hook_names}" hooks should be called with evaluation details')
|
|
def step_impl_should_have_eval_details(context, hook_names):
|
|
for hook_name in hook_names.split(", "):
|
|
hook = get_hook_from_name(context, hook_name)
|
|
for row in context.table:
|
|
flag_type, key, value = row
|
|
|
|
value = convert_value_from_key_and_flag_type(value, key, flag_type)
|
|
actual = hook.call_args[1]["details"].__dict__[key]
|
|
|
|
assert actual == value
|
|
|
|
|
|
def get_hook_from_name(context, hook_name):
|
|
if hook_name.lower() == "before":
|
|
return context.hook.before
|
|
elif hook_name.lower() == "after":
|
|
return context.hook.after
|
|
elif hook_name.lower() == "error":
|
|
return context.hook.error
|
|
elif hook_name.lower() == "finally":
|
|
return context.hook.finally_after
|
|
else:
|
|
raise ValueError(str(hook_name) + " is not a valid hook name")
|
|
|
|
|
|
def convert_value_from_key_and_flag_type(value, key, flag_type):
|
|
if value in ("None", "null"):
|
|
return None
|
|
if flag_type.lower() == "boolean":
|
|
return bool(value)
|
|
elif flag_type.lower() == "integer":
|
|
return int(value)
|
|
elif flag_type.lower() == "float":
|
|
return float(value)
|
|
elif key == "reason":
|
|
return Reason(value)
|
|
elif key == "error_code":
|
|
return ErrorCode(value)
|
|
return value
|