diff options
author | Devtools Arcadia <arcadia-devtools@yandex-team.ru> | 2022-02-07 18:08:42 +0300 |
---|---|---|
committer | Devtools Arcadia <arcadia-devtools@mous.vla.yp-c.yandex.net> | 2022-02-07 18:08:42 +0300 |
commit | 1110808a9d39d4b808aef724c861a2e1a38d2a69 (patch) | |
tree | e26c9fed0de5d9873cce7e00bc214573dc2195b7 /library/cpp/testing/hook/hook.cpp | |
download | ydb-1110808a9d39d4b808aef724c861a2e1a38d2a69.tar.gz |
intermediate changes
ref:cde9a383711a11544ce7e107a78147fb96cc4029
Diffstat (limited to 'library/cpp/testing/hook/hook.cpp')
-rw-r--r-- | library/cpp/testing/hook/hook.cpp | 45 |
1 files changed, 45 insertions, 0 deletions
diff --git a/library/cpp/testing/hook/hook.cpp b/library/cpp/testing/hook/hook.cpp new file mode 100644 index 0000000000..b3c599da89 --- /dev/null +++ b/library/cpp/testing/hook/hook.cpp @@ -0,0 +1,45 @@ +#include "hook.h" + +namespace { + NTesting::THook* BeforeInitHead = nullptr; + NTesting::THook* BeforeRunHead = nullptr; + NTesting::THook* AfterRunHead = nullptr; + + void RegisterHook(NTesting::THook*& head, NTesting::THook* hook) { + hook->Next = head; + head = hook; + } + + void CallHooks(NTesting::THook* head) { + while (nullptr != head) { + if (nullptr != head->Fn) { + (*head->Fn)(); + } + head = head->Next; + } + } +} + +void NTesting::THook::RegisterBeforeInit(NTesting::THook* hook) noexcept { + RegisterHook(BeforeInitHead, hook); +} + +void NTesting::THook::CallBeforeInit() { + CallHooks(BeforeInitHead); +} + +void NTesting::THook::RegisterBeforeRun(NTesting::THook* hook) noexcept { + RegisterHook(BeforeRunHead, hook); +} + +void NTesting::THook::CallBeforeRun() { + CallHooks(BeforeRunHead); +} + +void NTesting::THook::RegisterAfterRun(NTesting::THook* hook) noexcept { + RegisterHook(AfterRunHead, hook); +} + +void NTesting::THook::CallAfterRun() { + CallHooks(AfterRunHead); +} |