diff --git a/tests/urchin/cql_test_env.cc b/tests/urchin/cql_test_env.cc index 129d7d8217..2301c1d1fa 100644 --- a/tests/urchin/cql_test_env.cc +++ b/tests/urchin/cql_test_env.cc @@ -149,6 +149,10 @@ public: }); } + virtual database& local_db() override { + return _db->local(); + } + future<> start() { return _core_local.start(); } diff --git a/tests/urchin/cql_test_env.hh b/tests/urchin/cql_test_env.hh index 4a75a86296..2932ad6af1 100644 --- a/tests/urchin/cql_test_env.hh +++ b/tests/urchin/cql_test_env.hh @@ -16,6 +16,8 @@ #include "bytes.hh" #include "schema.hh" +class database; + class cql_test_env { public: virtual ~cql_test_env() {}; @@ -40,6 +42,8 @@ public: boost::any expected) = 0; virtual future<> stop() = 0; + + virtual database& local_db() = 0; }; future<::shared_ptr> make_env_for_test();