#include "runtime/runtime_spec_helper.h"
#include "runtime/helpers/read_test_entries.h"

extern "C" const TSLanguage *ts_language_javascript();
extern "C" const TSLanguage *ts_language_json();
extern "C" const TSLanguage *ts_language_arithmetic();
extern "C" const TSLanguage *ts_language_golang();

START_TEST

describe("Languages", [&]() {
  TSDocument *doc;

  before_each([&]() {
    doc = ts_document_make();
  });

  after_each([&]() {
    ts_document_free(doc);
  });

  auto run_tests_for_language = [&](string language_name, const TSLanguage *language) {
    describe(language_name.c_str(), [&]() {
      before_each([&]() {
        ts_document_set_language(doc, language);
      });

      for (auto &entry : test_entries_for_language(language_name)) {
        it(entry.description.c_str(), [&]() {
          ts_document_set_input_string(doc, entry.input.c_str());
          auto doc_string = ts_node_string(ts_document_root_node(doc));
          AssertThat(doc_string, Equals(("(DOCUMENT " + entry.tree_string + ")").c_str()));
          free((void *)doc_string);
        });
      }
    });
  };

  run_tests_for_language("json", ts_language_json());
  run_tests_for_language("arithmetic", ts_language_arithmetic());
  run_tests_for_language("javascript", ts_language_javascript());
  run_tests_for_language("golang", ts_language_golang());
});

END_TEST