#include "../kit/xml.h"

#define KIT_TEST_FILE xml
#include "../kit/kit_test.h"

TEST("xml parse tag") {
  is_handle_t        is  = IS_WRAP_STRING(SZ("<foo> </foo>"));
  xml_parse_result_t res = xml_parse(is, NULL);

  REQUIRE_EQ(res.status, KIT_OK);

  if (res.status == KIT_OK) {
    REQUIRE_EQ(res.xml.is_declaration, 0);
    REQUIRE(AR_EQUAL(res.xml.tag, SZ("foo")));
    xml_destroy(&res.xml);
  }

  is_destroy(is);
}

TEST("xml parse tag with dash") {
  is_handle_t        is  = IS_WRAP_STRING(SZ("<foo-bar> </foo-bar>"));
  xml_parse_result_t res = xml_parse(is, NULL);

  REQUIRE_EQ(res.status, KIT_OK);

  if (res.status == KIT_OK) {
    REQUIRE_EQ(res.xml.is_declaration, 0);
    REQUIRE(AR_EQUAL(res.xml.tag, SZ("foo-bar")));
    xml_destroy(&res.xml);
  }

  is_destroy(is);
}

TEST("xml parse tag not closed") {
  is_handle_t        is  = IS_WRAP_STRING(SZ("<foo>"));
  xml_parse_result_t res = xml_parse(is, NULL);

  REQUIRE_EQ(res.status, KIT_ERROR_INTERNAL);

  if (res.status == KIT_OK)
    xml_destroy(&res.xml);

  is_destroy(is);
}

TEST("xml parse tag text") {
  is_handle_t        is  = IS_WRAP_STRING(SZ("<foo> bar </foo>"));
  xml_parse_result_t res = xml_parse(is, NULL);

  REQUIRE_EQ(res.status, KIT_OK);

  if (res.status == KIT_OK) {
    REQUIRE_EQ(res.xml.is_declaration, 0);
    REQUIRE(AR_EQUAL(res.xml.tag, SZ("foo")));
    REQUIRE(AR_EQUAL(res.xml.text, SZ(" bar ")));
    xml_destroy(&res.xml);
  }

  is_destroy(is);
}

TEST("xml parse empty tag") {
  is_handle_t        is  = IS_WRAP_STRING(SZ("<foo />"));
  xml_parse_result_t res = xml_parse(is, NULL);

  REQUIRE_EQ(res.status, KIT_OK);

  if (res.status == KIT_OK) {
    REQUIRE_EQ(res.xml.is_declaration, 0);
    REQUIRE(AR_EQUAL(res.xml.tag, SZ("foo")));
    xml_destroy(&res.xml);
  }

  is_destroy(is);
}

TEST("xml parse tail") {
  is_handle_t        is  = IS_WRAP_STRING(SZ("<foo></foo> bar"));
  xml_parse_result_t res = xml_parse(is, NULL);

  REQUIRE_EQ(res.status, KIT_OK);

  if (res.status == KIT_OK) {
    REQUIRE(AR_EQUAL(res.xml.tag, SZ("foo")));
    REQUIRE(AR_EQUAL(res.xml.tail, SZ(" bar")));
    xml_destroy(&res.xml);
  }

  is_destroy(is);
}

TEST("xml parse empty tail") {
  is_handle_t        is  = IS_WRAP_STRING(SZ("<foo /> bar"));
  xml_parse_result_t res = xml_parse(is, NULL);

  REQUIRE_EQ(res.status, KIT_OK);

  if (res.status == KIT_OK) {
    REQUIRE(AR_EQUAL(res.xml.tag, SZ("foo")));
    REQUIRE(AR_EQUAL(res.xml.tail, SZ(" bar")));
    xml_destroy(&res.xml);
  }

  is_destroy(is);
}

TEST("xml parse property") {
  is_handle_t is = IS_WRAP_STRING(SZ("<foo bar=\"42\"></foo>"));
  xml_parse_result_t res = xml_parse(is, NULL);

  REQUIRE_EQ(res.status, KIT_OK);

  if (res.status == KIT_OK) {
    REQUIRE(AR_EQUAL(res.xml.tag, SZ("foo")));
    REQUIRE_EQ(res.xml.properties.size, 1);
    if (res.xml.properties.size == 1) {
      REQUIRE(AR_EQUAL(res.xml.properties.values[0].name, SZ("bar")));
      REQUIRE(AR_EQUAL(res.xml.properties.values[0].value, SZ("42")));
    }
    xml_destroy(&res.xml);
  }

  is_destroy(is);
}

TEST("xml parse empty property") {
  is_handle_t        is  = IS_WRAP_STRING(SZ("<foo bar=\"42\" />"));
  xml_parse_result_t res = xml_parse(is, NULL);

  REQUIRE_EQ(res.status, KIT_OK);

  if (res.status == KIT_OK) {
    REQUIRE(AR_EQUAL(res.xml.tag, SZ("foo")));
    REQUIRE_EQ(res.xml.properties.size, 1);
    if (res.xml.properties.size == 1) {
      REQUIRE(AR_EQUAL(res.xml.properties.values[0].name, SZ("bar")));
      REQUIRE(AR_EQUAL(res.xml.properties.values[0].value, SZ("42")));
    }
    xml_destroy(&res.xml);
  }

  is_destroy(is);
}

