wire in missing tests

2 files changed, 2 insertions, 0 deletions

diff --git a/searchlib/src/tests/query/customtypevisitor_test.cpp b/searchlib/src/tests/query/customtypevisitor_test.cpp
index a941694d375..c5eeac8543d 100644
--- a/searchlib/src/tests/query/customtypevisitor_test.cpp
+++ b/searchlib/src/tests/query/customtypevisitor_test.cpp
@@ -139,6 +139,7 @@ Test::Main()
     TEST_CALL(requireThatNodeIsVisited<MyONear>);
     TEST_CALL(requireThatNodeIsVisited<MyOr>);
     TEST_CALL(requireThatNodeIsVisited<MyPhrase>);
+    TEST_CALL(requireThatNodeIsVisited<MySameElement>);
     TEST_CALL(requireThatNodeIsVisited<MyRangeTerm>);
     TEST_CALL(requireThatNodeIsVisited<MyRank>);
     TEST_CALL(requireThatNodeIsVisited<MyNumberTerm>);
diff --git a/searchlib/src/tests/query/templatetermvisitor_test.cpp b/searchlib/src/tests/query/templatetermvisitor_test.cpp
index abf97af5d6c..95679c29ccb 100644
--- a/searchlib/src/tests/query/templatetermvisitor_test.cpp
+++ b/searchlib/src/tests/query/templatetermvisitor_test.cpp
@@ -76,6 +76,7 @@ void Test::requireThatAllTermsCanBeVisited() {
     EXPECT_TRUE(!checkVisit(new SimpleEquiv(17, Weight(100))));
     EXPECT_TRUE(!checkVisit(new SimpleNear(2)));
     EXPECT_TRUE(!checkVisit(new SimpleONear(2)));
+    EXPECT_TRUE(!checkVisit(new SimpleSameElement("foo")));
     EXPECT_TRUE(!checkVisit(new SimpleOr));
     EXPECT_TRUE(!checkVisit(new SimpleRank));
 }