1 files changed, 144 insertions, 0 deletions
diff --git a/searchcore/src/tests/proton/reference/gid_to_lid_mapper/gid_to_lid_mapper_test.cpp b/searchcore/src/tests/proton/reference/gid_to_lid_mapper/gid_to_lid_mapper_test.cpp
new file mode 100644
index 00000000000..3506e88cea5
--- /dev/null
+++ b/searchcore/src/tests/proton/reference/gid_to_lid_mapper/gid_to_lid_mapper_test.cpp
@@ -0,0 +1,144 @@
+// Copyright 2017 Yahoo Inc. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
+#include <vespa/vespalib/testkit/testapp.h>
+#include <vespa/searchcore/proton/bucketdb/bucket_db_owner.h>
+#include <vespa/searchcore/proton/documentmetastore/documentmetastore.h>
+#include <vespa/vespalib/stllike/string.h>
+#include <vespa/document/base/documentid.h>
+#include <vespa/document/bucket/bucketid.h>
+#include <vespa/searchcore/proton/reference/gid_to_lid_mapper.h>
+#include <vespa/searchcore/proton/reference/gid_to_lid_mapper_factory.h>
+#include <vespa/log/log.h>
+LOG_SETUP("gid_to_lid_mapper_test");
+
+using document::GlobalId;
+using document::BucketId;
+using document::DocumentId;
+using storage::spi::Timestamp;
+using vespalib::GenerationHandler;
+
+namespace proton {
+
+namespace {
+
+GlobalId toGid(vespalib::stringref docId) {
+    return DocumentId(docId).getGlobalId();
+}
+
+vespalib::string doc1("id:test:music::1");
+vespalib::string doc2("id:test:music::2");
+vespalib::string doc3("id:test:music::3");
+
+static constexpr uint32_t numBucketBits = UINT32_C(20);
+
+BucketId toBucketId(const GlobalId &gid) {
+    BucketId bucketId(gid.convertToBucketId());
+    bucketId.setUsedBits(numBucketBits);
+    return bucketId;
+}
+
+void assertLid(const std::unique_ptr<search::IGidToLidMapper> &mapper, const vespalib::string &docId, uint32_t lid) {
+    EXPECT_EQUAL(lid, mapper->mapGidToLid(toGid(docId)));
+}
+
+}
+
+struct Fixture
+{
+
+    BucketDBOwner::SP _bucketDB;
+    std::shared_ptr<DocumentMetaStore> _dms;
+    Timestamp _timestamp;
+    using generation_t = GenerationHandler::generation_t;
+
+    Fixture()
+        : _bucketDB(std::make_shared<BucketDBOwner>()),
+          _dms(std::make_shared<DocumentMetaStore>(_bucketDB))
+    {
+        populate();
+    }
+
+    void bumpTimeStamp() {
+        _timestamp = Timestamp(_timestamp.getValue() + 1);
+    }
+
+    void put(vespalib::stringref docId, uint32_t lid) {
+        bumpTimeStamp();
+        const GlobalId gid(toGid(docId));
+        _dms->put(gid, toBucketId(gid), _timestamp, lid);
+        _dms->commit();
+    }
+
+    uint32_t put(vespalib::stringref docId) {
+        auto inspectRes = _dms->inspect(toGid(docId));
+        uint32_t lid = inspectRes.getLid();
+        put(docId, lid);
+        return lid;
+    }
+
+    void remove(uint32_t lid) {
+        _dms->remove(lid);
+        _dms->removeComplete(lid);
+        _dms->commit();
+    }
+
+
+    void populate() {
+        put(doc1, 4);
+        put(doc2, 7);
+        _dms->constructFreeList();
+    }
+
+    std::shared_ptr<search::IGidToLidMapperFactory> getGidToLidMapperFactory() {
+        return std::make_shared<GidToLidMapperFactory>(_dms);
+    }
+
+    void assertGenerations(generation_t currentGeneration, generation_t firstUsedGeneration)
+    {
+        (void) currentGeneration;
+        (void) firstUsedGeneration;
+#if 1
+        const GenerationHandler &handler = _dms->getGenerationHandler();
+        EXPECT_EQUAL(currentGeneration, handler.getCurrentGeneration());
+        EXPECT_EQUAL(firstUsedGeneration, handler.getFirstUsedGeneration());
+#endif
+    }
+};
+
+TEST_F("Test that we can use gid mapper to get lids", Fixture)
+{
+    auto factory = f.getGidToLidMapperFactory();
+    auto mapper = factory->getMapper();
+    TEST_DO(assertLid(mapper, doc1, 4));
+    TEST_DO(assertLid(mapper, doc2, 7));
+    TEST_DO(assertLid(mapper, doc3, 0));
+}
+
+TEST_F("Test that mapper holds read guard", Fixture)
+{
+    TEST_DO(f.assertGenerations(3, 3));
+    auto factory = f.getGidToLidMapperFactory();
+    uint32_t lid3 = f.put(doc3);
+    EXPECT_EQUAL(1u, lid3);
+    TEST_DO(assertLid(factory->getMapper(), doc3, 1));
+    TEST_DO(f.assertGenerations(4, 4));
+    // Remove and readd withoug guard, old docid can be reused
+    f.remove(lid3);
+    lid3 = f.put(doc3);
+    EXPECT_EQUAL(1u, lid3);
+    TEST_DO(assertLid(factory->getMapper(), doc3, 1));
+    TEST_DO(f.assertGenerations(7, 7));
+    // Remove and readd withoug guard, old docid cannot be reused
+    auto mapper = factory->getMapper();
+    f.remove(lid3);
+    lid3 = f.put(doc3);
+    EXPECT_EQUAL(2u, lid3);
+    TEST_DO(assertLid(mapper, doc3, 2));
+    TEST_DO(f.assertGenerations(10, 7));
+}
+
+}
+
+TEST_MAIN()
+{
+    TEST_RUN_ALL();
+}