diff options
Diffstat (limited to 'storage/src/tests/distributor/distributortestutil.cpp')
-rw-r--r-- | storage/src/tests/distributor/distributortestutil.cpp | 12 |
1 files changed, 8 insertions, 4 deletions
diff --git a/storage/src/tests/distributor/distributortestutil.cpp b/storage/src/tests/distributor/distributortestutil.cpp index 2802b976256..b465edf5d16 100644 --- a/storage/src/tests/distributor/distributortestutil.cpp +++ b/storage/src/tests/distributor/distributortestutil.cpp @@ -4,6 +4,7 @@ #include <vespa/document/test/make_bucket_space.h> #include <vespa/document/test/make_document_bucket.h> #include <vespa/storage/distributor/distributor.h> +#include <vespa/storage/distributor/distributor_stripe.h> #include <vespa/storage/distributor/distributor_bucket_space.h> #include <vespa/storage/distributor/distributorcomponent.h> #include <vespa/vdslib/distribution/distribution.h> @@ -257,6 +258,7 @@ DistributorTestUtil::removeFromBucketDB(const document::BucketId& id) void DistributorTestUtil::addIdealNodes(const document::BucketId& id) { + // TODO STRIPE roundabout way of getting state bundle..! addIdealNodes(*distributor_component().getClusterStateBundle().getBaselineClusterState(), id); } @@ -338,20 +340,21 @@ DistributorTestUtil::disableBucketActivationInConfig(bool disable) BucketDBUpdater& DistributorTestUtil::getBucketDBUpdater() { - return _distributor->_bucketDBUpdater; + return _distributor->bucket_db_updater(); } IdealStateManager& DistributorTestUtil::getIdealStateManager() { - return _distributor->_idealStateManager; + return _distributor->ideal_state_manager(); } ExternalOperationHandler& DistributorTestUtil::getExternalOperationHandler() { - return _distributor->_externalOperationHandler; + return _distributor->external_operation_handler(); } storage::distributor::DistributorComponent& DistributorTestUtil::distributor_component() { - return _distributor->_component; + // TODO STRIPE tests use this to indirectly access bucket space repos/DBs! + return _distributor->distributor_component(); } bool @@ -369,6 +372,7 @@ DistributorTestUtil::tick() { DistributorConfiguration& DistributorTestUtil::getConfig() { + // TODO STRIPE avoid const cast return const_cast<DistributorConfiguration&>(_distributor->getConfig()); } |