Hawkin
/
arangodb
mirror of https://gitee.com/bigwinds/arangodb
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
arangodb/tests/VocBase/vocbase-test.cpp

376 lines
17 KiB
C++
Raw Permalink Blame History

////////////////////////////////////////////////////////////////////////////////
/// 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/Servers.h"
#include "RestServer/DatabaseFeature.h"
#include "RestServer/QueryRegistryFeature.h"
#include "RestServer/ViewTypesFeature.h"
#include "Sharding/ShardingFeature.h"
#include "StorageEngine/EngineSelectorFeature.h"
#include "VocBase/LogicalCollection.h"
#include "VocBase/LogicalView.h"
#include "velocypack/Parser.h"
#include <memory>
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<TestView>(vocbase, definition, planVersion);
return arangodb::Result();
}
};
} // namespace
// -----------------------------------------------------------------------------
// --SECTION-- setup / tear-down
// -----------------------------------------------------------------------------
class VocbaseTest : public ::testing::Test {
protected:
arangodb::tests::mocks::MockAqlServer server;
ViewFactory viewFactory;
VocbaseTest() {
// register view factory
server.getFeature<arangodb::ViewTypesFeature>().emplace(
arangodb::LogicalDataSource::Type::emplace(arangodb::velocypack::StringRef("testViewType")),
viewFactory);
}
};
// -----------------------------------------------------------------------------
// --SECTION-- test suite
// -----------------------------------------------------------------------------
TEST_F(VocbaseTest, test_isAllowedName) {
// direct (non-system)
std::string const ratherLong(256, 'x');
std::string const tooLong(257, 'x');
{
EXPECT_FALSE(TRI_vocbase_t::IsAllowedName(false, arangodb::velocypack::StringRef(nullptr, 0)));
EXPECT_FALSE(TRI_vocbase_t::IsAllowedName(false, arangodb::velocypack::StringRef("")));
EXPECT_TRUE(TRI_vocbase_t::IsAllowedName(false, arangodb::velocypack::StringRef("abc123")));
EXPECT_FALSE(TRI_vocbase_t::IsAllowedName(false, arangodb::velocypack::StringRef("123abc")));
EXPECT_FALSE(TRI_vocbase_t::IsAllowedName(false, arangodb::velocypack::StringRef("123")));
EXPECT_FALSE(TRI_vocbase_t::IsAllowedName(false, arangodb::velocypack::StringRef("_123")));
EXPECT_FALSE(TRI_vocbase_t::IsAllowedName(false, arangodb::velocypack::StringRef("_abc")));
EXPECT_TRUE(TRI_vocbase_t::IsAllowedName(false, arangodb::velocypack::StringRef("abcdefghijklmnopqrstuvwxyz0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789")));
EXPECT_TRUE(TRI_vocbase_t::IsAllowedName(false, arangodb::velocypack::StringRef(ratherLong)));
EXPECT_FALSE(TRI_vocbase_t::IsAllowedName(false, arangodb::velocypack::StringRef(tooLong))); // longer than TRI_COL_NAME_LENGTH
}
// direct (system)
{
EXPECT_FALSE(TRI_vocbase_t::IsAllowedName(true, arangodb::velocypack::StringRef(nullptr, 0)));
EXPECT_FALSE(TRI_vocbase_t::IsAllowedName(true, arangodb::velocypack::StringRef("")));
EXPECT_TRUE(TRI_vocbase_t::IsAllowedName(true, arangodb::velocypack::StringRef("abc123")));
EXPECT_FALSE(TRI_vocbase_t::IsAllowedName(true, arangodb::velocypack::StringRef("123abc")));
EXPECT_FALSE(TRI_vocbase_t::IsAllowedName(true, arangodb::velocypack::StringRef("123")));
EXPECT_TRUE(TRI_vocbase_t::IsAllowedName(true, arangodb::velocypack::StringRef("_123")));
EXPECT_TRUE(TRI_vocbase_t::IsAllowedName(true, arangodb::velocypack::StringRef("_abc")));
EXPECT_TRUE(TRI_vocbase_t::IsAllowedName(true, arangodb::velocypack::StringRef("abcdefghijklmnopqrstuvwxyz0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789")));
EXPECT_TRUE(TRI_vocbase_t::IsAllowedName(true, arangodb::velocypack::StringRef(ratherLong)));
EXPECT_FALSE(TRI_vocbase_t::IsAllowedName(true, arangodb::velocypack::StringRef(tooLong))); // longer than TRI_COL_NAME_LENGTH
}
// slice (default)
{
auto json0 = arangodb::velocypack::Parser::fromJson("{ }");
EXPECT_FALSE(TRI_vocbase_t::IsAllowedName(json0->slice()));
auto json1 = arangodb::velocypack::Parser::fromJson("{ \"name\": \"\" }");
EXPECT_FALSE(TRI_vocbase_t::IsAllowedName(json1->slice()));
auto json2 =
arangodb::velocypack::Parser::fromJson("{ \"name\": \"abc123\" }");
EXPECT_TRUE(TRI_vocbase_t::IsAllowedName(json2->slice()));
auto json3 =
arangodb::velocypack::Parser::fromJson("{ \"name\": \"123abc\" }");
EXPECT_FALSE(TRI_vocbase_t::IsAllowedName(json3->slice()));
auto json4 =
arangodb::velocypack::Parser::fromJson("{ \"name\": \"123\" }");
EXPECT_FALSE(TRI_vocbase_t::IsAllowedName(json4->slice()));
auto json5 =
arangodb::velocypack::Parser::fromJson("{ \"name\": \"_123\" }");
EXPECT_FALSE(TRI_vocbase_t::IsAllowedName(json5->slice()));
auto json6 =
arangodb::velocypack::Parser::fromJson("{ \"name\": \"_abc\" }");
EXPECT_FALSE(TRI_vocbase_t::IsAllowedName(json6->slice()));
auto json7 = arangodb::velocypack::Parser::fromJson(
"{ \"name\": "
"\"abcdefghijklmnopqrstuvwxyz0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZ012345"
"6789\" }");
EXPECT_TRUE(TRI_vocbase_t::IsAllowedName(json7->slice()));
auto json8 = arangodb::velocypack::Parser::fromJson("{\"name\":\"" + tooLong + "\"}");
EXPECT_FALSE(TRI_vocbase_t::IsAllowedName(json8->slice()));
}
// slice (non-system)
{
auto json0 =
arangodb::velocypack::Parser::fromJson("{ \"isSystem\": false }");
EXPECT_FALSE(TRI_vocbase_t::IsAllowedName(json0->slice()));
auto json1 = arangodb::velocypack::Parser::fromJson(
"{ \"isSystem\": false, \"name\": \"\" }");
EXPECT_FALSE(TRI_vocbase_t::IsAllowedName(json1->slice()));
auto json2 = arangodb::velocypack::Parser::fromJson(
"{ \"isSystem\": false, \"name\": \"abc123\" }");
EXPECT_TRUE(TRI_vocbase_t::IsAllowedName(json2->slice()));
auto json3 = arangodb::velocypack::Parser::fromJson(
"{ \"isSystem\": false, \"name\": \"123abc\" }");
EXPECT_FALSE(TRI_vocbase_t::IsAllowedName(json3->slice()));
auto json4 = arangodb::velocypack::Parser::fromJson(
"{ \"isSystem\": false, \"name\": \"123\" }");
EXPECT_FALSE(TRI_vocbase_t::IsAllowedName(json4->slice()));
auto json5 = arangodb::velocypack::Parser::fromJson(
"{\"isSystem\": false, \"name\": \"_123\" }");
EXPECT_FALSE(TRI_vocbase_t::IsAllowedName(json5->slice()));
auto json6 = arangodb::velocypack::Parser::fromJson(
"{ \"isSystem\": false, \"name\": \"_abc\" }");
EXPECT_FALSE(TRI_vocbase_t::IsAllowedName(json6->slice()));
auto json7 = arangodb::velocypack::Parser::fromJson(
"{ \"isSystem\": false, \"name\": 123 }");
EXPECT_FALSE(TRI_vocbase_t::IsAllowedName(json7->slice()));
auto json8 = arangodb::velocypack::Parser::fromJson(
"{ \"isSystem\": 123, \"name\": \"abc\" }");
EXPECT_TRUE(TRI_vocbase_t::IsAllowedName(json8->slice()));
auto json9 = arangodb::velocypack::Parser::fromJson(
"{ \"isSystem\": false, \"name\": "
"\"abcdefghijklmnopqrstuvwxyz0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZ012345"
"6789\" }");
EXPECT_TRUE(TRI_vocbase_t::IsAllowedName(json9->slice()));
auto json10 = arangodb::velocypack::Parser::fromJson("{\"isSystem\":false, \"name\":\"" + tooLong + "\"}");
EXPECT_FALSE(TRI_vocbase_t::IsAllowedName(json10->slice()));
}
// slice (system)
{
auto json0 =
arangodb::velocypack::Parser::fromJson("{ \"isSystem\": true }");
EXPECT_FALSE(TRI_vocbase_t::IsAllowedName(json0->slice()));
auto json1 = arangodb::velocypack::Parser::fromJson(
"{ \"isSystem\": true, \"name\": \"\" }");
EXPECT_FALSE(TRI_vocbase_t::IsAllowedName(json1->slice()));
auto json2 = arangodb::velocypack::Parser::fromJson(
"{ \"isSystem\": true, \"name\": \"abc123\" }");
EXPECT_TRUE(TRI_vocbase_t::IsAllowedName(json2->slice()));
auto json3 = arangodb::velocypack::Parser::fromJson(
"{ \"isSystem\": true, \"name\": \"123abc\" }");
EXPECT_FALSE(TRI_vocbase_t::IsAllowedName(json3->slice()));
auto json4 = arangodb::velocypack::Parser::fromJson(
"{ \"isSystem\": true, \"name\": \"123\" }");
EXPECT_FALSE(TRI_vocbase_t::IsAllowedName(json4->slice()));
auto json5 = arangodb::velocypack::Parser::fromJson(
"{ \"isSystem\": true, \"name\": \"_123\" }");
EXPECT_TRUE(TRI_vocbase_t::IsAllowedName(json5->slice()));
auto json6 = arangodb::velocypack::Parser::fromJson(
"{ \"isSystem\": true, \"name\": \"_abc\" }");
EXPECT_TRUE(TRI_vocbase_t::IsAllowedName(json6->slice()));
auto json7 = arangodb::velocypack::Parser::fromJson(
"{ \"isSystem\": true, \"name\": 123 }");
EXPECT_FALSE(TRI_vocbase_t::IsAllowedName(json7->slice()));
auto json8 = arangodb::velocypack::Parser::fromJson(
"{ \"isSystem\": 123, \"name\": \"_abc\" }");
EXPECT_FALSE(TRI_vocbase_t::IsAllowedName(json8->slice()));
auto json9 = arangodb::velocypack::Parser::fromJson(
"{ \"isSystem\": true, \"name\": "
"\"abcdefghijklmnopqrstuvwxyz0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZ012345"
"6789\" }");
EXPECT_TRUE(TRI_vocbase_t::IsAllowedName(json9->slice()));
auto json10 = arangodb::velocypack::Parser::fromJson(
"{ \"isSystem\": true, \"name\": \"" + tooLong + "\"}");
EXPECT_FALSE(TRI_vocbase_t::IsAllowedName(json10->slice()));
}
}
TEST_F(VocbaseTest, test_isSystemName) {
EXPECT_FALSE(TRI_vocbase_t::IsSystemName(""));
EXPECT_TRUE(TRI_vocbase_t::IsSystemName("_"));
EXPECT_TRUE(TRI_vocbase_t::IsSystemName("_abc"));
EXPECT_FALSE(TRI_vocbase_t::IsSystemName("abc"));
}
TEST_F(VocbaseTest, test_lookupDataSource) {
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, testDBInfo(server.server()));
// not present collection (no datasource)
{
EXPECT_FALSE(vocbase.lookupDataSource(100));
EXPECT_FALSE(vocbase.lookupDataSource("100"));
EXPECT_FALSE(vocbase.lookupDataSource("testCollection"));
EXPECT_FALSE(vocbase.lookupDataSource("testCollectionGUID"));
EXPECT_FALSE(vocbase.lookupCollection(100));
EXPECT_FALSE(vocbase.lookupCollection("100"));
EXPECT_FALSE(vocbase.lookupCollection("testCollection"));
EXPECT_FALSE(vocbase.lookupCollection("testCollectionGUID"));
}
// not present view (no datasource)
{
EXPECT_FALSE(vocbase.lookupDataSource(200));
EXPECT_FALSE(vocbase.lookupDataSource("200"));
EXPECT_FALSE(vocbase.lookupDataSource("testView"));
EXPECT_FALSE(vocbase.lookupDataSource("testViewGUID"));
EXPECT_FALSE(vocbase.lookupView(200));
EXPECT_FALSE(vocbase.lookupView("200"));
EXPECT_FALSE(vocbase.lookupView("testView"));
EXPECT_FALSE(vocbase.lookupView("testViewGUID"));
}
auto collection = vocbase.createCollection(collectionJson->slice());
auto view = vocbase.createView(viewJson->slice());
EXPECT_FALSE(collection->deleted());
EXPECT_FALSE(view->deleted());
// not present collection (is view)
{
EXPECT_FALSE(!vocbase.lookupDataSource(200));
EXPECT_FALSE(!vocbase.lookupDataSource("200"));
EXPECT_FALSE(!vocbase.lookupDataSource("testView"));
EXPECT_FALSE(vocbase.lookupDataSource("testViewGUID"));
EXPECT_FALSE(vocbase.lookupCollection(200));
EXPECT_FALSE(vocbase.lookupCollection("200"));
EXPECT_FALSE(vocbase.lookupCollection("testView"));
EXPECT_FALSE(vocbase.lookupCollection("testViewGUID"));
EXPECT_FALSE(vocbase.lookupCollectionByUuid("testView"));
EXPECT_FALSE(vocbase.lookupCollectionByUuid("testViewGUID"));
}
// not preset view (is collection)
{
EXPECT_FALSE(!vocbase.lookupDataSource(100));
EXPECT_FALSE(!vocbase.lookupDataSource("100"));
EXPECT_FALSE(!vocbase.lookupDataSource("testCollection"));
EXPECT_FALSE(!vocbase.lookupDataSource("testCollectionGUID"));
EXPECT_FALSE(vocbase.lookupView(100));
EXPECT_FALSE(vocbase.lookupView("100"));
EXPECT_FALSE(vocbase.lookupView("testCollection"));
EXPECT_FALSE(vocbase.lookupView("testCollectionGUID"));
}
// present collection
{
EXPECT_FALSE(!vocbase.lookupDataSource(100));
EXPECT_FALSE(!vocbase.lookupDataSource("100"));
EXPECT_FALSE(!vocbase.lookupDataSource("testCollection"));
EXPECT_FALSE(!vocbase.lookupDataSource("testCollectionGUID"));
EXPECT_FALSE(!vocbase.lookupCollection(100));
EXPECT_FALSE(!vocbase.lookupCollection("100"));
EXPECT_FALSE(!vocbase.lookupCollection("testCollection"));
EXPECT_FALSE(!vocbase.lookupCollection("testCollectionGUID"));
EXPECT_FALSE(vocbase.lookupCollectionByUuid("testCollection"));
EXPECT_TRUE(
(false == !vocbase.lookupCollectionByUuid("testCollectionGUID")));
}
// present view
{
EXPECT_FALSE(!vocbase.lookupDataSource(200));
EXPECT_FALSE(!vocbase.lookupDataSource("200"));
EXPECT_FALSE(!vocbase.lookupDataSource("testView"));
EXPECT_FALSE(vocbase.lookupDataSource("testViewGUID"));
EXPECT_FALSE(!vocbase.lookupView(200));
EXPECT_FALSE(!vocbase.lookupView("200"));
EXPECT_FALSE(!vocbase.lookupView("testView"));
EXPECT_FALSE(vocbase.lookupView("testViewGUID"));
}
EXPECT_TRUE(vocbase.dropCollection(collection->id(), true, 0).ok());
EXPECT_TRUE(view->drop().ok());
EXPECT_TRUE(collection->deleted());
EXPECT_TRUE(view->deleted());
// not present collection (deleted)
{
EXPECT_FALSE(vocbase.lookupDataSource(100));
EXPECT_FALSE(vocbase.lookupDataSource("100"));
EXPECT_FALSE(vocbase.lookupDataSource("testCollection"));
EXPECT_FALSE(vocbase.lookupDataSource("testCollectionGUID"));
EXPECT_FALSE(vocbase.lookupCollection(100));
EXPECT_FALSE(vocbase.lookupCollection("100"));
EXPECT_FALSE(vocbase.lookupCollection("testCollection"));
EXPECT_FALSE(vocbase.lookupCollection("testCollectionGUID"));
EXPECT_FALSE(vocbase.lookupCollectionByUuid("testCollection"));
EXPECT_TRUE(
(true == !vocbase.lookupCollectionByUuid("testCollectionGUID")));
}
// not present view (deleted)
{
EXPECT_FALSE(vocbase.lookupDataSource(200));
EXPECT_FALSE(vocbase.lookupDataSource("200"));
EXPECT_FALSE(vocbase.lookupDataSource("testView"));
EXPECT_FALSE(vocbase.lookupDataSource("testViewGUID"));
EXPECT_FALSE(vocbase.lookupView(200));
EXPECT_FALSE(vocbase.lookupView("200"));
EXPECT_FALSE(vocbase.lookupView("testView"));
EXPECT_FALSE(vocbase.lookupView("testViewGUID"));
EXPECT_FALSE(vocbase.lookupCollectionByUuid("testCollection"));
EXPECT_TRUE(
(true == !vocbase.lookupCollectionByUuid("testCollectionGUID")));
}
}
Reference in New Issue View Git Blame Copy Permalink