meta db unit tests and fixes
This commit is contained in:
parent
be96d79281
commit
88736fc58a
@ -22,10 +22,11 @@
|
|||||||
#include "db/meta_db.hpp"
|
#include "db/meta_db.hpp"
|
||||||
|
|
||||||
#include "app_config.hpp"
|
#include "app_config.hpp"
|
||||||
|
#include "db/rdb_meta_db.hpp"
|
||||||
#include "db/sqlite_meta_db.hpp"
|
#include "db/sqlite_meta_db.hpp"
|
||||||
|
|
||||||
namespace repertory {
|
namespace repertory {
|
||||||
auto create_meta_db(const app_config &cfg) -> std::unique_ptr<i_meta_db> {
|
auto create_meta_db(const app_config &cfg) -> std::unique_ptr<i_meta_db> {
|
||||||
return std::make_unique<sqlite_meta_db>(cfg);
|
return std::make_unique<rdb_meta_db>(cfg);
|
||||||
}
|
}
|
||||||
} // namespace repertory
|
} // namespace repertory
|
||||||
|
@ -26,13 +26,16 @@
|
|||||||
#include "utils/error_utils.hpp"
|
#include "utils/error_utils.hpp"
|
||||||
#include "utils/path.hpp"
|
#include "utils/path.hpp"
|
||||||
#include "utils/string.hpp"
|
#include "utils/string.hpp"
|
||||||
|
#include <memory>
|
||||||
|
|
||||||
namespace {
|
namespace {
|
||||||
void create_rocksdb(
|
[[nodiscard]] auto
|
||||||
const repertory::app_config &cfg, const std::string &name,
|
create_rocksdb(const repertory::app_config &cfg, const std::string &name,
|
||||||
const std::vector<rocksdb::ColumnFamilyDescriptor> &families,
|
const std::vector<rocksdb::ColumnFamilyDescriptor> &families,
|
||||||
std::vector<rocksdb::ColumnFamilyHandle *> &handles,
|
std::vector<rocksdb::ColumnFamilyHandle *> &handles)
|
||||||
std::unique_ptr<rocksdb::DB> &db) {
|
-> std::unique_ptr<rocksdb::DB> {
|
||||||
|
REPERTORY_USES_FUNCTION_NAME();
|
||||||
|
|
||||||
rocksdb::Options options{};
|
rocksdb::Options options{};
|
||||||
options.create_if_missing = true;
|
options.create_if_missing = true;
|
||||||
options.create_missing_column_families = true;
|
options.create_missing_column_families = true;
|
||||||
@ -44,13 +47,12 @@ void create_rocksdb(
|
|||||||
options,
|
options,
|
||||||
repertory::utils::path::combine(cfg.get_data_directory(), {name}),
|
repertory::utils::path::combine(cfg.get_data_directory(), {name}),
|
||||||
families, &handles, &ptr);
|
families, &handles, &ptr);
|
||||||
if (status.ok()) {
|
if (not status.ok()) {
|
||||||
db.reset(ptr);
|
repertory::utils::error::raise_error(function_name, status.ToString());
|
||||||
return;
|
throw repertory::startup_exception(status.ToString());
|
||||||
}
|
}
|
||||||
|
|
||||||
repertory::utils::error::raise_error(__FUNCTION__, status.ToString());
|
return std::unique_ptr<rocksdb::DB>(ptr);
|
||||||
throw repertory::startup_exception(status.ToString());
|
|
||||||
}
|
}
|
||||||
} // namespace
|
} // namespace
|
||||||
|
|
||||||
@ -67,7 +69,7 @@ rdb_meta_db::rdb_meta_db(const app_config &cfg) {
|
|||||||
families.emplace_back("source", rocksdb::ColumnFamilyOptions());
|
families.emplace_back("source", rocksdb::ColumnFamilyOptions());
|
||||||
|
|
||||||
auto handles = std::vector<rocksdb::ColumnFamilyHandle *>();
|
auto handles = std::vector<rocksdb::ColumnFamilyHandle *>();
|
||||||
create_rocksdb(cfg, name, families, handles, db_);
|
db_ = create_rocksdb(cfg, name, families, handles);
|
||||||
|
|
||||||
std::size_t idx{};
|
std::size_t idx{};
|
||||||
default_family_ = handles[idx++];
|
default_family_ = handles[idx++];
|
||||||
|
@ -0,0 +1,64 @@
|
|||||||
|
/*
|
||||||
|
Copyright <2018-2024> <scott.e.graves@protonmail.com>
|
||||||
|
|
||||||
|
Permission is hereby granted, free of charge, to any person obtaining a copy
|
||||||
|
of this software and associated documentation files (the "Software"), to deal
|
||||||
|
in the Software without restriction, including without limitation the rights
|
||||||
|
to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
||||||
|
copies of the Software, and to permit persons to whom the Software is
|
||||||
|
furnished to do so, subject to the following conditions:
|
||||||
|
|
||||||
|
The above copyright notice and this permission notice shall be included in all
|
||||||
|
copies or substantial portions of the Software.
|
||||||
|
|
||||||
|
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
||||||
|
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
||||||
|
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
||||||
|
AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
||||||
|
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
||||||
|
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
|
||||||
|
SOFTWARE.
|
||||||
|
*/
|
||||||
|
#ifndef REPERTORY_TEST_INCLUDE_FIXTURES_META_DB_FIXTURE_HPP
|
||||||
|
#define REPERTORY_TEST_INCLUDE_FIXTURES_META_DB_FIXTURE_HPP
|
||||||
|
|
||||||
|
#include "test_common.hpp"
|
||||||
|
|
||||||
|
#include "app_config.hpp"
|
||||||
|
#include "db/i_meta_db.hpp"
|
||||||
|
#include "db/rdb_meta_db.hpp"
|
||||||
|
#include "db/sqlite_meta_db.hpp"
|
||||||
|
|
||||||
|
namespace repertory {
|
||||||
|
template <typename db_t> class meta_db_test : public ::testing::Test {
|
||||||
|
protected:
|
||||||
|
static std::unique_ptr<app_config> config;
|
||||||
|
static std::unique_ptr<db_t> meta_db;
|
||||||
|
|
||||||
|
protected:
|
||||||
|
static void SetUpTestCase() {
|
||||||
|
static std::uint64_t idx{};
|
||||||
|
|
||||||
|
auto cfg_directory = utils::path::combine(test::get_test_output_dir(),
|
||||||
|
{
|
||||||
|
"meta_db_test",
|
||||||
|
std::to_string(++idx),
|
||||||
|
});
|
||||||
|
config = std::make_unique<app_config>(provider_type::s3, cfg_directory);
|
||||||
|
meta_db = std::make_unique<db_t>(*config);
|
||||||
|
}
|
||||||
|
|
||||||
|
static void TearDownTestCase() {
|
||||||
|
meta_db.reset();
|
||||||
|
config.reset();
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
using meta_db_types = ::testing::Types<rdb_meta_db, sqlite_meta_db>;
|
||||||
|
|
||||||
|
template <typename db_t> std::unique_ptr<app_config> meta_db_test<db_t>::config;
|
||||||
|
|
||||||
|
template <typename db_t> std::unique_ptr<db_t> meta_db_test<db_t>::meta_db;
|
||||||
|
} // namespace repertory
|
||||||
|
|
||||||
|
#endif // REPERTORY_TEST_INCLUDE_FIXTURES_META_DB_FIXTURE_HPP
|
54
repertory/repertory_test/src/meta_db_test.cpp
Normal file
54
repertory/repertory_test/src/meta_db_test.cpp
Normal file
@ -0,0 +1,54 @@
|
|||||||
|
/*
|
||||||
|
Copyright <2018-2024> <scott.e.graves@protonmail.com>
|
||||||
|
|
||||||
|
Permission is hereby granted, free of charge, to any person obtaining a copy
|
||||||
|
of this software and associated documentation files (the "Software"), to deal
|
||||||
|
in the Software without restriction, including without limitation the rights
|
||||||
|
to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
||||||
|
copies of the Software, and to permit persons to whom the Software is
|
||||||
|
furnished to do so, subject to the following conditions:
|
||||||
|
|
||||||
|
The above copyright notice and this permission notice shall be included in all
|
||||||
|
copies or substantial portions of the Software.
|
||||||
|
|
||||||
|
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
||||||
|
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
||||||
|
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
||||||
|
AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
||||||
|
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
||||||
|
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
|
||||||
|
SOFTWARE.
|
||||||
|
*/
|
||||||
|
|
||||||
|
#include "fixtures/meta_db_fixture.hpp"
|
||||||
|
|
||||||
|
namespace {
|
||||||
|
[[nodiscard]] auto create_test_file() -> std::string {
|
||||||
|
static std::atomic<std::uint64_t> idx{};
|
||||||
|
return "/test" + std::to_string(++idx);
|
||||||
|
}
|
||||||
|
} // namespace
|
||||||
|
|
||||||
|
namespace repertory {
|
||||||
|
TYPED_TEST_CASE(meta_db_test, meta_db_types);
|
||||||
|
|
||||||
|
TYPED_TEST(meta_db_test, can_get_api_path_from_source_path) {
|
||||||
|
auto test_file = create_test_file();
|
||||||
|
auto test_source = create_test_file();
|
||||||
|
EXPECT_EQ(api_error::success,
|
||||||
|
this->meta_db->set_item_meta(test_file, META_SOURCE, test_source));
|
||||||
|
|
||||||
|
std::string api_path;
|
||||||
|
EXPECT_EQ(api_error::success,
|
||||||
|
this->meta_db->get_api_path(test_source, api_path));
|
||||||
|
EXPECT_STREQ(test_file.c_str(), api_path.c_str());
|
||||||
|
}
|
||||||
|
|
||||||
|
TYPED_TEST(meta_db_test,
|
||||||
|
get_api_path_return_item_not_found_if_source_does_not_exist) {
|
||||||
|
std::string api_path;
|
||||||
|
EXPECT_EQ(api_error::item_not_found,
|
||||||
|
this->meta_db->get_api_path(create_test_file(), api_path));
|
||||||
|
EXPECT_TRUE(api_path.empty());
|
||||||
|
}
|
||||||
|
} // namespace repertory
|
Loading…
x
Reference in New Issue
Block a user