open-feature
/
python-sdk
mirror of https://github.com/open-feature/python-sdk.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

test: add unit tests to assert hook_hints are called in hooks 

Signed-off-by: Tom Carrio <tom@carrio.dev>
This commit is contained in:
Tom Carrio 2022-10-29 01:58:29 -04:00
parent ca0b6f48a8
commit 63825b16cf
1 changed files with 13 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
tests/hooks/test_hook_support.py
View File

@ -7,26 +7,31 @@ from open_feature.hooks.hook_support import (
before_hooks,
error_hooks,
)
from unittest.mock import ANY
def test_error_hooks_run_error_method(mock_hook):
# Given
hook_context = HookContext("flag_key", FlagType.BOOLEAN, True, "")
hook_hints = MappingProxyType(dict())
# When
error_hooks(FlagType.BOOLEAN, hook_context, Exception, [mock_hook], {})
error_hooks(FlagType.BOOLEAN, hook_context, Exception, [mock_hook], hook_hints)
# Then
mock_hook.supports_flag_value_type.assert_called_once()
mock_hook.error.assert_called_once()
mock_hook.error.assert_called_with(hook_context, ANY, hook_hints)
def test_before_hooks_run_before_method(mock_hook):
# Given
hook_context = HookContext("flag_key", FlagType.BOOLEAN, True, "")
hook_hints = MappingProxyType(dict())
# When
before_hooks(FlagType.BOOLEAN, hook_context, [mock_hook], {})
before_hooks(FlagType.BOOLEAN, hook_context, [mock_hook], hook_hints)
# Then
mock_hook.supports_flag_value_type.assert_called_once()
mock_hook.before.assert_called_once()
mock_hook.error.assert_called_with(hook_context, hook_hints)
def test_after_hooks_run_after_method(mock_hook):
@ -35,20 +40,24 @@ def test_after_hooks_run_after_method(mock_hook):
flag_evaluation_details = FlagEvaluationDetails(
hook_context.flag_key, "val", "unknown"
)
hook_hints = MappingProxyType(dict())
# When
after_hooks(
FlagType.BOOLEAN, hook_context, flag_evaluation_details, [mock_hook], {}
FlagType.BOOLEAN, hook_context, flag_evaluation_details, [mock_hook], hook_hints
)
# Then
mock_hook.supports_flag_value_type.assert_called_once()
mock_hook.after.assert_called_once()
mock_hook.error.assert_called_with(hook_context, flag_evaluation_details, hook_hints)
def test_finally_after_hooks_run_finally_after_method(mock_hook):
# Given
hook_context = HookContext("flag_key", FlagType.BOOLEAN, True, "")
hook_hints = MappingProxyType(dict())
# When
after_all_hooks(FlagType.BOOLEAN, hook_context, [mock_hook], {})
after_all_hooks(FlagType.BOOLEAN, hook_context, [mock_hook], hook_hints)
# Then
mock_hook.supports_flag_value_type.assert_called_once()
mock_hook.finally_after.assert_called_once()
mock_hook.error.assert_called_with(hook_context, hook_hints)