library/cpp/testing/unittest/plugin.cpp


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
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);
        }

    }
}