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/unittest/plugin.cpp | |
download | ydb-1110808a9d39d4b808aef724c861a2e1a38d2a69.tar.gz |
intermediate changes
ref:cde9a383711a11544ce7e107a78147fb96cc4029
Diffstat (limited to 'library/cpp/testing/unittest/plugin.cpp')
-rw-r--r-- | library/cpp/testing/unittest/plugin.cpp | 50 |
1 files changed, 50 insertions, 0 deletions
diff --git a/library/cpp/testing/unittest/plugin.cpp b/library/cpp/testing/unittest/plugin.cpp new file mode 100644 index 0000000000..543112f7ac --- /dev/null +++ b/library/cpp/testing/unittest/plugin.cpp @@ -0,0 +1,50 @@ +#include "plugin.h" + +#include <util/generic/singleton.h> +#include <util/generic/vector.h> +#include <util/generic/utility.h> + +namespace NUnitTest { + namespace NPlugin { + namespace { + class TPlugins { + public: + void OnStartMain(int argc, char* argv[]) const { + for (const auto& plugin : Plugins) { + plugin->OnStartMain(argc, argv); + } + } + + void OnStopMain(int argc, char* argv[]) const { + for (const auto& plugin : Plugins) { + plugin->OnStopMain(argc, argv); + } + } + + void Register(TSimpleSharedPtr<IPlugin> plugin) { + Plugins.emplace_back(std::move(plugin)); + } + + static TPlugins& Instance() { + return *Singleton<TPlugins>(); + } + + private: + TVector<TSimpleSharedPtr<IPlugin>> Plugins; + }; + } // anonymous namespace + + TPluginRegistrator::TPluginRegistrator(TSimpleSharedPtr<IPlugin> plugin) { + TPlugins::Instance().Register(std::move(plugin)); + } + + void OnStartMain(int argc, char* argv[]) { + TPlugins::Instance().OnStartMain(argc, argv); + } + + void OnStopMain(int argc, char* argv[]) { + TPlugins::Instance().OnStopMain(argc, argv); + } + + } +} |