#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::Recover(durability_dir, db_);
  }

  GraphDb db_;
};

TEST_F(RecoveryTest, TestVerticesRecovered) {
  GraphDbAccessor dba(db_);
  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) {
  GraphDbAccessor dba(db_);
  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) {
  GraphDbAccessor dba(db_);
  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();
}