Use ClusterState from DistributorBucketSpace in PendingBucketSpaceDbTransition.

1 files changed, 7 insertions, 2 deletions

diff --git a/storage/src/tests/distributor/bucketdbupdatertest.cpp b/storage/src/tests/distributor/bucketdbupdatertest.cpp
index 7831328460d..08262e30fe6 100644
--- a/storage/src/tests/distributor/bucketdbupdatertest.cpp
+++ b/storage/src/tests/distributor/bucketdbupdatertest.cpp
@@ -182,12 +182,17 @@ protected:
 
     bool bucketExistsThatHasNode(int bucketCount, uint16_t node) const;
 
-    ClusterInformation::CSP createClusterInfo(const std::string& clusterState) {
+    ClusterInformation::CSP createClusterInfo(const std::string& clusterStateString) {
+        auto clusterState = std::make_shared<lib::ClusterState>(clusterStateString);
         ClusterInformation::CSP clusterInfo(
                 new SimpleClusterInformation(
                         getBucketDBUpdater().getDistributorComponent().getIndex(),
-                        lib::ClusterState(clusterState),
+                        *clusterState,
                         "ui"));
+        auto &repo = getBucketSpaceRepo();
+        for (auto &elem : repo) {
+            elem.second->setClusterState(clusterState);
+        }
         return clusterInfo;
     }