1
0
Fork 0
arangodb/tests/Mocks/Servers.h

86 lines
2.1 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 "ApplicationFeatures/ApplicationServer.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 {
public:
MockServer();
virtual ~MockServer();
void init();
TRI_vocbase_t& getSystemDatabase() const;
protected:
// Implementation knows the place when all features are included
void startFeatures();
private:
// Will be called by destructor
void stopFeatures();
protected:
arangodb::application_features::ApplicationServer _server;
StorageEngineMock _engine;
std::unique_ptr<TRI_vocbase_t> _system;
std::vector<std::pair<arangodb::application_features::ApplicationFeature*, bool>> _features;
};
class MockAqlServer : public MockServer {
public:
MockAqlServer();
~MockAqlServer();
std::shared_ptr<arangodb::transaction::Methods> createFakeTransaction() const;
std::unique_ptr<arangodb::aql::Query> createFakeQuery() const;
};
} // mocks
} // tests
} // arangodb
#endif