mirror of https://gitee.com/bigwinds/arangodb
145 lines
3.6 KiB
C++
145 lines
3.6 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 Michael Hackstein
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
|
|
#ifndef ARANGODB_TESTS_MOCKS_SERVERS_H
|
|
#define ARANGODB_TESTS_MOCKS_SERVERS_H 1
|
|
|
|
#include "StorageEngineMock.h"
|
|
|
|
#include "Agency/Store.h"
|
|
#include "ApplicationFeatures/ApplicationServer.h"
|
|
#include "Cluster/ServerState.h"
|
|
|
|
struct TRI_vocbase_t;
|
|
|
|
namespace arangodb {
|
|
|
|
namespace transaction {
|
|
class Methods;
|
|
}
|
|
|
|
namespace aql {
|
|
class Query;
|
|
}
|
|
|
|
namespace application_features {
|
|
class ApplicationFeature;
|
|
}
|
|
|
|
namespace tests {
|
|
namespace mocks {
|
|
|
|
class MockServer {
|
|
protected:
|
|
// You can only create specialized types
|
|
MockServer();
|
|
virtual ~MockServer();
|
|
|
|
public:
|
|
void init();
|
|
|
|
TRI_vocbase_t& getSystemDatabase() const;
|
|
std::string const testFilesystemPath() const {
|
|
return _testFilesystemPath;
|
|
}
|
|
|
|
protected:
|
|
// Implementation knows the place when all features are included
|
|
virtual void startFeatures();
|
|
|
|
private:
|
|
// Will be called by destructor
|
|
void stopFeatures();
|
|
|
|
protected:
|
|
|
|
arangodb::application_features::ApplicationServer::State
|
|
_oldApplicationServerState;
|
|
arangodb::application_features::ApplicationServer _server;
|
|
StorageEngineMock _engine;
|
|
std::unordered_map<arangodb::application_features::ApplicationFeature*, bool> _features;
|
|
std::string _testFilesystemPath;
|
|
|
|
};
|
|
|
|
class MockAqlServer : public MockServer {
|
|
public:
|
|
MockAqlServer();
|
|
~MockAqlServer();
|
|
|
|
std::shared_ptr<arangodb::transaction::Methods> createFakeTransaction() const;
|
|
std::unique_ptr<arangodb::aql::Query> createFakeQuery() const;
|
|
};
|
|
|
|
class MockRestServer : public MockServer {
|
|
public:
|
|
MockRestServer();
|
|
~MockRestServer();
|
|
};
|
|
|
|
class MockClusterServer : public MockServer {
|
|
public:
|
|
virtual TRI_vocbase_t* createDatabase(std::string const& name) = 0;
|
|
virtual void dropDatabase(std::string const& name) = 0;
|
|
arangodb::consensus::Store& getAgencyStore() { return _agencyStore; };
|
|
|
|
// You can only create specialized types
|
|
protected:
|
|
MockClusterServer();
|
|
~MockClusterServer();
|
|
|
|
protected:
|
|
// Implementation knows the place when all features are included
|
|
void startFeatures() override;
|
|
void agencyTrx(std::string const& key, std::string const& value);
|
|
void agencyCreateDatabase(std::string const& name);
|
|
void agencyDropDatabase(std::string const& name);
|
|
|
|
private:
|
|
arangodb::consensus::Store _agencyStore;
|
|
arangodb::ServerState::RoleEnum _oldRole;
|
|
};
|
|
|
|
class MockDBServer : public MockClusterServer {
|
|
public:
|
|
MockDBServer();
|
|
~MockDBServer();
|
|
|
|
TRI_vocbase_t* createDatabase(std::string const& name) override;
|
|
void dropDatabase(std::string const& name) override;
|
|
};
|
|
|
|
class MockCoordinator : public MockClusterServer {
|
|
public:
|
|
MockCoordinator();
|
|
~MockCoordinator();
|
|
|
|
TRI_vocbase_t* createDatabase(std::string const& name) override;
|
|
void dropDatabase(std::string const& name) override;
|
|
};
|
|
|
|
} // namespace mocks
|
|
} // namespace tests
|
|
} // namespace arangodb
|
|
|
|
#endif
|