1 files changed, 6 insertions, 0 deletions
diff --git a/searchcore/src/tests/proton/documentdb/documentbucketmover/bucketmover_common.cpp b/searchcore/src/tests/proton/documentdb/documentbucketmover/bucketmover_common.cpp
index 147161f7791..714d519cab3 100644
--- a/searchcore/src/tests/proton/documentdb/documentbucketmover/bucketmover_common.cpp
+++ b/searchcore/src/tests/proton/documentdb/documentbucketmover/bucketmover_common.cpp
@@ -1,6 +1,7 @@
 // Copyright Verizon Media. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
 
 #include "bucketmover_common.h"
+#include <vespa/searchcore/proton/documentmetastore/documentmetastore.h>
 #include <vespa/vespalib/testkit/test_macros.h>
 
 using vespalib::IDestructorCallback;
@@ -81,4 +82,9 @@ assertEqual(const document::BucketId &bucket, const proton::test::Document &doc,
     return true;
 }
 
+void
+MySubDb::setBucketState(const BucketId &bucketId, bool active) {
+    _metaStore.setBucketState(bucketId, active);
+}
+
 }