9f460914ed
Summary: GraphDbAccessor is now constructed only through GraphDb. This allows the concrete GraphDb to instantiate a concrete GraphDbAccessor. This allows us to use virtual calls, so that the implementation may be kept separate. The major downside of doing things this way is heap allocation of GraphDbAccessor. In case it turns out to be a real performance issues, another solution with pointer to static implementation may be used. InsertVertexIntoRemote is now a non-member function, which reduces coupling. It made no sense for it to be member function because it used only the public parts of GraphDbAccessor. Reviewers: msantl, mtomic, mferencevic Reviewed By: msantl Subscribers: pullbot Differential Revision: https://phabricator.memgraph.io/D1504
74 lines
2.3 KiB
C++
74 lines
2.3 KiB
C++
#include <experimental/optional>
|
|
#include <string>
|
|
|
|
#include "gflags/gflags.h"
|
|
#include "gtest/gtest.h"
|
|
|
|
#include "database/graph_db.hpp"
|
|
#include "database/graph_db_accessor.hpp"
|
|
#include "durability/recovery.hpp"
|
|
#include "query/typed_value.hpp"
|
|
|
|
static const char *usage =
|
|
"--durability-dir DURABILITY_DIR\n"
|
|
"Check that Memgraph can recover the snapshot. This tool should be "
|
|
"invoked through 'test_mg_import' wrapper, so as to check that 'mg_import' "
|
|
"tools work correctly.\n";
|
|
|
|
DEFINE_string(durability_dir, "", "Path to where the durability directory");
|
|
|
|
class RecoveryTest : public ::testing::Test {
|
|
protected:
|
|
void SetUp() override {
|
|
std::string durability_dir(FLAGS_durability_dir);
|
|
durability::RecoveryData recovery_data;
|
|
durability::RecoverOnlySnapshot(durability_dir, &db_, &recovery_data,
|
|
std::experimental::nullopt);
|
|
durability::RecoverWalAndIndexes(durability_dir, &db_, &recovery_data);
|
|
}
|
|
|
|
database::SingleNode db_;
|
|
};
|
|
|
|
TEST_F(RecoveryTest, TestVerticesRecovered) {
|
|
auto dba = db_.Access();
|
|
EXPECT_EQ(dba->VerticesCount(), 10);
|
|
EXPECT_EQ(dba->VerticesCount(dba->Label("Comment")), 5);
|
|
for (const auto &vertex : dba->Vertices(dba->Label("Comment"), false)) {
|
|
EXPECT_TRUE(vertex.has_label(dba->Label("Message")));
|
|
}
|
|
EXPECT_EQ(dba->VerticesCount(dba->Label("Forum")), 5);
|
|
}
|
|
|
|
TEST_F(RecoveryTest, TestPropertyNull) {
|
|
auto dba = db_.Access();
|
|
bool found = false;
|
|
for (const auto &vertex : dba->Vertices(dba->Label("Comment"), false)) {
|
|
auto id_prop = query::TypedValue(vertex.PropsAt(dba->Property("id")));
|
|
auto browser = query::TypedValue(vertex.PropsAt(dba->Property("browser")));
|
|
if (id_prop.IsString() && id_prop.Value<std::string>() == "2") {
|
|
EXPECT_FALSE(found);
|
|
found = true;
|
|
EXPECT_TRUE(browser.IsNull());
|
|
} else {
|
|
EXPECT_FALSE(browser.IsNull());
|
|
}
|
|
}
|
|
ASSERT_TRUE(found);
|
|
}
|
|
|
|
TEST_F(RecoveryTest, TestEdgesRecovered) {
|
|
auto dba = db_.Access();
|
|
EXPECT_EQ(dba->EdgesCount(), 5);
|
|
for (const auto &edge : dba->Edges(false)) {
|
|
EXPECT_TRUE(edge.EdgeType() == dba->EdgeType("POSTED_ON"));
|
|
}
|
|
}
|
|
|
|
int main(int argc, char *argv[]) {
|
|
::testing::InitGoogleTest(&argc, argv);
|
|
gflags::SetUsageMessage(usage);
|
|
gflags::ParseCommandLineFlags(&argc, &argv, true);
|
|
return RUN_ALL_TESTS();
|
|
}
|