Files
scylladb/test/lib/test_services.hh
Raphael S. Carvalho 7d82373e3a test: Rename column_family_for_tests to table_for_tests
To avoid confusion, as replica::column_family was already renamed
to replica::table.

Signed-off-by: Raphael S. Carvalho <raphaelsc@scylladb.com>
2022-10-05 21:37:19 -03:00

83 lines
2.7 KiB
C++

/*
* This file is open source software, licensed to you under the terms
* of the Apache License, Version 2.0 (the "License"). See the NOTICE file
* distributed with this work for additional information regarding copyright
* ownership. You may not use this file except in compliance with the License.
*
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
* KIND, either express or implied. See the License for the
* specific language governing permissions and limitations
* under the License.
*/
/*
* Copyright (C) 2016-present ScyllaDB
*/
#pragma once
#include <memory>
#include <seastar/core/seastar.hh>
#include "schema.hh"
#include "schema_builder.hh"
#include "row_cache.hh"
#include "replica/database.hh"
#include "cell_locking.hh"
#include "compaction/compaction_manager.hh"
#include "compaction/table_state.hh"
#include "sstables/sstables_manager.hh"
extern db::config test_db_config;
extern gms::feature_service test_feature_service;
struct table_for_tests {
class table_state;
struct data {
schema_ptr s;
reader_concurrency_semaphore semaphore;
cache_tracker tracker;
replica::cf_stats cf_stats{0};
replica::column_family::config cfg;
cell_locker_stats cl_stats;
compaction_manager cm{compaction_manager::for_testing_tag{}};
lw_shared_ptr<replica::column_family> cf;
std::unique_ptr<table_state> table_s;
data();
~data();
};
lw_shared_ptr<data> _data;
explicit table_for_tests(sstables::sstables_manager& sstables_manager);
explicit table_for_tests(sstables::sstables_manager& sstables_manager, schema_ptr s, std::optional<sstring> datadir = {});
schema_ptr schema() { return _data->s; }
const replica::cf_stats& cf_stats() const noexcept { return _data->cf_stats; }
operator lw_shared_ptr<replica::column_family>() { return _data->cf; }
replica::column_family& operator*() { return *_data->cf; }
replica::column_family* operator->() { return _data->cf.get(); }
compaction_manager& get_compaction_manager() noexcept { return _data->cm; }
compaction::table_state& as_table_state() noexcept;
future<> stop();
future<> stop_and_keep_alive() {
return stop().finally([cf = *this] {});
}
};
dht::token create_token_from_key(const dht::i_partitioner&, sstring key);
range<dht::token> create_token_range_from_keys(const dht::sharder& sharder, const dht::i_partitioner&, sstring start_key, sstring end_key);