Add checks for number of documents in target attribute.

2 files changed, 4 insertions, 0 deletions

diff --git a/searchlib/src/tests/attribute/imported_attribute_vector/imported_attribute_vector_test.cpp b/searchlib/src/tests/attribute/imported_attribute_vector/imported_attribute_vector_test.cpp
index 7d0bd9bc065..e55344aded0 100644
--- a/searchlib/src/tests/attribute/imported_attribute_vector/imported_attribute_vector_test.cpp
+++ b/searchlib/src/tests/attribute/imported_attribute_vector/imported_attribute_vector_test.cpp
@@ -245,8 +245,10 @@ TEST_F("Original target lid range is used by read guard", Fixture)
     reset_with_single_value_reference_mappings<IntegerAttribute, int32_t>(
             f, BasicType::INT32,
             {});
+    EXPECT_EQUAL(11u, f.target_attr->getNumDocs());
     auto first_guard = f.get_imported_attr();
     add_n_docs_with_undefined_values(*f.target_attr, 1);
+    EXPECT_EQUAL(12u, f.target_attr->getNumDocs());
     auto typed_target_attr = f.template target_attr_as<IntegerAttribute>();
     ASSERT_TRUE(typed_target_attr->update(11, 2345));
     f.target_attr->commit();
diff --git a/searchlib/src/tests/attribute/imported_search_context/imported_search_context_test.cpp b/searchlib/src/tests/attribute/imported_search_context/imported_search_context_test.cpp
index fac1284f1a6..19327245083 100644
--- a/searchlib/src/tests/attribute/imported_search_context/imported_search_context_test.cpp
+++ b/searchlib/src/tests/attribute/imported_search_context/imported_search_context_test.cpp
@@ -431,8 +431,10 @@ TEST_F("original lid range is used by search context", SingleValueFixture)
 
 TEST_F("Original target lid range is used by search context", SingleValueFixture)
 {
+    EXPECT_EQUAL(11u, f.target_attr->getNumDocs());
     auto first_ctx = f.create_context(word_term("2345"));
     add_n_docs_with_undefined_values(*f.target_attr, 1);
+    EXPECT_EQUAL(12u, f.target_attr->getNumDocs());
     auto typed_target_attr = f.template target_attr_as<IntegerAttribute>();
     ASSERT_TRUE(typed_target_attr->update(11, 2345));
     f.target_attr->commit();