1 files changed, 55 insertions, 0 deletions
diff --git a/searchlib/src/tests/common/matching_elements_fields/matching_elements_fields_test.cpp b/searchlib/src/tests/common/matching_elements_fields/matching_elements_fields_test.cpp
new file mode 100644
index 00000000000..766be30f8f1
--- /dev/null
+++ b/searchlib/src/tests/common/matching_elements_fields/matching_elements_fields_test.cpp
@@ -0,0 +1,55 @@
+// Copyright 2019 Oath Inc. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
+
+#include <vespa/vespalib/gtest/gtest.h>
+#include <vespa/searchlib/common/matching_elements_fields.h>
+
+using namespace search;
+
+namespace {
+
+std::string str(const vespalib::string &s) { return std::string(s.data(), s.size()); }
+
+}
+
+struct MatchingElementsFieldsTest : ::testing::Test {
+    MatchingElementsFields fields;
+    MatchingElementsFieldsTest() : fields() {
+        fields.add_mapping("foo", "foo.a");
+        fields.add_mapping("foo", "foo.b");
+        fields.add_mapping("bar", "bar.x");
+        fields.add_mapping("bar", "bar.y");
+        fields.add_field("baz");
+    }
+    ~MatchingElementsFieldsTest() = default;
+};
+
+TEST_F(MatchingElementsFieldsTest, require_that_field_can_be_identified) {
+    EXPECT_TRUE(fields.has_field("foo"));
+    EXPECT_TRUE(fields.has_field("bar"));
+    EXPECT_TRUE(fields.has_field("baz"));
+    EXPECT_TRUE(!fields.has_field("foo.a"));
+    EXPECT_TRUE(!fields.has_field("bar.x"));
+    EXPECT_TRUE(!fields.has_field("bogus"));
+}
+
+TEST_F(MatchingElementsFieldsTest, require_that_struct_field_can_be_identified) {
+    EXPECT_TRUE(!fields.has_struct_field("foo"));
+    EXPECT_TRUE(!fields.has_struct_field("bar"));
+    EXPECT_TRUE(!fields.has_struct_field("baz"));
+    EXPECT_TRUE(fields.has_struct_field("foo.a"));
+    EXPECT_TRUE(fields.has_struct_field("bar.x"));
+    EXPECT_TRUE(!fields.has_struct_field("bogus"));
+}
+
+TEST_F(MatchingElementsFieldsTest, require_that_struct_field_maps_to_enclosing_field_name) {
+    EXPECT_EQ(str(fields.get_enclosing_field("foo.a")), str("foo"));
+    EXPECT_EQ(str(fields.get_enclosing_field("foo.b")), str("foo"));
+    EXPECT_EQ(str(fields.get_enclosing_field("bar.x")), str("bar"));
+    EXPECT_EQ(str(fields.get_enclosing_field("bar.y")), str("bar"));
+}
+
+TEST_F(MatchingElementsFieldsTest, require_that_nonexisting_struct_field_maps_to_empty_string) {
+    EXPECT_EQ(str(fields.get_enclosing_field("bogus")), str(""));
+}
+
+GTEST_MAIN_RUN_ALL_TESTS()