diff options
| -rw-r--r-- | tests/query/injection_spec.lua | 7 | ||||
| -rw-r--r-- | tests/query/injections/cuda/macro-self-injection.cu | 2 |
2 files changed, 8 insertions, 1 deletions
diff --git a/tests/query/injection_spec.lua b/tests/query/injection_spec.lua index abbd7a8a..7a98dd27 100644 --- a/tests/query/injection_spec.lua +++ b/tests/query/injection_spec.lua @@ -22,6 +22,7 @@ local function check_assertions(file) local parser = parsers.get_parser(buf, lang) local self = highlighter.new(parser, {}) + local top_level_root = parser:parse()[1]:root() for _, assertion in ipairs(assertions) do local row = assertion.position.row @@ -34,7 +35,11 @@ local function check_assertions(file) end local root = tstree:root() - if ts_utils.is_in_node_range(root, row, col) and assertion.expected_capture_name == tree:lang() then + if + ts_utils.is_in_node_range(root, row, col) + and assertion.expected_capture_name == tree:lang() + and root ~= top_level_root + then found = true end end, true) diff --git a/tests/query/injections/cuda/macro-self-injection.cu b/tests/query/injections/cuda/macro-self-injection.cu new file mode 100644 index 00000000..1acef197 --- /dev/null +++ b/tests/query/injections/cuda/macro-self-injection.cu @@ -0,0 +1,2 @@ +#define FOO(X,Y) X + Y +// ^ cuda |