traxys/tree-sitter
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

feat(lib): add ts_query_end_byte_for_pattern

Browse source
This commit is contained in:
Amaan Qureshi 2024-07-04 20:51:27 -04:00
parent 3095fbe07b
commit 25c7189180
5 changed files with 52 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

26
cli/src/tests/query_test.rs
View file

@ -3640,30 +3640,27 @@ fn test_query_text_callback_returns_chunks() {
}
#[test]
fn test_query_start_byte_for_pattern() {
fn test_query_start_end_byte_for_pattern() {
let language = get_language("javascript");
let patterns_1 = r#"
let patterns_1 = indoc! {r#"
"+" @operator
"-" @operator
"*" @operator
"=" @operator
"=>" @operator
"#
.trim_start();
"#};
let patterns_2 = "
let patterns_2 = indoc! {"
(identifier) @a
(string) @b
"
.trim_start();
"};
let patterns_3 = "
let patterns_3 = indoc! {"
((identifier) @b (#match? @b i))
(function_declaration name: (identifier) @c)
(method_definition name: (property_identifier) @d)
"
.trim_start();
"};
let mut source = String::new();
source += patterns_1;
@ -3673,11 +3670,20 @@ fn test_query_start_byte_for_pattern() {
let query = Query::new(&language, &source).unwrap();
assert_eq!(query.start_byte_for_pattern(0), 0);
assert_eq!(query.end_byte_for_pattern(0), "\"+\" @operator\n".len());
assert_eq!(query.start_byte_for_pattern(5), patterns_1.len());
assert_eq!(
query.end_byte_for_pattern(5),
patterns_1.len() + "(identifier) @a\n".len()
);
assert_eq!(
query.start_byte_for_pattern(7),
patterns_1.len() + patterns_2.len()
);
assert_eq!(
query.end_byte_for_pattern(7),
patterns_1.len() + patterns_2.len() + "((identifier) @b (#match? @b i))\n".len()
);
}
#[test]