mirror of https://gitee.com/bigwinds/arangodb
276 lines
10 KiB
C++
276 lines
10 KiB
C++
////////////////////////////////////////////////////////////////////////////////
|
|
/// DISCLAIMER
|
|
///
|
|
/// Copyright 2018 ArangoDB GmbH, Cologne, Germany
|
|
///
|
|
/// Licensed under the Apache License, Version 2.0 (the "License");
|
|
/// 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 holder is ArangoDB GmbH, Cologne, Germany
|
|
///
|
|
/// @author Andrey Abramov
|
|
/// @author Vasiliy Nabatchikov
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
|
|
#include "catch.hpp"
|
|
#include "../IResearch/common.h"
|
|
#include "../IResearch/StorageEngineMock.h"
|
|
#include "IResearch/ApplicationServerHelper.h"
|
|
#include "RestServer/DatabaseFeature.h"
|
|
#include "RestServer/QueryRegistryFeature.h"
|
|
#include "RestServer/ViewTypesFeature.h"
|
|
#include "StorageEngine/EngineSelectorFeature.h"
|
|
#include "Utils/CollectionNameResolver.h"
|
|
#include "VocBase/LogicalCollection.h"
|
|
#include "VocBase/LogicalView.h"
|
|
#include "velocypack/Parser.h"
|
|
|
|
namespace {
|
|
|
|
std::shared_ptr<arangodb::LogicalView> makeTestView(
|
|
TRI_vocbase_t& vocbase,
|
|
arangodb::velocypack::Slice const& info,
|
|
uint64_t planVersion,
|
|
arangodb::LogicalView::PreCommitCallback const& preCommit
|
|
) {
|
|
struct Impl: public arangodb::DBServerLogicalView {
|
|
Impl(
|
|
TRI_vocbase_t& vocbase,
|
|
arangodb::velocypack::Slice const& info,
|
|
uint64_t planVersion
|
|
): arangodb::DBServerLogicalView(vocbase, info, planVersion) {
|
|
}
|
|
arangodb::Result create() { return DBServerLogicalView::create(*this); }
|
|
virtual arangodb::Result dropImpl() override { return arangodb::Result(); }
|
|
virtual void getPropertiesVPack(
|
|
arangodb::velocypack::Builder&,
|
|
bool
|
|
) const override {
|
|
}
|
|
virtual void open() override {}
|
|
virtual arangodb::Result updateProperties(
|
|
arangodb::velocypack::Slice const&,
|
|
bool
|
|
) override {
|
|
return arangodb::Result();
|
|
}
|
|
virtual bool visitCollections(
|
|
std::function<bool(TRI_voc_cid_t)> const&
|
|
) const override {
|
|
return true;
|
|
}
|
|
};
|
|
|
|
auto view = std::make_shared<Impl>(vocbase, info, planVersion);
|
|
|
|
return
|
|
(!preCommit || preCommit(std::static_pointer_cast<arangodb::LogicalView>(view)))
|
|
&& view->create().ok()
|
|
? view : nullptr;
|
|
}
|
|
|
|
}
|
|
|
|
// -----------------------------------------------------------------------------
|
|
// --SECTION-- setup / tear-down
|
|
// -----------------------------------------------------------------------------
|
|
|
|
struct CollectionNameResolverSetup {
|
|
StorageEngineMock engine;
|
|
arangodb::application_features::ApplicationServer server;
|
|
std::vector<std::pair<arangodb::application_features::ApplicationFeature*, bool>> features;
|
|
|
|
CollectionNameResolverSetup(): server(nullptr, nullptr) {
|
|
arangodb::EngineSelectorFeature::ENGINE = &engine;
|
|
|
|
// setup required application features
|
|
features.emplace_back(new arangodb::DatabaseFeature(&server), false); // required for TRI_vocbase_t::dropCollection(...)
|
|
features.emplace_back(new arangodb::QueryRegistryFeature(&server), false); // required for TRI_vocbase_t instantiation
|
|
features.emplace_back(new arangodb::ViewTypesFeature(&server), false); // required for TRI_vocbase_t::createView(...)
|
|
|
|
for (auto& f: features) {
|
|
arangodb::application_features::ApplicationServer::server->addFeature(f.first);
|
|
}
|
|
|
|
for (auto& f: features) {
|
|
f.first->prepare();
|
|
}
|
|
|
|
for (auto& f: features) {
|
|
if (f.second) {
|
|
f.first->start();
|
|
}
|
|
}
|
|
|
|
// register view factory
|
|
arangodb::iresearch::getFeature<arangodb::ViewTypesFeature>()->emplace(
|
|
arangodb::LogicalDataSource::Type::emplace(
|
|
arangodb::velocypack::StringRef("testViewType")
|
|
),
|
|
makeTestView
|
|
);
|
|
}
|
|
|
|
~CollectionNameResolverSetup() {
|
|
arangodb::application_features::ApplicationServer::server = nullptr;
|
|
arangodb::EngineSelectorFeature::ENGINE = nullptr;
|
|
|
|
// destroy application features
|
|
for (auto& f: features) {
|
|
if (f.second) {
|
|
f.first->stop();
|
|
}
|
|
}
|
|
|
|
for (auto& f: features) {
|
|
f.first->unprepare();
|
|
}
|
|
}
|
|
};
|
|
|
|
// -----------------------------------------------------------------------------
|
|
// --SECTION-- test suite
|
|
// -----------------------------------------------------------------------------
|
|
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
/// @brief setup
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
|
|
TEST_CASE("CollectionNameResolverTest", "[vocbase]") {
|
|
CollectionNameResolverSetup s;
|
|
(void)(s);
|
|
|
|
SECTION("test_getDataSource") {
|
|
auto collectionJson = arangodb::velocypack::Parser::fromJson("{ \"globallyUniqueId\": \"testCollectionGUID\", \"id\": 100, \"name\": \"testCollection\" }");
|
|
auto viewJson = arangodb::velocypack::Parser::fromJson("{ \"id\": 200, \"name\": \"testView\", \"type\": \"testViewType\" }"); // any arbitrary view type
|
|
Vocbase vocbase(TRI_vocbase_type_e::TRI_VOCBASE_TYPE_NORMAL, 1, "testVocbase");
|
|
arangodb::CollectionNameResolver resolver(&vocbase);
|
|
|
|
// not present collection (no datasource)
|
|
{
|
|
CHECK((true == !resolver.getDataSource(100)));
|
|
CHECK((true == !resolver.getDataSource("100")));
|
|
CHECK((true == !resolver.getDataSource("testCollection")));
|
|
CHECK((true == !resolver.getDataSource("testCollectionGUID")));
|
|
CHECK((true == !resolver.getCollection(100)));
|
|
CHECK((true == !resolver.getCollection("100")));
|
|
CHECK((true == !resolver.getCollection("testCollection")));
|
|
CHECK((true == !resolver.getCollection("testCollectionGUID")));
|
|
}
|
|
|
|
// not present view (no datasource)
|
|
{
|
|
CHECK((true == !resolver.getDataSource(200)));
|
|
CHECK((true == !resolver.getDataSource("200")));
|
|
CHECK((true == !resolver.getDataSource("testView")));
|
|
CHECK((true == !resolver.getDataSource("testViewGUID")));
|
|
CHECK((true == !resolver.getView(200)));
|
|
CHECK((true == !resolver.getView("200")));
|
|
CHECK((true == !resolver.getView("testView")));
|
|
CHECK((true == !resolver.getView("testViewGUID")));
|
|
}
|
|
|
|
auto* collection = vocbase.createCollection(collectionJson->slice());
|
|
auto view = vocbase.createView(viewJson->slice());
|
|
|
|
CHECK((false == collection->deleted()));
|
|
CHECK((false == view->deleted()));
|
|
|
|
// not present collection (is view)
|
|
{
|
|
CHECK((false == !resolver.getDataSource(200)));
|
|
CHECK((false == !resolver.getDataSource("200")));
|
|
CHECK((false == !resolver.getDataSource("testView")));
|
|
CHECK((true == !resolver.getDataSource("testViewGUID")));
|
|
CHECK((true == !resolver.getCollection(200)));
|
|
CHECK((true == !resolver.getCollection("200")));
|
|
CHECK((true == !resolver.getCollection("testView")));
|
|
CHECK((true == !resolver.getCollection("testViewGUID")));
|
|
}
|
|
|
|
// not preset view (is collection)
|
|
{
|
|
CHECK((false == !resolver.getDataSource(100)));
|
|
CHECK((false == !resolver.getDataSource("100")));
|
|
CHECK((false == !resolver.getDataSource("testCollection")));
|
|
CHECK((false == !resolver.getDataSource("testCollectionGUID")));
|
|
CHECK((true == !resolver.getView(100)));
|
|
CHECK((true == !resolver.getView("100")));
|
|
CHECK((true == !resolver.getView("testCollection")));
|
|
CHECK((true == !resolver.getView("testCollectionGUID")));
|
|
}
|
|
|
|
// present collection
|
|
{
|
|
CHECK((false == !resolver.getDataSource(100)));
|
|
CHECK((false == !resolver.getDataSource("100")));
|
|
CHECK((false == !resolver.getDataSource("testCollection")));
|
|
CHECK((false == !resolver.getDataSource("testCollectionGUID")));
|
|
CHECK((false == !resolver.getCollection(100)));
|
|
CHECK((false == !resolver.getCollection("100")));
|
|
CHECK((false == !resolver.getCollection("testCollection")));
|
|
CHECK((false == !resolver.getCollection("testCollectionGUID")));
|
|
}
|
|
|
|
// present view
|
|
{
|
|
CHECK((false == !resolver.getDataSource(200)));
|
|
CHECK((false == !resolver.getDataSource("200")));
|
|
CHECK((false == !resolver.getDataSource("testView")));
|
|
CHECK((true == !resolver.getDataSource("testViewGUID")));
|
|
CHECK((false == !resolver.getView(200)));
|
|
CHECK((false == !resolver.getView("200")));
|
|
CHECK((false == !resolver.getView("testView")));
|
|
CHECK((true == !resolver.getView("testViewGUID")));
|
|
}
|
|
|
|
CHECK((TRI_ERROR_NO_ERROR == vocbase.dropCollection(collection, true, 0)));
|
|
CHECK((true == vocbase.dropView(*view).ok()));
|
|
CHECK((true == collection->deleted()));
|
|
CHECK((true == view->deleted()));
|
|
|
|
// present collection (deleted, cached)
|
|
{
|
|
CHECK((false == !resolver.getDataSource(100)));
|
|
CHECK((false == !resolver.getDataSource("100")));
|
|
CHECK((false == !resolver.getDataSource("testCollection")));
|
|
CHECK((false == !resolver.getDataSource("testCollectionGUID")));
|
|
CHECK((false == !resolver.getCollection(100)));
|
|
CHECK((false == !resolver.getCollection("100")));
|
|
CHECK((false == !resolver.getCollection("testCollection")));
|
|
CHECK((false == !resolver.getCollection("testCollectionGUID")));
|
|
CHECK((true == resolver.getCollection(100)->deleted()));
|
|
}
|
|
|
|
// present view (deleted, cached)
|
|
{
|
|
CHECK((false == !resolver.getDataSource(200)));
|
|
CHECK((false == !resolver.getDataSource("200")));
|
|
CHECK((false == !resolver.getDataSource("testView")));
|
|
CHECK((true == !resolver.getDataSource("testViewGUID")));
|
|
CHECK((false == !resolver.getView(200)));
|
|
CHECK((false == !resolver.getView("200")));
|
|
CHECK((false == !resolver.getView("testView")));
|
|
CHECK((true == !resolver.getView("testViewGUID")));
|
|
CHECK((true == resolver.getView(200)->deleted()));
|
|
}
|
|
}
|
|
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
/// @brief generate tests
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
|
|
}
|
|
|
|
// -----------------------------------------------------------------------------
|
|
// --SECTION-- END-OF-FILE
|
|
// -----------------------------------------------------------------------------
|