1 files changed, 132 insertions, 0 deletions
diff --git a/src/lib/cc/tests/user_context_unittests.cc b/src/lib/cc/tests/user_context_unittests.cc
new file mode 100644
index 0000000..e88b421
--- /dev/null
+++ b/src/lib/cc/tests/user_context_unittests.cc
@@ -0,0 +1,132 @@
+// Copyright (C) 2018-2020 Internet Systems Consortium, Inc. ("ISC")
+//
+// This Source Code Form is subject to the terms of the Mozilla Public
+// License, v. 2.0. If a copy of the MPL was not distributed with this
+// file, You can obtain one at http://mozilla.org/MPL/2.0/.
+
+#include <config.h>
+#include <cc/user_context.h>
+#include <gtest/gtest.h>
+
+using namespace isc::data;
+
+namespace {
+
+ElementPtr gen() {
+    std::string content = "{ \"foo\": 1, \"bar\": \"xyz\" }";
+    return (Element::fromJSON(content));
+}
+
+TEST(UserContext, setget) {
+    UserContext parent;
+    EXPECT_FALSE(parent.getContext());
+    ConstElementPtr map = gen();
+    parent.setContext(map);
+    ConstElementPtr ctx = parent.getContext();
+    EXPECT_EQ(*ctx, *map);
+}
+
+TEST(UserContext, null) {
+    UserContext parent;
+    ElementPtr map = gen();
+    parent.contextToElement(map);
+    ElementPtr expected = gen();
+    EXPECT_EQ(*expected, *map);
+}
+
+TEST(UserContext, notMap) {
+    UserContext parent;
+    ConstElementPtr ctx = Element::create("foo");
+    parent.setContext(ctx);
+    ElementPtr map = gen();
+    parent.contextToElement(map);
+    ElementPtr expected = gen();
+    expected->set("user-context", ctx);
+    EXPECT_EQ(*expected, *map);
+}
+
+TEST(UserContext, empty) {
+    UserContext parent;
+    ConstElementPtr ctx = Element::createMap();
+    parent.setContext(ctx);
+    ElementPtr map = gen();
+    parent.contextToElement(map);
+    ElementPtr expected = gen();
+    expected->set("user-context", ctx);
+    EXPECT_EQ(*expected, *map);
+}
+
+TEST(UserContext, noComment) {
+    UserContext parent;
+    ConstElementPtr ctx = Element::fromJSON("{ \"version\": 1 }");
+    parent.setContext(ctx);
+    ElementPtr map = gen();
+    parent.contextToElement(map);
+    ElementPtr expected = gen();
+    expected->set("user-context", ctx);
+    EXPECT_EQ(*expected, *map);
+};
+
+TEST(UserContext, onlyComment) {
+    UserContext parent;
+    ConstElementPtr ctx = Element::fromJSON("{ \"comment\": \"foobar\" }");
+    parent.setContext(ctx);
+    ElementPtr map = gen();
+    parent.contextToElement(map);
+    ElementPtr expected = gen();
+    expected->set("user-context", ctx);
+    EXPECT_EQ(*expected, *map);
+}
+
+TEST(UserContext, both) {
+    UserContext parent;
+    ConstElementPtr ctx =
+        Element::fromJSON("{ \"comment\": \"foobar\", \"version\": 1 }");
+    parent.setContext(ctx);
+    ElementPtr map = gen();
+    parent.contextToElement(map);
+    ElementPtr expected = gen();
+    expected->set("user-context", ctx);
+    EXPECT_EQ(*expected, *map);
+}
+
+TEST(toElement, notMap) {
+    ConstElementPtr arg = Element::create("foo");
+    ConstElementPtr result = UserContext::toElement(arg);
+    EXPECT_EQ(*result, *arg);
+}
+
+TEST(toElement, empty) {
+    ElementPtr map = gen();
+    ConstElementPtr ctx = Element::createMap();
+    map->set("user-context", ctx);
+    ConstElementPtr result = UserContext::toElement(map);
+    EXPECT_EQ(*result, *map);
+}
+
+TEST(toElement, noComment) {
+    ElementPtr map = gen();
+    ConstElementPtr ctx = Element::fromJSON("{ \"version\": 1 }");
+    map->set("user-context", ctx);
+    ConstElementPtr result = UserContext::toElement(map);
+    EXPECT_EQ(*result, *map);
+}
+
+TEST(toElement, onlyComment) {
+    ElementPtr map = gen();
+    ConstElementPtr ctx = Element::fromJSON("{ \"comment\": \"foobar\" }");
+    map->set("user-context", ctx);
+    ConstElementPtr result = UserContext::toElement(map);
+    EXPECT_EQ(*result, *map);
+}
+
+TEST(toElement, both) {
+    ElementPtr map = gen();
+    ConstElementPtr ctx =
+        Element::fromJSON("{ \"comment\": \"foobar\", \"version\": 1 }");
+    map->set("user-context", ctx);
+    ConstElementPtr result = UserContext::toElement(map);
+    EXPECT_EQ(*result, *map);
+}
+
+}