library/cpp/testing/unittest/gtest.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
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67

#include "gtest.h" 
#include "simple.h"
 
#include <util/generic/map.h> 
#include <util/generic/vector.h> 
#include <util/system/type_name.h>
 
using namespace NUnitTest; 
using namespace NUnitTest::NPrivate; 
 
IGTestFactory::~IGTestFactory() {
} 
 
namespace { 
    struct TCurrentTest: public TSimpleTestExecutor {
        inline TCurrentTest(TStringBuf name) 
            : MyName(name) 
        { 
        } 
 
        TString TypeId() const override {
            return TypeName(*this) + "-" + MyName;
        } 
 
        TString Name() const noexcept override {
            return TString(MyName);
        } 
 
        const TStringBuf MyName; 
    }; 
 
    struct TGTestFactory: public IGTestFactory { 
        inline TGTestFactory(TStringBuf name) 
            : Test(name) 
        { 
        } 
 
        ~TGTestFactory() override {
        } 
 
        TString Name() const noexcept override {
            return Test.Name(); 
        } 
 
        TTestBase* ConstructTest() override {
            return new TCurrentTest(Test); 
        } 
 
        void AddTest(const char* name, void (*body)(TTestContext&), bool forceFork) override {
            Test.Tests.push_back(TBaseTestCase(name, body, forceFork));
        } 
 
        TCurrentTest Test; 
    }; 
} 
 
IGTestFactory* NUnitTest::NPrivate::ByName(const char* name) { 
    static TMap<TStringBuf, TAutoPtr<TGTestFactory>> tests;
 
    auto& ret = tests[name]; 
 
    if (!ret) { 
        ret = new TGTestFactory(name); 
    } 
 
    return ret.Get(); 
}