tree-sitter/spec/runtime/node_position_spec.cc
2014-07-14 21:11:15 -07:00

46 lines
1.4 KiB
C++

#include "runtime/runtime_spec_helper.h"
#include "runtime/tree.h"
extern "C" TSParser * ts_parser_json();
START_TEST
describe("tracking the positions of AST nodes", []() {
TSDocument *doc;
before_each([&]() {
doc = ts_document_make();
ts_document_set_parser(doc, ts_parser_json());
});
after_each([&]() {
ts_document_free(doc);
});
it("records the widths and offsets of nodes", [&]() {
ts_document_set_input_string(doc, " [12, 5]");
const TSTree *tree = ts_document_tree(doc);
const TSTree *array = ts_tree_children(tree, NULL)[0];
const TSTree *number1 = ts_tree_children(array, NULL)[0];
const TSTree *number2 = ts_tree_children(array, NULL)[1];
AssertThat(ts_document_symbol_name(doc, array), Equals("array"));
AssertThat(ts_document_symbol_name(doc, number1), Equals("number"));
AssertThat(ts_document_symbol_name(doc, number2), Equals("number"));
AssertThat(number1->offset, Equals<size_t>(0));
AssertThat(number1->size, Equals<size_t>(2));
AssertThat(number2->offset, Equals<size_t>(1));
AssertThat(number2->size, Equals<size_t>(1));
AssertThat(array->offset, Equals<size_t>(2));
AssertThat(array->size, Equals<size_t>(7));
AssertThat(tree->offset, Equals<size_t>(2));
AssertThat(tree->size, Equals<size_t>(7));
});
});
END_TEST