mirror of https://gitee.com/bigwinds/arangodb
118 lines
4.0 KiB
C++
118 lines
4.0 KiB
C++
////////////////////////////////////////////////////////////////////////////////
|
|
/// DISCLAIMER
|
|
///
|
|
/// Copyright 2017-2017 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
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
|
|
#include "gtest/gtest.h"
|
|
|
|
#include "fakeit.hpp"
|
|
|
|
#include "Aql/AqlValue.h"
|
|
#include "Aql/Query.h"
|
|
#include "Cluster/ServerState.h"
|
|
#include "Graph/ClusterTraverserCache.h"
|
|
#include "Transaction/Methods.h"
|
|
|
|
#include <velocypack/Builder.h>
|
|
#include <velocypack/Slice.h>
|
|
#include <velocypack/velocypack-aliases.h>
|
|
|
|
using namespace arangodb;
|
|
using namespace arangodb::aql;
|
|
using namespace arangodb::graph;
|
|
|
|
namespace arangodb {
|
|
namespace tests {
|
|
namespace cluster_traverser_cache_test {
|
|
|
|
class ClusterTraverserCacheTest : public ::testing::Test {
|
|
protected:
|
|
ServerState* ss;
|
|
ServerState::RoleEnum oldRole;
|
|
|
|
ClusterTraverserCacheTest()
|
|
: ss(ServerState::instance()), oldRole(ss->getRole()) {
|
|
ss->setRole(ServerState::ROLE_COORDINATOR);
|
|
}
|
|
|
|
~ClusterTraverserCacheTest() { ss->setRole(oldRole); }
|
|
};
|
|
|
|
TEST_F(ClusterTraverserCacheTest, it_should_return_a_null_aqlvalue_if_vertex_not_cached) {
|
|
std::unordered_map<ServerID, traverser::TraverserEngineID> engines;
|
|
std::string vertexId = "UnitTest/Vertex";
|
|
std::string expectedMessage = "vertex '" + vertexId + "' not found";
|
|
|
|
fakeit::Mock<transaction::Methods> trxMock;
|
|
transaction::Methods& trx = trxMock.get();
|
|
|
|
fakeit::Mock<Query> queryMock;
|
|
Query& query = queryMock.get();
|
|
fakeit::When(Method(queryMock, trx)).AlwaysReturn(&trx);
|
|
fakeit::When(OverloadedMethod(queryMock, registerWarning, void(int, const char*)))
|
|
.Do([&](int code, char const* message) {
|
|
ASSERT_TRUE(code == TRI_ERROR_ARANGO_DOCUMENT_NOT_FOUND);
|
|
ASSERT_TRUE(strcmp(message, expectedMessage.c_str()) == 0);
|
|
});
|
|
|
|
ClusterTraverserCache testee(&query, &engines);
|
|
|
|
// NOTE: we do not put anything into the cache, so we get null for any vertex
|
|
AqlValue val = testee.fetchVertexAqlResult(arangodb::velocypack::StringRef(vertexId));
|
|
ASSERT_TRUE(val.isNull(false));
|
|
fakeit::Verify(OverloadedMethod(queryMock, registerWarning, void(int, const char*)))
|
|
.Exactly(1);
|
|
}
|
|
|
|
TEST_F(ClusterTraverserCacheTest, it_should_insert_a_null_vpack_if_vertex_not_cached) {
|
|
std::unordered_map<ServerID, traverser::TraverserEngineID> engines;
|
|
std::string vertexId = "UnitTest/Vertex";
|
|
std::string expectedMessage = "vertex '" + vertexId + "' not found";
|
|
|
|
fakeit::Mock<transaction::Methods> trxMock;
|
|
transaction::Methods& trx = trxMock.get();
|
|
|
|
fakeit::Mock<Query> queryMock;
|
|
Query& query = queryMock.get();
|
|
fakeit::When(Method(queryMock, trx)).AlwaysReturn(&trx);
|
|
fakeit::When(OverloadedMethod(queryMock, registerWarning, void(int, const char*)))
|
|
.Do([&](int code, char const* message) {
|
|
ASSERT_TRUE(code == TRI_ERROR_ARANGO_DOCUMENT_NOT_FOUND);
|
|
ASSERT_TRUE(strcmp(message, expectedMessage.c_str()) == 0);
|
|
});
|
|
|
|
VPackBuilder result;
|
|
|
|
ClusterTraverserCache testee(&query, &engines);
|
|
|
|
// NOTE: we do not put anything into the cache, so we get null for any vertex
|
|
testee.insertVertexIntoResult(arangodb::velocypack::StringRef(vertexId), result);
|
|
|
|
VPackSlice sl = result.slice();
|
|
ASSERT_TRUE(sl.isNull());
|
|
|
|
fakeit::Verify(OverloadedMethod(queryMock, registerWarning, void(int, const char*)))
|
|
.Exactly(1);
|
|
}
|
|
|
|
} // namespace cluster_traverser_cache_test
|
|
} // namespace tests
|
|
} // namespace arangodb
|