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

extern "C" TSLanguage *ts_language_javascript;
extern "C" TSLanguage *ts_language_json;
extern "C" TSLanguage *ts_language_arithmetic;
extern "C" 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, 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_document_string(doc);
                    AssertThat(doc_string, Equals(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