1 files changed, 3 insertions, 0 deletions
diff --git a/storage/src/tests/distributor/distributor_stripe_test_util.h b/storage/src/tests/distributor/distributor_stripe_test_util.h
index 272301bf4a6..2892cec6fcf 100644
--- a/storage/src/tests/distributor/distributor_stripe_test_util.h
+++ b/storage/src/tests/distributor/distributor_stripe_test_util.h
@@ -138,6 +138,7 @@ public:
 
     // TODO explicit notion of bucket spaces for tests
     DistributorBucketSpace& getDistributorBucketSpace();
+    const DistributorBucketSpace& getDistributorBucketSpace() const;
     BucketDatabase& getBucketDatabase(); // Implicit default space only
     BucketDatabase& getBucketDatabase(document::BucketSpace space);
     const BucketDatabase& getBucketDatabase() const; // Implicit default space only
@@ -175,6 +176,8 @@ public:
     void set_redundancy(uint32_t redundancy);
 
     void trigger_distribution_change(std::shared_ptr<lib::Distribution> distr);
+    void simulate_distribution_config_change(std::shared_ptr<lib::Distribution> new_config);
+    static std::shared_ptr<lib::Distribution> make_default_distribution_config(uint16_t redundancy, uint16_t node_count);
 
     using ConfigBuilder = vespa::config::content::core::StorDistributormanagerConfigBuilder;