bahman/scylla
1
0
Fork 0
Code Issues Pull Requests Actions 3 Packages Projects Releases Wiki Activity

test: cql_test_env: Expose topology_state_machine

Browse Source
This commit is contained in:
Tomasz Grabiec
2025-02-01 00:07:24 +01:00
parent 61532eb53b
commit 3bb9d2fbdb
2 changed files with 6 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
test/lib/cql_test_env.cc
View File

@@ -411,6 +411,10 @@ public:
return _token_metadata;
}
virtual sharded<service::topology_state_machine>& get_topology_state_machine() override {
return _topology_state_machine;
}
virtual future<> refresh_client_state() override {
return _core_local.invoke_on_all([] (core_local_state& state) {
return state.client_state.maybe_update_per_service_level_params();

2
test/lib/cql_test_env.hh
View File

@@ -189,6 +189,8 @@ public:
virtual sharded<locator::shared_token_metadata>& get_shared_token_metadata() = 0;
virtual sharded<service::topology_state_machine>& get_topology_state_machine() = 0;
data_dictionary::database data_dictionary();
virtual sharded<qos::service_level_controller>& service_level_controller_service() = 0;