#include <catch2/catch_test_macros.hpp>

#include "matador/object/repository.hpp"

#include "../../models/department.hpp"
#include "../../models/recipe.hpp"

struct node {};

using namespace matador;

struct person {
  virtual ~person() = default;
  template < typename Operator >
  void process(Operator &/*op*/) {}
};

struct student final : person {};
struct teacher final : person {};

struct names {
  unsigned int id{};
  std::vector<std::string> names_list;

  template<typename Operator>
  void process(Operator &op) {
    namespace field = matador::access;
    field::primary_key(op, "id", id);
    field::has_many(op, "name_list", names_list, "names_id", utils::fetch_type::EAGER);
  }
};

TEST_CASE("Test empty prototype tree", "[schema_node][empty]") {
  const object::repository repo;

  REQUIRE( repo.empty() );
}

TEST_CASE("Test add type to prototype tree", "[schema_node][add]") {
  object::repository repo;

  REQUIRE( repo.empty() );

  auto res = repo.attach<person>("person");
  REQUIRE( res.is_ok() );
  res = repo.attach<student, person>("student");
  REQUIRE( res.is_ok() );
  res = repo.attach<teacher, person>("teacher");
  REQUIRE( res.is_ok() );

  REQUIRE( !repo.empty() );
  REQUIRE( repo.size() == 3 );
}

TEST_CASE("Test next and previous of schema node", "[schema_node][next][previous]") {
  object::repository repo;

  REQUIRE( repo.empty() );

  auto res = repo.attach<person>("person");
  REQUIRE( res.is_ok() );

  REQUIRE( repo.size() == 1 );

  auto it = repo.begin();
  REQUIRE( it->name() == "person" );
  REQUIRE( (--it)->name() == "person" );
  REQUIRE( ++it == repo.end() );
}

TEST_CASE("Test automatic creating of a relation table with foreign key", "[schema][relation_table][foreign_key]") {
  object::repository repo;

  REQUIRE( repo.empty() );

  auto res = repo.attach<test::department>("department");
  REQUIRE( res.is_ok() );
  REQUIRE(repo.size() == 2);
}

TEST_CASE("Test automatic creating of a relation table with values", "[schema][relation_table][values]") {
  object::repository repo;

  REQUIRE( repo.empty() );

  auto res = repo.attach<names>("names");
  REQUIRE( res.is_ok() );
}

TEST_CASE("Test one to many", "[relation][one-to-many]") {
  object::repository repo;

  REQUIRE( repo.empty() );

  auto res = repo.attach<test::department>("departments")
    .and_then( [&repo] { return repo.attach<test::employee>("employees"); } );
  REQUIRE( res.is_ok() );
  REQUIRE(repo.size() == 2);
  REQUIRE(repo.contains("departments"));
  REQUIRE(repo.contains("employees"));
}

TEST_CASE("Test many to many relation", "[relation][many-to-many]") {
  object::repository repo;

  REQUIRE(repo.empty());

  auto result = repo.attach<test::recipe>("recipes")
    .and_then( [&repo] { return repo.attach<test::ingredient>("ingredients"); } );
  REQUIRE(result);

  REQUIRE(repo.size() == 3);
  REQUIRE(repo.contains("ingredients"));
  REQUIRE(repo.contains("recipes"));
  REQUIRE(repo.contains("recipe_ingredients"));
}