def test_deregister_unknown_hook(self):
"""
When deregistering a hook
When the module is not known
No exceptions are raised
A log entry is generated
"""
hooks = ModuleHookRegistry()
module_hook = mock.Mock()
# Ensure we do not have the module registered
module_name = "test.module.name"
hooks.register(module_name, module_hook)
# Ensure our hook was registered
assert hooks.hooks[module_name] == set([module_hook])
# Deregistering a different hook
unknown_hook = mock.Mock()
with mock.patch("ddtrace.internal.import_hooks.log") as log_mock:
hooks.deregister(module_name, unknown_hook)
# Ensure we didn't remove the other hook of ours
assert hooks.hooks[module_name] == set([module_hook])
# Ensure a log message is generated
log_mock.debug.assert_has_calls([
mock.call("No hook %r registered for module %r", mock.ANY,
"test.module.name")
])
def test_deregister(self):
"""
When deregistering a hook
The hook is removed from the registry
The hook is not called when the registry hooks call
"""
hooks = ModuleHookRegistry()
module_hook = mock.Mock()
# Register the hook
module_name = "tests.test_module"
hooks.register(module_name, module_hook)
assert hooks.hooks[module_name] == set([module_hook])
# Deregister the hook
hooks.deregister(module_name, module_hook)
# Ensure it was removed
assert hooks.hooks[module_name] == set()
import tests.test_module
hooks.call(module_name, module=tests.test_module)
# Ensure it was not called
module_hook.assert_not_called()
def test_register_module_not_loaded(self):
"""
When registering a module hook
When the module is not already loaded
We register the hook for the module name
The module hook is not called
"""
hooks = ModuleHookRegistry()
module_hook = mock.Mock()
module_name = "test.module.name"
hooks.register(module_name, module_hook)
# The hook was registered
assert len(hooks.hooks) == 1
assert module_name in hooks.hooks
assert len(hooks.hooks[module_name]) == 1
assert hooks.hooks[module_name] == set([module_hook])
# The hook was not called
module_hook.assert_not_called()
def test_register_module_loaded(self):
"""
When registering a module hook
When the module is already loaded
We register the hook for the module name
We immediately call the hook function
"""
hooks = ModuleHookRegistry()
module_name = "tests.test_module"
import tests.test_module
module_hook = mock.Mock()
hooks.register(module_name, module_hook)
assert len(hooks.hooks) == 1
assert module_name in hooks.hooks
assert len(hooks.hooks[module_name]) == 1
assert hooks.hooks[module_name] == set([module_hook])
# Assert it was called once with the appropriate arguments
module_hook.assert_called_once_with(tests.test_module)
def test_call_after_import(self):
"""
When calling module hooks
All hooks registered for the module are called
"""
hooks = ModuleHookRegistry()
module_name = "tests.test_module"
hook_one = mock.Mock()
hook_two = mock.Mock()
hook_three = mock.Mock()
hooks.register(module_name, hook_one)
hooks.register(module_name, hook_two)
hooks.register(module_name, hook_three)
import tests.test_module
hooks.call(module_name)
# Assert all hooks were called with the module
hook_one.assert_called_once_with(tests.test_module)
hook_two.assert_called_once_with(tests.test_module)
hook_three.assert_called_once_with(tests.test_module)