From 4e73b4d8adaf697edbd560f83afb76b9bd310798 Mon Sep 17 00:00:00 2001
From: Pavel Emelyanov <xemul@scylladb.com>
Date: Thu, 25 Jul 2024 18:51:36 +0300
Subject: [PATCH] test/cql_test_env: Export task manager from cql test env

To be used by one of the next patches

Signed-off-by: Pavel Emelyanov <xemul@scylladb.com>
---
 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 d0e78b276c..f9120d4684 100644
--- a/test/lib/cql_test_env.cc
+++ b/test/lib/cql_test_env.cc
@@ -394,6 +394,10 @@ public:
         return _ss;
     }
 
+    virtual sharded<tasks::task_manager>& get_task_manager() override {
+        return _task_manager;
+    }
+
     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 ef533a7bec..9c25b5b14a 100644
--- a/test/lib/cql_test_env.hh
+++ b/test/lib/cql_test_env.hh
@@ -181,6 +181,8 @@ public:
 
     virtual sharded<service::storage_service>& get_storage_service() = 0;
 
+    virtual sharded<tasks::task_manager>& get_task_manager() = 0;
+
     data_dictionary::database data_dictionary();
 };