aboutsummaryrefslogtreecommitdiffstats
path: root/config/src/tests/configgen/vector_inserter.cpp
blob: 80ebf0fc95f9d872e13f27969df11a56b08c51c3 (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
// Copyright Yahoo. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.

#include <vespa/vespalib/testkit/test_kit.h>
#include <vespa/config/common/types.h>
#include <vespa/config/configgen/vector_inserter.hpp>
#include <vespa/vespalib/data/slime/slime.h>

using namespace config;
using namespace config::internal;
using namespace vespalib;
using namespace vespalib::slime;

struct MyType{
    MyType() : foo(0), bar(0) {}
    MyType(const ConfigPayload & payload)
    {
        foo = payload.get()["foo"].asLong();
        bar = payload.get()["bar"].asLong();
    }
    int foo;
    int bar;
};

TEST("require that vector of ints can be inserted") {
    std::vector<int32_t> vector;
    Slime slime;
    Cursor & root = slime.setArray();
    root.addLong(3);
    root.addLong(2);
    root.addLong(6);
    VectorInserter inserter(vector);
    root.traverse(inserter);
    ASSERT_EQUAL(3u, vector.size());
    ASSERT_EQUAL(3, vector[0]);
    ASSERT_EQUAL(2, vector[1]);
    ASSERT_EQUAL(6, vector[2]);
}

TEST("require that vector of struct can be inserted") {
    std::vector<MyType> typeVector;
    Slime slime;
    Cursor & root = slime.setArray();
    Cursor & one = root.addObject();
    one.setLong("foo", 3);
    one.setLong("bar", 4);
    Cursor & two = root.addObject();
    two.setLong("foo", 1);
    two.setLong("bar", 6);
    VectorInserter inserter(typeVector);
    root.traverse(inserter);
    ASSERT_EQUAL(2u, typeVector.size());
    ASSERT_EQUAL(3, typeVector[0].foo);
    ASSERT_EQUAL(4, typeVector[0].bar);
    ASSERT_EQUAL(1, typeVector[1].foo);
    ASSERT_EQUAL(6, typeVector[1].bar);
}

TEST("require that vector of long can be inserted") {
    std::vector<int64_t> vector;
    Slime slime;
    Cursor & root = slime.setArray();
    root.addLong(3);
    root.addLong(2);
    root.addLong(6);
    VectorInserter inserter(vector);
    root.traverse(inserter);
    ASSERT_EQUAL(3u, vector.size());
    ASSERT_EQUAL(3, vector[0]);
    ASSERT_EQUAL(2, vector[1]);
    ASSERT_EQUAL(6, vector[2]);
}

TEST("require that vector of double can be inserted") {
    std::vector<double> vector;
    Slime slime;
    Cursor & root = slime.setArray();
    root.addDouble(3.1);
    root.addDouble(2.4);
    root.addDouble(6.6);
    VectorInserter inserter(vector);
    root.traverse(inserter);
    ASSERT_EQUAL(3u, vector.size());
    ASSERT_EQUAL(3.1, vector[0]);
    ASSERT_EQUAL(2.4, vector[1]);
    ASSERT_EQUAL(6.6, vector[2]);
}

TEST("require that vector of bool can be inserted") {
    std::vector<bool> vector;
    Slime slime;
    Cursor & root = slime.setArray();
    root.addBool(true);
    root.addBool(false);
    root.addBool(true);
    VectorInserter inserter(vector);
    root.traverse(inserter);
    ASSERT_EQUAL(3u, vector.size());
    ASSERT_TRUE(vector[0]);
    ASSERT_FALSE(vector[1]);
    ASSERT_TRUE(vector[2]);
}

template<typename V>
void
verify_vector_strings_can_be_inserted(V vector) {
    Slime slime;
    Cursor & root = slime.setArray();
    root.addString("foo");
    root.addString("bar");
    root.addString("baz");
    VectorInserter inserter(vector);
    root.traverse(inserter);
    ASSERT_EQUAL(3u, vector.size());
    ASSERT_EQUAL("foo", vector[0]);
    ASSERT_EQUAL("bar", vector[1]);
    ASSERT_EQUAL("baz", vector[2]);
}

TEST("require that different vectors of strings can be inserted") {
    verify_vector_strings_can_be_inserted(std::vector<vespalib::string>());
    verify_vector_strings_can_be_inserted(StringVector());
}

TEST_MAIN() { TEST_RUN_ALL(); }