diff options
Diffstat (limited to 'storage/src/tests/distributor/distributor_stripe_test_util.h')
-rw-r--r-- | storage/src/tests/distributor/distributor_stripe_test_util.h | 14 |
1 files changed, 13 insertions, 1 deletions
diff --git a/storage/src/tests/distributor/distributor_stripe_test_util.h b/storage/src/tests/distributor/distributor_stripe_test_util.h index 4b489d41f7d..707418512f9 100644 --- a/storage/src/tests/distributor/distributor_stripe_test_util.h +++ b/storage/src/tests/distributor/distributor_stripe_test_util.h @@ -40,7 +40,7 @@ class DistributorStripeTestUtil : public DoneInitializeHandler, public StripeHostInfoNotifier { public: DistributorStripeTestUtil(); - ~DistributorStripeTestUtil(); + ~DistributorStripeTestUtil() override; /** * Sets up the storage link chain. @@ -211,6 +211,18 @@ public: void simulate_set_pending_cluster_state(const vespalib::string& state_str); void clear_pending_cluster_state_bundle(); + template <typename CmdType> + requires std::is_base_of_v<api::StorageCommand, CmdType> + [[nodiscard]] std::shared_ptr<CmdType> sent_command(size_t idx) { + assert(idx < _sender.commands().size()); + auto cmd = std::dynamic_pointer_cast<CmdType>(_sender.command(idx)); + assert(cmd != nullptr); + return cmd; + } + + void config_enable_condition_probing(bool enable); + void tag_content_node_supports_condition_probing(uint16_t index, bool supported); + protected: vdstestlib::DirConfig _config; std::unique_ptr<TestDistributorApp> _node; |