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

@@ -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();

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();
 };