Expose document type name from `IPersistenceHandler` interface

1 files changed, 10 insertions, 4 deletions

diff --git a/searchcore/src/tests/proton/persistenceengine/persistenceengine_test.cpp b/searchcore/src/tests/proton/persistenceengine/persistenceengine_test.cpp
index 787b0c365e4..21ac6893356 100644
--- a/searchcore/src/tests/proton/persistenceengine/persistenceengine_test.cpp
+++ b/searchcore/src/tests/proton/persistenceengine/persistenceengine_test.cpp
@@ -161,8 +161,9 @@ struct MyHandler : public IPersistenceHandler, IBucketFreezer {
     const Document              *document;
     std::multiset<uint64_t>      frozen;
     std::multiset<uint64_t>      was_frozen;
+    DocTypeName                  _doc_type_name;
 
-    MyHandler()
+    MyHandler(const DocTypeName &type_name)
         : initialized(false),
           lastBucket(),
           lastTimestamp(),
@@ -180,7 +181,8 @@ struct MyHandler : public IPersistenceHandler, IBucketFreezer {
           _createBucketResult(),
           document(nullptr),
           frozen(),
-          was_frozen()
+          was_frozen(),
+          _doc_type_name(type_name)
     {
     }
 
@@ -287,6 +289,10 @@ struct MyHandler : public IPersistenceHandler, IBucketFreezer {
         resultHandler.handle(Result());
     }
 
+    const DocTypeName &doc_type_name() const noexcept override {
+        return _doc_type_name;
+    }
+
     void freezeBucket(BucketId bucket) override {
         frozen.insert(bucket.getId());
         was_frozen.insert(bucket.getId());
@@ -311,8 +317,8 @@ struct HandlerSet {
 };
 
 HandlerSet::HandlerSet()
-    : phandler1(std::make_shared<MyHandler>()),
-      phandler2(std::make_shared<MyHandler>()),
+    : phandler1(std::make_shared<MyHandler>(DocTypeName("type1"))),
+      phandler2(std::make_shared<MyHandler>(DocTypeName("type2"))),
       handler1(dynamic_cast<MyHandler &>(*phandler1.get())),
       handler2(dynamic_cast<MyHandler &>(*phandler2.get()))
 {}