#include <catch2/catch_test_macros.hpp>

#include "matador/sql/column_name_generator.hpp"
#include "matador/sql/schema.hpp"

#include "models/order.hpp"
#include "models/product.hpp"

using namespace matador::sql;

TEST_CASE("Generate column names from object", "[column name generator]") {
  schema s("main");
  s.attach<matador::test::product>("product");

  auto columns = column_name_generator::generate<matador::test::product>(s);

  const std::vector<std::string> expected_columns = {
    "product_name",
    "supplier_id",
    "category_id",
    "quantity_per_unit",
    "unit_price",
    "units_in_stock",
    "units_in_order",
    "reorder_level",
    "discontinued"
  };
  REQUIRE(!columns.empty());
  REQUIRE(columns.size() == expected_columns.size());

  for (size_t i = 0; i != expected_columns.size(); ++i) {
    REQUIRE(expected_columns[i] == columns[i].name);
  }
}

TEST_CASE("Generate column names for object with has many relation", "[column][relation]") {
  schema s("main");
  s.attach<matador::test::product>("product");
  s.attach<matador::test::order_details>("order_details");
  s.attach<matador::test::order>("order");

  auto columns = column_name_generator::generate<matador::test::order>(s);

  const std::vector<std::string> expected_columns = {
  "order_id",
  "order_date",
  "required_date",
  "shipped_date",
  "ship_via",
  "freight",
  "ship_name",
  "ship_address",
  "ship_city",
  "ship_region",
  "ship_postal_code",
  "ship_country",
  "order_details_id",
  "order_id",
  "product_id",
  };
  REQUIRE(!columns.empty());
  REQUIRE(columns.size() == expected_columns.size());

  for (size_t i = 0; i != expected_columns.size(); ++i) {
    REQUIRE(expected_columns[i] == columns[i].name);
  }

}