//////////////////////////////////////////////////////////////////////////////// /// 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 "gtest/gtest.h" #include "../IResearch/common.h" #include "../Mocks/StorageEngineMock.h" #include "RestServer/DatabaseFeature.h" #include "RestServer/QueryRegistryFeature.h" #include "RestServer/ViewTypesFeature.h" #include "Sharding/ShardingFeature.h" #include "StorageEngine/EngineSelectorFeature.h" #include "Utils/CollectionNameResolver.h" #include "VocBase/LogicalCollection.h" #include "VocBase/LogicalView.h" #include "velocypack/Parser.h" namespace { struct TestView : public arangodb::LogicalView { TestView(TRI_vocbase_t& vocbase, arangodb::velocypack::Slice const& definition, uint64_t planVersion) : arangodb::LogicalView(vocbase, definition, planVersion) {} virtual arangodb::Result appendVelocyPackImpl( arangodb::velocypack::Builder&, Serialization) const override { return arangodb::Result(); } virtual arangodb::Result dropImpl() override { return arangodb::LogicalViewHelperStorageEngine::drop(*this); } virtual void open() override {} virtual arangodb::Result renameImpl(std::string const& oldName) override { return arangodb::LogicalViewHelperStorageEngine::rename(*this, oldName); } virtual arangodb::Result properties(arangodb::velocypack::Slice const&, bool) override { return arangodb::Result(); } virtual bool visitCollections(CollectionVisitor const& visitor) const override { return true; } }; struct ViewFactory : public arangodb::ViewFactory { virtual arangodb::Result create(arangodb::LogicalView::ptr& view, TRI_vocbase_t& vocbase, arangodb::velocypack::Slice const& definition) const override { view = vocbase.createView(definition); return arangodb::Result(); } virtual arangodb::Result instantiate(arangodb::LogicalView::ptr& view, TRI_vocbase_t& vocbase, arangodb::velocypack::Slice const& definition, uint64_t planVersion) const override { view = std::make_shared(vocbase, definition, planVersion); return arangodb::Result(); } }; } // namespace // ----------------------------------------------------------------------------- // --SECTION-- setup / tear-down // ----------------------------------------------------------------------------- class CollectionNameResolverTest : public ::testing::Test { protected: StorageEngineMock engine; arangodb::application_features::ApplicationServer server; std::vector> features; ViewFactory viewFactory; CollectionNameResolverTest() : engine(server), 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::ShardingFeature(server), false); 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::application_features::ApplicationServer::lookupFeature() ->emplace(arangodb::LogicalDataSource::Type::emplace( arangodb::velocypack::StringRef("testViewType")), viewFactory); } ~CollectionNameResolverTest() { 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 // ----------------------------------------------------------------------------- TEST_F(CollectionNameResolverTest, 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) { EXPECT_TRUE((true == !resolver.getDataSource(100))); EXPECT_TRUE((true == !resolver.getDataSource("100"))); EXPECT_TRUE((true == !resolver.getDataSource("testCollection"))); EXPECT_TRUE((true == !resolver.getDataSource("testCollectionGUID"))); EXPECT_TRUE((true == !resolver.getCollection(100))); EXPECT_TRUE((true == !resolver.getCollection("100"))); EXPECT_TRUE((true == !resolver.getCollection("testCollection"))); EXPECT_TRUE((true == !resolver.getCollection("testCollectionGUID"))); } // not present view (no datasource) { EXPECT_TRUE((true == !resolver.getDataSource(200))); EXPECT_TRUE((true == !resolver.getDataSource("200"))); EXPECT_TRUE((true == !resolver.getDataSource("testView"))); EXPECT_TRUE((true == !resolver.getDataSource("testViewGUID"))); EXPECT_TRUE((true == !resolver.getView(200))); EXPECT_TRUE((true == !resolver.getView("200"))); EXPECT_TRUE((true == !resolver.getView("testView"))); EXPECT_TRUE((true == !resolver.getView("testViewGUID"))); } auto collection = vocbase.createCollection(collectionJson->slice()); auto view = vocbase.createView(viewJson->slice()); EXPECT_TRUE((false == collection->deleted())); EXPECT_TRUE((false == view->deleted())); // not present collection (is view) { EXPECT_TRUE((false == !resolver.getDataSource(200))); EXPECT_TRUE((false == !resolver.getDataSource("200"))); EXPECT_TRUE((false == !resolver.getDataSource("testView"))); EXPECT_TRUE((true == !resolver.getDataSource("testViewGUID"))); EXPECT_TRUE((true == !resolver.getCollection(200))); EXPECT_TRUE((true == !resolver.getCollection("200"))); EXPECT_TRUE((true == !resolver.getCollection("testView"))); EXPECT_TRUE((true == !resolver.getCollection("testViewGUID"))); } // not preset view (is collection) { EXPECT_TRUE((false == !resolver.getDataSource(100))); EXPECT_TRUE((false == !resolver.getDataSource("100"))); EXPECT_TRUE((false == !resolver.getDataSource("testCollection"))); EXPECT_TRUE((false == !resolver.getDataSource("testCollectionGUID"))); EXPECT_TRUE((true == !resolver.getView(100))); EXPECT_TRUE((true == !resolver.getView("100"))); EXPECT_TRUE((true == !resolver.getView("testCollection"))); EXPECT_TRUE((true == !resolver.getView("testCollectionGUID"))); } // present collection { EXPECT_TRUE((false == !resolver.getDataSource(100))); EXPECT_TRUE((false == !resolver.getDataSource("100"))); EXPECT_TRUE((false == !resolver.getDataSource("testCollection"))); EXPECT_TRUE((false == !resolver.getDataSource("testCollectionGUID"))); EXPECT_TRUE((false == !resolver.getCollection(100))); EXPECT_TRUE((false == !resolver.getCollection("100"))); EXPECT_TRUE((false == !resolver.getCollection("testCollection"))); EXPECT_TRUE((false == !resolver.getCollection("testCollectionGUID"))); } // present view { EXPECT_TRUE((false == !resolver.getDataSource(200))); EXPECT_TRUE((false == !resolver.getDataSource("200"))); EXPECT_TRUE((false == !resolver.getDataSource("testView"))); EXPECT_TRUE((true == !resolver.getDataSource("testViewGUID"))); EXPECT_TRUE((false == !resolver.getView(200))); EXPECT_TRUE((false == !resolver.getView("200"))); EXPECT_TRUE((false == !resolver.getView("testView"))); EXPECT_TRUE((true == !resolver.getView("testViewGUID"))); } EXPECT_TRUE((true == vocbase.dropCollection(collection->id(), true, 0).ok())); EXPECT_TRUE((true == view->drop().ok())); EXPECT_TRUE((true == collection->deleted())); EXPECT_TRUE((true == view->deleted())); // present collection (deleted, cached) { EXPECT_TRUE((false == !resolver.getDataSource(100))); EXPECT_TRUE((false == !resolver.getDataSource("100"))); EXPECT_TRUE((false == !resolver.getDataSource("testCollection"))); EXPECT_TRUE((false == !resolver.getDataSource("testCollectionGUID"))); EXPECT_TRUE((false == !resolver.getCollection(100))); EXPECT_TRUE((false == !resolver.getCollection("100"))); EXPECT_TRUE((false == !resolver.getCollection("testCollection"))); EXPECT_TRUE((false == !resolver.getCollection("testCollectionGUID"))); EXPECT_TRUE((true == resolver.getCollection(100)->deleted())); } // present view (deleted, cached) { EXPECT_TRUE((false == !resolver.getDataSource(200))); EXPECT_TRUE((false == !resolver.getDataSource("200"))); EXPECT_TRUE((false == !resolver.getDataSource("testView"))); EXPECT_TRUE((true == !resolver.getDataSource("testViewGUID"))); EXPECT_TRUE((false == !resolver.getView(200))); EXPECT_TRUE((false == !resolver.getView("200"))); EXPECT_TRUE((false == !resolver.getView("testView"))); EXPECT_TRUE((true == !resolver.getView("testViewGUID"))); EXPECT_TRUE((true == resolver.getView(200)->deleted())); } }