test returned value also

3 files changed, 5 insertions, 0 deletions

diff --git a/searchlib/src/tests/attribute/attribute_test.cpp b/searchlib/src/tests/attribute/attribute_test.cpp
index 67791f89d54..32ac836302a 100644
--- a/searchlib/src/tests/attribute/attribute_test.cpp
+++ b/searchlib/src/tests/attribute/attribute_test.cpp
@@ -1298,6 +1298,7 @@ AttributeTest::testWeightedSet()
             IAttributeVector::EnumHandle e;
             EXPECT_TRUE(ptr->findEnum("1", e));
             EXPECT_EQUAL(1u, ptr->findFoldedEnums("1").size());
+            EXPECT_EQUAL(e, ptr->findFoldedEnums("1")[0]);
 
         }
     }
@@ -1323,6 +1324,7 @@ AttributeTest::testWeightedSet()
             IAttributeVector::EnumHandle e;
             EXPECT_TRUE(ptr->findEnum("1", e));
             EXPECT_EQUAL(1u, ptr->findFoldedEnums("1").size());
+            EXPECT_EQUAL(e, ptr->findFoldedEnums("1")[0]);
         }
     }
     { // StringAttribute
@@ -1349,6 +1351,7 @@ AttributeTest::testWeightedSet()
             IAttributeVector::EnumHandle e;
             EXPECT_TRUE(ptr->findEnum("string00", e));
             EXPECT_EQUAL(1u, ptr->findFoldedEnums("StRiNg00").size());
+            EXPECT_EQUAL(e, ptr->findFoldedEnums("StRiNg00")[0]);
         }
     }
 }
diff --git a/searchlib/src/tests/attribute/enumstore/enumstore_test.cpp b/searchlib/src/tests/attribute/enumstore/enumstore_test.cpp
index 2eec5944f23..89dd1cfdab4 100644
--- a/searchlib/src/tests/attribute/enumstore/enumstore_test.cpp
+++ b/searchlib/src/tests/attribute/enumstore/enumstore_test.cpp
@@ -355,6 +355,7 @@ EnumStoreTest::testAddEnum(bool hasPostings)
         EXPECT_EQUAL(i, e);
         EXPECT_TRUE(ses.findEnum(unique[i].c_str(), e));
         EXPECT_EQUAL(1u, ses.findFoldedEnums(unique[i].c_str()).size());
+        EXPECT_EQUAL(e, ses.findFoldedEnums(unique[i].c_str())[0]);
         EXPECT_TRUE(ses.getEnum(datastore::EntryRef(e)) == i);
         EXPECT_TRUE(ses.findIndex(unique[i].c_str(), idx));
         EXPECT_TRUE(idx == indices[i]);
diff --git a/searchlib/src/tests/attribute/stringattribute/stringattribute_test.cpp b/searchlib/src/tests/attribute/stringattribute/stringattribute_test.cpp
index 15b0a06634c..2adfdd135df 100644
--- a/searchlib/src/tests/attribute/stringattribute/stringattribute_test.cpp
+++ b/searchlib/src/tests/attribute/stringattribute/stringattribute_test.cpp
@@ -154,6 +154,7 @@ StringAttributeTest::testMultiValue(Attribute & attr, uint32_t numDocs)
             uint32_t e;
             EXPECT_TRUE(attr.findEnum(uniqueStrings[0].c_str(), e));
             EXPECT_EQUAL(1u, attr.findFoldedEnums(uniqueStrings[0].c_str()).size());
+            EXPECT_EQUAL(e, attr.findFoldedEnums(uniqueStrings[0].c_str())[0]);
             EXPECT_TRUE(attr.getEnum(doc) == e);
         }