Correct tracking of docIdLimit in FixedSourceSelector.

1 files changed, 17 insertions, 0 deletions

diff --git a/searchlib/src/tests/attribute/sourceselector/sourceselector_test.cpp b/searchlib/src/tests/attribute/sourceselector/sourceselector_test.cpp
index 1eda0b7cb67..75a0ccf6ab9 100644
--- a/searchlib/src/tests/attribute/sourceselector/sourceselector_test.cpp
+++ b/searchlib/src/tests/attribute/sourceselector/sourceselector_test.cpp
@@ -49,6 +49,7 @@ private:
     template <typename SelectorType>
     void requireThatSourcesAreCountedCorrectly();
     void requireThatSourcesAreCountedCorrectly();
+    void requireThatDocIdLimitIsCorrect();
 };
 
 int
@@ -64,6 +65,7 @@ Test::Main()
     TEST_DO(requireThatSelectorCanSaveAndLoad());
     TEST_DO(requireThatCompleteSourceRangeIsHandled());
     TEST_DO(requireThatSourcesAreCountedCorrectly());
+    TEST_DO(requireThatDocIdLimitIsCorrect());
 
     TEST_DONE();
 }
@@ -219,6 +221,21 @@ Test::requireThatSourcesAreCountedCorrectly()
     requireThatSourcesAreCountedCorrectly<FixedSourceSelector>();
 }
 
+void
+Test::requireThatDocIdLimitIsCorrect()
+{
+    FixedSourceSelector selector(default_source, base_file_name);
+    EXPECT_EQUAL(0u, selector.getDocIdLimit());
+    selector.setSource(8, 10);
+    EXPECT_EQUAL(9u, selector.getDocIdLimit());
+    selector.compactLidSpace(4);
+    EXPECT_EQUAL(4u, selector.getDocIdLimit());
+    selector.setSource(6, 10);
+    EXPECT_EQUAL(7u, selector.getDocIdLimit());
+    auto selector2 = selector.cloneAndSubtract(base_file_name2, 3);
+    EXPECT_EQUAL(7u, selector2->getDocIdLimit());
+}
+
 }  // namespace
 
 TEST_APPHOOK(Test);