diff --git a/test/boost/sstable_test.hh b/test/boost/sstable_test.hh index 30790ab29c..efc0eba8df 100644 --- a/test/boost/sstable_test.hh +++ b/test/boost/sstable_test.hh @@ -37,40 +37,6 @@ #include #include -constexpr auto la = sstables::sstable::version_types::la; -constexpr auto big = sstables::sstable::format_types::big; - -class column_family_test { - lw_shared_ptr _cf; -public: - column_family_test(lw_shared_ptr cf) : _cf(cf) {} - - void add_sstable(sstables::shared_sstable sstable) { - _cf->_sstables->insert(std::move(sstable)); - } - - // NOTE: must run in a thread - void rebuild_sstable_list(const std::vector& new_sstables, - const std::vector& sstables_to_remove) { - _cf->_sstables = _cf->build_new_sstable_list(new_sstables, sstables_to_remove).get0(); - } - - static void update_sstables_known_generation(column_family& cf, unsigned generation) { - cf.update_sstables_known_generation(generation); - } - - static uint64_t calculate_generation_for_new_table(column_family& cf) { - return cf.calculate_generation_for_new_table(); - } - - static int64_t calculate_shard_from_sstable_generation(int64_t generation) { - return column_family::calculate_shard_from_sstable_generation(generation); - } - - future try_flush_memtable_to_sstable(lw_shared_ptr mt) { - return _cf->try_flush_memtable_to_sstable(mt, sstable_write_permit::unconditional()); - } -}; namespace sstables { diff --git a/test/lib/sstable_test_env.hh b/test/lib/sstable_test_env.hh index e0ac3234a3..e126386562 100644 --- a/test/lib/sstable_test_env.hh +++ b/test/lib/sstable_test_env.hh @@ -31,6 +31,41 @@ #include "test/lib/test_services.hh" #include "test/lib/log.hh" +constexpr auto la = sstables::sstable::version_types::la; +constexpr auto big = sstables::sstable::format_types::big; + +class column_family_test { + lw_shared_ptr _cf; +public: + column_family_test(lw_shared_ptr cf) : _cf(cf) {} + + void add_sstable(sstables::shared_sstable sstable) { + _cf->_sstables->insert(std::move(sstable)); + } + + // NOTE: must run in a thread + void rebuild_sstable_list(const std::vector& new_sstables, + const std::vector& sstables_to_remove) { + _cf->_sstables = _cf->build_new_sstable_list(new_sstables, sstables_to_remove).get0(); + } + + static void update_sstables_known_generation(column_family& cf, unsigned generation) { + cf.update_sstables_known_generation(generation); + } + + static uint64_t calculate_generation_for_new_table(column_family& cf) { + return cf.calculate_generation_for_new_table(); + } + + static int64_t calculate_shard_from_sstable_generation(int64_t generation) { + return column_family::calculate_shard_from_sstable_generation(generation); + } + + future try_flush_memtable_to_sstable(lw_shared_ptr mt) { + return _cf->try_flush_memtable_to_sstable(mt, sstable_write_permit::unconditional()); + } +}; + namespace sstables { class test_env_sstables_manager : public sstables_manager {