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& 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& get_storage_service() = 0; + virtual sharded& get_task_manager() = 0; + data_dictionary::database data_dictionary(); };