test: cql_test_env: Expose db config

This commit is contained in:
Tomasz Grabiec
2025-02-11 14:21:30 +01:00
parent 3d01ce3707
commit f3b63bfeff
2 changed files with 8 additions and 0 deletions

View File

@@ -168,6 +168,7 @@ private:
sharded<gms::gossip_address_map> _gossip_address_map;
sharded<service::direct_fd_pinger> _fd_pinger;
sharded<cdc::cdc_service> _cdc;
db::config* _db_config;
service::raft_group0_client* _group0_client;
@@ -651,6 +652,7 @@ private:
_lang_manager.invoke_on_all(&lang::manager::start).get();
_db_config = &*cfg;
_db.start(std::ref(*cfg), dbcfg, std::ref(_mnotifier), std::ref(_feature_service), std::ref(_token_metadata), std::ref(_cm), std::ref(_sstm), std::ref(_lang_manager), std::ref(_sst_dir_semaphore), std::ref(abort_sources), utils::cross_shard_barrier()).get();
auto stop_db = defer_verbose_shutdown("database", [this] {
_db.stop().get();
@@ -1152,6 +1154,10 @@ public:
virtual sharded<qos::service_level_controller>& service_level_controller_service() override {
return _sl_controller;
}
db::config& db_config() override {
return *_db_config;
}
};
std::atomic<bool> single_node_cql_env::active = { false };

View File

@@ -194,6 +194,8 @@ public:
data_dictionary::database data_dictionary();
virtual sharded<qos::service_level_controller>& service_level_controller_service() = 0;
virtual db::config& db_config() = 0;
};
future<> do_with_cql_env(std::function<future<>(cql_test_env&)> func, cql_test_config = {}, std::optional<cql_test_init_configurables> = {});