aboutsummaryrefslogtreecommitdiffstats
path: root/library/cpp/type_info/ut/builder.cpp
blob: 43b7bb8c9bf902dfed81478a92e9f6faa05984aa (plain) (blame)
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
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
#include <library/cpp/testing/unittest/gtest.h>

#include <library/cpp/type_info/type_info.h>

#include "utils.h"

class Builder: public NTesting::TTest {
public:
    void SetUp() override {
        F = NTi::PoolFactory(false);
    }

    void TearDown() override {
        F.Reset();
    }

    NTi::IPoolTypeFactoryPtr F;
};

TEST_F(Builder, TaggedBuilder) {
    auto builder = NTi::TTaggedBuilderRaw(*F);

    ASSERT_FALSE(builder.CanBuild());
    ASSERT_FALSE(builder.HasTag());
    ASSERT_FALSE(builder.HasItem());

    UNIT_ASSERT_EQUAL(builder.GetTag(), Nothing());
    UNIT_ASSERT_EQUAL(builder.GetItem(), Nothing());

    {
        auto tag = TString("Url");
        builder.SetTag(tag);
    }

    ASSERT_FALSE(builder.CanBuild());
    ASSERT_TRUE(builder.HasTag());
    ASSERT_FALSE(builder.HasItem());

    UNIT_ASSERT_EQUAL(builder.GetTag(), MakeMaybe<TStringBuf>("Url"));
    UNIT_ASSERT_EQUAL(builder.GetItem(), Nothing());

    {
        auto ty = NTi::String();
        builder.SetItem(ty);
    }

    ASSERT_TRUE(builder.CanBuild());
    ASSERT_TRUE(builder.HasTag());
    ASSERT_TRUE(builder.HasItem());

    UNIT_ASSERT_EQUAL(builder.GetTag(), MakeMaybe<TStringBuf>("Url"));
    UNIT_ASSERT_EQUAL(builder.GetItem(), MakeMaybe<const NTi::TType*>(NTi::TStringType::InstanceRaw()));

    {
        auto tagged = builder.Build();
        ASSERT_STRICT_EQ(tagged, NTi::Tagged(NTi::String(), "Url"));
    }

    ASSERT_TRUE(builder.CanBuild());

    builder.Reset();

    ASSERT_FALSE(builder.CanBuild());
    ASSERT_FALSE(builder.HasTag());
    ASSERT_FALSE(builder.HasItem());

    UNIT_ASSERT_EQUAL(builder.GetTag(), Nothing());
    UNIT_ASSERT_EQUAL(builder.GetItem(), Nothing());

    builder.SetTag("T");
    builder.SetItem(NTi::String());

    ASSERT_TRUE(builder.CanBuild());
    ASSERT_TRUE(builder.HasTag());
    ASSERT_TRUE(builder.HasItem());

    UNIT_ASSERT_EQUAL(builder.GetTag(), MakeMaybe<TStringBuf>("T"));
    UNIT_ASSERT_EQUAL(builder.GetItem(), MakeMaybe<const NTi::TType*>(NTi::TStringType::InstanceRaw()));

    builder.DiscardItem();

    ASSERT_FALSE(builder.CanBuild());
    ASSERT_TRUE(builder.HasTag());
    ASSERT_FALSE(builder.HasItem());

    UNIT_ASSERT_EQUAL(builder.GetTag(), MakeMaybe<TStringBuf>("T"));
    UNIT_ASSERT_EQUAL(builder.GetItem(), Nothing());

    builder.DiscardTag();

    builder.SetTag("T");
    builder.SetItem(NTi::String());

    ASSERT_TRUE(builder.CanBuild());
    ASSERT_TRUE(builder.HasTag());
    ASSERT_TRUE(builder.HasItem());
}

TEST_F(Builder, TaggedBuilderChaining) {
    auto builder = NTi::TTaggedBuilderRaw(*F)
                       .SetTag("Uuu")
                       .SetItem(NTi::Optional(NTi::String()));

    ASSERT_STRICT_EQ(builder.Build(), NTi::Tagged(NTi::Optional(NTi::String()), "Uuu"));

    builder = std::move(builder)
                  .DiscardTag()
                  .SetTag("Urls");

    ASSERT_STRICT_EQ(builder.Build(), NTi::Tagged(NTi::Optional(NTi::String()), "Urls"));

    builder = std::move(builder)
                  .DiscardItem()
                  .SetItem(F->ListRaw(F->StringRaw()));

    ASSERT_STRICT_EQ(builder.Build(), NTi::Tagged(NTi::List(NTi::String()), "Urls"));

    auto type = std::move(builder)
                    .Reset()
                    .SetTag("Url")
                    .SetItem(F->StringRaw())
                    .Build();

    ASSERT_STRICT_EQ(type, NTi::Tagged(NTi::String(), "Url"));
}