Cannot get Treesitter to detect Zig tests

[lawrence-laz]

Hi, I am trying to write an adapter for Zig and currently have a very basic setup but can't seem to get treesitter parser to detect tests in a file.

const std = @import("std");
const testing = std.testing;

pub fn main() !void {
    std.debug.print("All your {s} are belong to us.\n", .{"codebase"});

    const stdout_file = std.io.getStdOut().writer();
    var bw = std.io.bufferedWriter(stdout_file);
    const stdout = bw.writer();

    try stdout.print("Run `zig build test` to run the tests.\n", .{});

    try bw.flush(); // don't forget to flush!
}

test "my test" {
    const foo = 2 + 2;
    try testing.expectEqual(4, foo);
}

test "my other test" {
    const foo = 2 + 2;
    try testing.expectEqual(4, foo);
}

Parsing script:

local async = require("neotest.async")
local lib = require("neotest.lib")
local ts = lib.treesitter

async.run(function()
	local parse_queries = [[
      ;;query
      (TestDecl)
    ]]
	local tree = ts.parse_positions("main.zig", parse_queries)
	vim.print(tree:to_list())
end)

Output:

{ {
    id = "main.zig",
    name = "main.zig",
    path = "main.zig",
    range = { 0, 0, 24, 0 },
    type = "file"
  } }

Looking at the range it seems like the whole file is selected and only one entry is returned, while there are two test declarations.

Am I doing something wrong?

[lawrence-laz]

Turns out I was missing capture groups:

local async = require("neotest.async")
local lib = require("neotest.lib")
local ts = lib.treesitter

async.run(function()
	local parse_queries = [[
      ;;query
      (TestDecl
      	(STRINGLITERALSINGLE) @test.name
      ) @test.definition
    ]]
	local tree = ts.parse_positions("main.zig", parse_queries)
	vim.print(tree:to_list())
end)

now returns:

{ {
    id = "main.zig",
    name = "main.zig",
    path = "main.zig",
    range = { 0, 0, 24, 0 },
    type = "file"
  }, { {
      id = 'main.zig::"my test"',
      name = '"my test"',
      path = "main.zig",
      range = { 15, 0, 18, 1 },
      type = "test"
    } }, { {
      id = 'main.zig::"my other test"',
      name = '"my other test"',
      path = "main.zig",
      range = { 20, 0, 23, 1 },
      type = "test"
    } } }