Publish

1 files changed, 118 insertions, 0 deletions

diff --git a/config/src/tests/configgen/vector_inserter.cpp b/config/src/tests/configgen/vector_inserter.cpp
new file mode 100644
index 00000000000..e9f16f804b3
--- /dev/null
+++ b/config/src/tests/configgen/vector_inserter.cpp
@@ -0,0 +1,118 @@
+// Copyright 2016 Yahoo Inc. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
+#include <vespa/fastos/fastos.h>
+#include <vespa/log/log.h>
+LOG_SETUP("vector_inserter");
+#include <vespa/vespalib/testkit/test_kit.h>
+#include <vespa/config/configgen/vector_inserter.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<int32_t> 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<MyType> 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<int64_t> 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<double> 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<bool> inserter(vector);
+    root.traverse(inserter);
+    ASSERT_EQUAL(3u, vector.size());
+    ASSERT_TRUE(vector[0]);
+    ASSERT_FALSE(vector[1]);
+    ASSERT_TRUE(vector[2]);
+}
+
+TEST("require that vector of string can be inserted") {
+    std::vector<vespalib::string> vector;
+    Slime slime;
+    Cursor & root = slime.setArray();
+    root.addString("foo");
+    root.addString("bar");
+    root.addString("baz");
+    VectorInserter<vespalib::string> 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_MAIN() { TEST_RUN_ALL(); }