#include <catch2/catch_test_macros.hpp>
#include <fmt/core.h>
#include <nlohmann/json.hpp>
#include <fstream>
#include "pathing.hpp"
#include "matrix.hpp"
#include "ai.hpp"

using namespace fmt;
using namespace nlohmann;
using std::string;

json load_test_pathing(const string &fname) {
  std::ifstream infile(fname);
  return json::parse(infile);
}

TEST_CASE("dijkstra algo test", "[pathing]") {
  json data = load_test_pathing("./tests/dijkstra.json");

  for(auto &test : data) {
    Matrix expected = test["expected"];
    Matrix walls = test["walls"];

    Pathing pathing(walls[0].size(), walls.size());

    pathing.$input = test["input"];

    REQUIRE(pathing.INVARIANT());
    pathing.compute_paths(walls);

    REQUIRE(pathing.INVARIANT());

    matrix::dump("PATHING RESULT", pathing.$paths);
    matrix::dump("PATHING EXPECTED", expected);
    REQUIRE(pathing.$paths == expected);
  }
}

TEST_CASE("random flood", "[pathing]") {
  json data = load_test_pathing("./tests/dijkstra.json");
  auto test = data[0];

  Matrix expected = test["expected"];
  Matrix walls = test["walls"];

  Pathing pathing(walls[0].size(), walls.size());
  pathing.$input = test["input"];

  REQUIRE(pathing.INVARIANT());
  pathing.compute_paths(walls);
}