TEST("xml parse child") {
  is_handle_t is = IS_WRAP_STRING(SZ("<foo><bar></bar></foo>"));
  xml_parse_result_t res = xml_parse(is, NULL);

  REQUIRE_EQ(res.status, KIT_OK);

  if (res.status == KIT_OK) {
    REQUIRE(AR_EQUAL(res.xml.tag, SZ("foo")));
    REQUIRE_EQ(res.xml.children.size, 1);
    if (res.xml.children.size == 1)
      REQUIRE(AR_EQUAL(res.xml.children.values[0].tag, SZ("bar")));
    xml_destroy(&res.xml);
  }

  is_destroy(is);
}

TEST("xml parse declaration") {
  is_handle_t        is  = IS_WRAP_STRING(SZ("<?foo ?>"));
  xml_parse_result_t res = xml_parse(is, NULL);

  REQUIRE_EQ(res.status, KIT_OK);

  if (res.status == KIT_OK) {
    REQUIRE_EQ(res.xml.is_declaration, 1);
    REQUIRE(AR_EQUAL(res.xml.tag, SZ("foo")));
    xml_destroy(&res.xml);
  }

  is_destroy(is);
}

TEST("xml parse comment") {
  is_handle_t        is  = IS_WRAP_STRING(SZ("<!-- foo --><bar />"));
  xml_parse_result_t res = xml_parse(is, NULL);

  REQUIRE_EQ(res.status, KIT_OK);

  if (res.status == KIT_OK) {
    REQUIRE_EQ(res.xml.is_declaration, 0);
    REQUIRE(AR_EQUAL(res.xml.tag, SZ("bar")));
    xml_destroy(&res.xml);
  }

  is_destroy(is);
}

TEST("xml parse comment before text") {
  is_handle_t is = IS_WRAP_STRING(SZ("<!-- foo --> bar <tag />"));
  xml_parse_result_t res = xml_parse(is, NULL);

  REQUIRE_EQ(res.status, KIT_OK);

  if (res.status == KIT_OK) {
    REQUIRE_EQ(res.xml.is_declaration, 0);
    REQUIRE(AR_EQUAL(res.xml.text, SZ(" bar ")));
    xml_destroy(&res.xml);
  }

  is_destroy(is);
}

TEST("xml parse comment after text") {
  is_handle_t is = IS_WRAP_STRING(SZ("foo <!-- bar --><tag />"));
  xml_parse_result_t res = xml_parse(is, NULL);

  REQUIRE_EQ(res.status, KIT_OK);

  if (res.status == KIT_OK) {
    REQUIRE_EQ(res.xml.is_declaration, 0);
    REQUIRE(AR_EQUAL(res.xml.text, SZ("foo ")));
    xml_destroy(&res.xml);
  }

  is_destroy(is);
}

TEST("xml parse comment between text") {
  is_handle_t is = IS_WRAP_STRING(
      SZ("foo<!-- comment --> bar<tag />"));
  xml_parse_result_t res = xml_parse(is, NULL);

  REQUIRE_EQ(res.status, KIT_OK);

  if (res.status == KIT_OK) {
    REQUIRE_EQ(res.xml.is_declaration, 0);
    REQUIRE(AR_EQUAL(res.xml.text, SZ("foo bar")));
    xml_destroy(&res.xml);
  }

  is_destroy(is);
}

TEST("xml parse comment tail") {
  is_handle_t        is  = IS_WRAP_STRING(SZ("<foo /><!-- tail -->"));
  xml_parse_result_t res = xml_parse(is, NULL);

  REQUIRE_EQ(res.status, KIT_OK);

  if (res.status == KIT_OK) {
    REQUIRE_EQ(res.xml.is_declaration, 0);
    REQUIRE(AR_EQUAL(res.xml.tag, SZ("foo")));
    xml_destroy(&res.xml);
  }

  is_destroy(is);
}

TEST("xml parse comment tail before text") {
  is_handle_t is = IS_WRAP_STRING(SZ("<foo /><!-- tail --> bar"));
  xml_parse_result_t res = xml_parse(is, NULL);

  REQUIRE_EQ(res.status, KIT_OK);

  if (res.status == KIT_OK) {
    REQUIRE_EQ(res.xml.is_declaration, 0);
    REQUIRE(AR_EQUAL(res.xml.tag, SZ("foo")));
    REQUIRE(AR_EQUAL(res.xml.tail, SZ(" bar")));
    xml_destroy(&res.xml);
  }

  is_destroy(is);
}

TEST("xml parse comment tail after text") {
  is_handle_t is = IS_WRAP_STRING(SZ("<foo /> bar <!-- tail -->"));
  xml_parse_result_t res = xml_parse(is, NULL);

  REQUIRE_EQ(res.status, KIT_OK);

  if (res.status == KIT_OK) {
    REQUIRE_EQ(res.xml.is_declaration, 0);
    REQUIRE(AR_EQUAL(res.xml.tag, SZ("foo")));
    REQUIRE(AR_EQUAL(res.xml.tail, SZ(" bar ")));
    xml_destroy(&res.xml);
  }

  is_destroy(is);
}

TEST("xml parse comment tail between text") {
  is_handle_t is = IS_WRAP_STRING(SZ("<tag />foo<!-- tail --> bar"));
  xml_parse_result_t res = xml_parse(is, NULL);

  REQUIRE_EQ(res.status, KIT_OK);

  if (res.status == KIT_OK) {
    REQUIRE_EQ(res.xml.is_declaration, 0);
    REQUIRE(AR_EQUAL(res.xml.tag, SZ("tag")));
    REQUIRE(AR_EQUAL(res.xml.tail, SZ("foo bar")));
    xml_destroy(&res.xml);
  }

  is_destroy(is);
}

#undef KIT_TEST_FILE