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

fix(cli): gracefully handle OOB assertions

Browse source
This commit is contained in:
Amaan Qureshi 2024-12-15 23:45:53 -05:00
parent 6c4a50a9c7
commit 2374bca62a
1 changed files with 9 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

10
cli/src/query_testing.rs
View file

@ -190,13 +190,21 @@ pub fn parse_position_comments(
let mut i = 0;
let lines = source.lines_with_terminator().collect::<Vec<_>>();
for assertion in &mut result {
let original_position = assertion.position;
loop {
let on_assertion_line = assertion_ranges[i..]
.iter()
.any(|(start, _)| start.row == assertion.position.row);
let on_empty_line = lines[assertion.position.row].len() <= assertion.position.column;
if on_assertion_line || on_empty_line {
assertion.position.row -= 1;
if assertion.position.row > 0 {
assertion.position.row -= 1;
} else {
return Err(anyhow!(
"Error: could not find a line that corresponds to the assertion `{}` located at {original_position}",
assertion.expected_capture_name
));
}
} else {
while i < assertion_ranges.len()
&& assertion_ranges[i].0.row < assertion.position.row