From 3bb9d2fbdb79f9920c1e5e8c66544bae36f69d61 Mon Sep 17 00:00:00 2001
From: Tomasz Grabiec <tgrabiec@scylladb.com>
Date: Sat, 1 Feb 2025 00:07:24 +0100
Subject: [PATCH] test: cql_test_env: Expose topology_state_machine

---
 test/lib/cql_test_env.cc | 4 ++++
 test/lib/cql_test_env.hh | 2 ++
 2 files changed, 6 insertions(+)

diff --git a/test/lib/cql_test_env.cc b/test/lib/cql_test_env.cc
index 4335784dd9..9b77149850 100644
--- a/test/lib/cql_test_env.cc
+++ b/test/lib/cql_test_env.cc
@@ -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();
diff --git a/test/lib/cql_test_env.hh b/test/lib/cql_test_env.hh
index f313aae003..dced13e400 100644
--- a/test/lib/cql_test_env.hh
+++ b/test/lib/cql_test_env.hh
@@ -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;