diff --git a/test/lib/cql_test_env.cc b/test/lib/cql_test_env.cc index fc4b5874b2..38d0fb2187 100644 --- a/test/lib/cql_test_env.cc +++ b/test/lib/cql_test_env.cc @@ -132,6 +132,7 @@ public: static std::atomic active; private: sharded& _db; + sharded& _feature_service; sharded& _sstm; sharded& _proxy; sharded& _qp; @@ -188,6 +189,7 @@ private: public: single_node_cql_env( sharded& db, + sharded& feature_service, sharded& sstm, sharded& proxy, sharded& qp, @@ -203,6 +205,7 @@ public: sharded& group0_registry, sharded& sys_ks) : _db(db) + , _feature_service(feature_service) , _sstm(sstm) , _proxy(proxy) , _qp(qp) @@ -446,6 +449,10 @@ public: return _proxy; } + virtual sharded& get_feature_service() override { + return _feature_service; + } + virtual sharded& get_sstorage_manager() override { return _sstm; } @@ -983,7 +990,7 @@ public: notify_set.notify_all(configurable::system_state::started).get(); - single_node_cql_env env(db, sstm, proxy, qp, auth_service, view_builder, view_update_generator, mm_notif, mm, std::ref(sl_controller), bm, gossiper, group0_client, raft_gr, sys_ks); + single_node_cql_env env(db, feature_service, sstm, proxy, qp, auth_service, view_builder, view_update_generator, mm_notif, mm, std::ref(sl_controller), bm, gossiper, group0_client, raft_gr, sys_ks); env.start().get(); auto stop_env = defer([&env] { env.stop().get(); }); diff --git a/test/lib/cql_test_env.hh b/test/lib/cql_test_env.hh index 0f7b655239..5fb2341538 100644 --- a/test/lib/cql_test_env.hh +++ b/test/lib/cql_test_env.hh @@ -178,6 +178,8 @@ public: virtual sharded& get_storage_proxy() = 0; + virtual sharded& get_feature_service() = 0; + virtual sharded& get_sstorage_manager() = 0; data_dictionary::database data_dictionary();