Exclude test helpers from codecov

Author: mtodor

Makefile

@@ -56,7 +56,7 @@ test: ## Run unit tests
 .PHONY: test-coverage-and-junit
 test-coverage-and-junit: ## Run unit tests with coverage and junit output
 	go install github.com/jstemmer/go-junit-report/v2@v2.1.0
-	$(GOTEST) -v -cover -race -coverprofile=$(COVERAGE_OUT) ./... -json 2>&1 | go-junit-report -parser gojson > $(JUNIT_OUT)
+	$(GOTEST) -v -cover -race -coverprofile=$(COVERAGE_OUT) ./... -json 2>&1 | go-junit-report -set-exit-code -parser gojson > $(JUNIT_OUT)
 
 .PHONY: coverage-html
 coverage-html: test ## Generate and open HTML coverage report

cmd/stackrox-mcp/main_test.go

@@ -50,7 +50,7 @@ func TestGracefulShutdown(t *testing.T) {
 	}()
 
 	serverURL := "http://" + net.JoinHostPort(cfg.Server.Address, strconv.Itoa(cfg.Server.Port))
-	err = testutil.WaitForServerReady(serverURL, 3*time.Second)
+	err = testutil.WaitForServerReady(t, serverURL, 3*time.Second)
 	require.NoError(t, err, "Server should start within timeout")
 
 	// Establish actual HTTP connection to verify server is responding.

internal/server/server_test.go

@@ -137,7 +137,7 @@ func TestServer_Start(t *testing.T) {
 	}()
 
 	serverURL := "http://" + net.JoinHostPort(cfg.Server.Address, strconv.Itoa(cfg.Server.Port))
-	err := testutil.WaitForServerReady(serverURL, 3*time.Second)
+	err := testutil.WaitForServerReady(t, serverURL, 3*time.Second)
 	require.NoError(t, err, "Server should start within timeout")
 
 	// Verify tools were registered.
@@ -184,7 +184,7 @@ func TestServer_HealthEndpoint(t *testing.T) {
 	}()
 
 	serverURL := "http://" + net.JoinHostPort(cfg.Server.Address, strconv.Itoa(cfg.Server.Port))
-	err := testutil.WaitForServerReady(serverURL, 3*time.Second)
+	err := testutil.WaitForServerReady(t, serverURL, 3*time.Second)
 	require.NoError(t, err, "Server should start within timeout")
 
 	// Test health endpoint.

internal/testutil/config_test.go

@@ -0,0 +1,23 @@
+package testutil
+
+import (
+	"os"
+	"path/filepath"
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+	"github.com/stretchr/testify/require"
+)
+
+func TestWriteYAMLFile(t *testing.T) {
+	content := "test: value\n"
+
+	path := WriteYAMLFile(t, content)
+
+	filename := filepath.Base(path)
+	assert.Contains(t, filename, t.Name(), "Filename should contain test name")
+
+	info, err := os.Stat(path)
+	require.NoError(t, err, "File should exist")
+	assert.Equal(t, int64(len(content)), info.Size(), "File should not be empty")
+}

internal/testutil/server.go

@@ -3,6 +3,7 @@ package testutil
 import (
 	"fmt"
 	"net/http"
+	"testing"
 	"time"
 )
 
@@ -11,7 +12,9 @@ const timeoutDuration = 100 * time.Millisecond
 // WaitForServerReady polls the server until it's ready to accept connections.
 // This function is useful for integration tests where you need to wait for
 // a server to start before making requests to it.
-func WaitForServerReady(address string, timeout time.Duration) error {
+func WaitForServerReady(t *testing.T, address string, timeout time.Duration) error {
+	t.Helper()
+
 	deadline := time.Now().Add(timeout)
 	client := &http.Client{Timeout: timeoutDuration}
 

internal/toolsets/mock/toolset_test.go

@@ -0,0 +1,34 @@
+package mock
+
+import (
+	"testing"
+
+	"github.com/stackrox/stackrox-mcp/internal/toolsets"
+	"github.com/stretchr/testify/assert"
+	"github.com/stretchr/testify/require"
+)
+
+func TestToolset_GetName(t *testing.T) {
+	toolset := NewToolset("my-toolset", true, nil)
+
+	require.NotNil(t, toolset)
+	assert.Equal(t, "my-toolset", toolset.GetName())
+}
+
+func TestToolset_IsEnabled_True(t *testing.T) {
+	tools := []toolsets.Tool{&Tool{NameValue: "test-tool"}}
+	toolset := NewToolset("test", true, tools)
+
+	require.NotNil(t, toolset)
+	assert.True(t, toolset.IsEnabled())
+	assert.Equal(t, tools, toolset.GetTools(), "Should return tools when enabled")
+}
+
+func TestToolset_IsEnabled_False(t *testing.T) {
+	tools := []toolsets.Tool{&Tool{NameValue: "test-tool"}}
+	toolset := NewToolset("test", false, tools)
+
+	require.NotNil(t, toolset)
+	assert.False(t, toolset.IsEnabled())
+	assert.Empty(t, toolset.GetTools(), "Should return empty slice when disabled")
+}