fix: do not detect TestMain as test function (#183)

lua/neotest-golang/features/testify/query.lua

@@ -16,13 +16,13 @@ M.namespace_query = [[
         ; name: (identifier)
         type: (pointer_type
           (type_identifier) @namespace.name )))) @namespace.definition
-    name: (field_identifier) @test_function (#match? @test_function "^(Test|Example)")
+    name: (field_identifier) @test_function (#match? @test_function "^(Test|Example)") (#not-match? @test.name "^TestMain$")
   ]]
 
 M.test_method_query = [[
    ; query for test method
    (method_declaration
-    name: (field_identifier) @test.name (#match? @test.name "^(Test|Example)")) @test.definition
+    name: (field_identifier) @test.name (#match? @test.name "^(Test|Example)") (#not-match? @test.name "^TestMain$")) @test.definition
   ]]
 
 --- Run a TreeSitter query on a file and return the matches.

lua/neotest-golang/query.lua

@@ -10,7 +10,7 @@ local M = {}
 M.test_function = [[
     ; query for test function
     ((function_declaration
-      name: (identifier) @test.name) (#match? @test.name "^(Test|Example)"))
+      name: (identifier) @test.name) (#match? @test.name "^(Test|Example)") (#not-match? @test.name "^TestMain$"))
       @test.definition
 
     ; query for subtest, like t.Run()

tests/go/positions_test.go

@@ -1,9 +1,16 @@
 package main
 
 import (
+	"os"
 	"testing"
 )
 
+// This function should not be detected by neotest-golang.
+func TestMain(m *testing.M) {
+	code := m.Run()
+	os.Exit(code)
+}
+
 // Vanilla top-level test.
 func TestTopLevel(t *testing.T) {
 	if Add(1, 2) != 3 {

tests/unit/golist_spec.lua

@@ -73,7 +73,7 @@ describe("go list output", function()
       StaleReason = "build ID mismatch",
       Target = "/Users/fredrik/go/bin/neotest-golang",
       TestGoFiles = { "positions_test.go", "testname_test.go" },
-      TestImports = { "testing" },
+      TestImports = { "os", "testing" },
     }
 
     -- ignored keys, as they might differ between OS/CI/platforms/too often