test: improve testify assertion specificity in targeted tests

Agent-Logs-Url: https://github.com/github/gh-aw-mcpg/sessions/b3e143bf-9353-45dd-8d9b-80cebf30c68e

Co-authored-by: lpcox <[email protected]>

Author: Copilot

internal/config/config_test.go

@@ -1089,7 +1089,7 @@ port 3000
 	errMsg := err.Error()
 	assert.Contains(t, errMsg, "line", "Error should mention line number")
 	// Our improved format includes "column" explicitly when ParseError is detected
-	assert.True(t, strings.Contains(errMsg, "column") || strings.Contains(errMsg, "line 2"),
+	assert.Regexp(t, `column|line 2`, errMsg,
 		"Error should mention column or line position, got: %s", errMsg)
 }
 

internal/config/expand_raw_json_test.go

@@ -1,7 +1,6 @@
 package config
 
 import (
-	"strings"
 	"testing"
 
 	"github.com/stretchr/testify/assert"
@@ -566,7 +565,7 @@ func TestValidateMounts(t *testing.T) {
 			if tt.shouldErr {
 				require.Error(t, err, "Expected an error but got none")
 				if tt.errMsg != "" {
-					assert.True(t, strings.Contains(err.Error(), tt.errMsg),
+					assert.Contains(t, err.Error(), tt.errMsg,
 						"Error message %q should contain %q", err.Error(), tt.errMsg)
 				}
 			} else {

internal/config/fetch_and_fix_schema_test.go

@@ -245,7 +245,7 @@ func TestFixSchemaBytes_AddRegistryAndGuardPoliciesToStdioConfig(t *testing.T) {
 	require.True(t, hasGuardPolicies, "guard-policies field should be added to stdioServerConfig.properties")
 	gpMap := guardPolicies.(map[string]interface{})
 	assert.Equal(t, "object", gpMap["type"], "guard-policies.type should be 'object'")
-	assert.Equal(t, true, gpMap["additionalProperties"], "guard-policies.additionalProperties should be true")
+	assert.True(t, gpMap["additionalProperties"].(bool), "guard-policies.additionalProperties should be true")
 }
 
 // TestFixSchemaBytes_AddRegistryAndGuardPoliciesToHttpConfig covers the injection

internal/difc/labels_test.go

@@ -1,7 +1,6 @@
 package difc
 
 import (
-	"strings"
 	"testing"
 
 	"github.com/stretchr/testify/assert"
@@ -590,11 +589,11 @@ func TestViolationError_Error(t *testing.T) {
 			msg := tt.err.Error()
 			assert.NotEmpty(t, msg)
 			for _, want := range tt.wantContains {
-				assert.True(t, strings.Contains(msg, want),
+				assert.Contains(t, msg, want,
 					"expected %q in error message %q", want, msg)
 			}
 			for _, absent := range tt.wantAbsent {
-				assert.False(t, strings.Contains(msg, absent),
+				assert.NotContains(t, msg, absent,
 					"expected %q NOT to be in error message %q", absent, msg)
 			}
 		})
@@ -616,7 +615,7 @@ func TestViolationError_Detailed(t *testing.T) {
 		base := err.Error()
 
 		// Detailed should contain the base error message
-		assert.True(t, strings.Contains(detailed, base), "detailed should include base error")
+		assert.Contains(t, detailed, base, "detailed should include base error")
 
 		// Detailed should include agent and resource tag context
 		assert.Contains(t, detailed, "Agent")

internal/mcp/tool_result_test.go

@@ -268,7 +268,7 @@ func TestParseToolArguments(t *testing.T) {
 		require.NotNil(t, args)
 		assert.Equal(t, "search term", args["query"])
 		assert.Equal(t, float64(10), args["limit"])
-		assert.Equal(t, true, args["active"])
+		assert.True(t, args["active"].(bool))
 	})
 
 	t.Run("nested object arguments are parsed correctly", func(t *testing.T) {

internal/mcp/unmarshal_params_test.go

@@ -153,7 +153,7 @@ func TestUnmarshalParams(t *testing.T) {
 		// JSON unmarshal converts numbers to float64 by default
 		assert.Equal(t, float64(42), target["int_val"])
 		assert.Equal(t, 3.14, target["float_val"])
-		assert.Equal(t, true, target["bool_val"])
+		assert.True(t, target["bool_val"].(bool))
 		assert.Len(t, target["array_val"], 3)
 		assert.NotNil(t, target["object_val"])
 		assert.Nil(t, target["null_val"])

internal/middleware/jqschema_integration_test.go

@@ -195,7 +195,7 @@ func TestMiddlewareWithLargePayload(t *testing.T) {
 	// Verify preview truncation in Content field (preview ends with ... when truncated)
 	previewInContent := contentMap["payloadPreview"].(string)
 	if strings.HasSuffix(previewInContent, "...") {
-		assert.True(t, len(previewInContent) <= 503, "Preview in Content should be truncated")
+		assert.LessOrEqual(t, len(previewInContent), 503, "Preview in Content should be truncated")
 	}
 
 	// Also check data return value
@@ -204,7 +204,7 @@ func TestMiddlewareWithLargePayload(t *testing.T) {
 	// Verify preview truncation (check if it ends with ...)
 	preview := dataMap["payloadPreview"].(string)
 	if strings.HasSuffix(preview, "...") {
-		assert.True(t, len(preview) <= 503, "Preview should be truncated")
+		assert.LessOrEqual(t, len(preview), 503, "Preview should be truncated")
 	}
 
 	// Verify payload file has complete data

internal/proxy/proxy_test.go

@@ -682,7 +682,7 @@ func TestRewrapSearchResponse(t *testing.T) {
 		result := rewrapSearchResponse(original, filtered)
 		m := result.(map[string]interface{})
 		assert.Equal(t, float64(2), m["total_count"])
-		assert.Equal(t, false, m["incomplete_results"])
+		assert.False(t, m["incomplete_results"].(bool))
 		assert.Len(t, m["items"].([]interface{}), 2)
 	})
 

internal/proxy/rest_backend_caller_tool_test.go

@@ -6,7 +6,6 @@ import (
 	"fmt"
 	"net/http"
 	"net/http/httptest"
-	"strings"
 	"testing"
 
 	"github.com/stretchr/testify/assert"
@@ -574,5 +573,5 @@ func TestRestBackendCaller_IssueRead_ResponseFormat(t *testing.T) {
 
 	text, ok := content[0]["text"].(string)
 	require.True(t, ok)
-	assert.True(t, strings.Contains(text, "Bug report"), "response text should contain the issue title")
+	assert.Contains(t, text, "Bug report", "response text should contain the issue title")
 }

internal/server/call_backend_tool_test.go

@@ -5,7 +5,6 @@ import (
 	"encoding/json"
 	"net/http"
 	"net/http/httptest"
-	"strings"
 	"testing"
 	"time"
 
@@ -482,8 +481,8 @@ func TestCallBackendTool_AllowedToolsError_MessageFormat(t *testing.T) {
 	require.True(result.IsError)
 	require.Len(result.Content, 1)
 	text := result.Content[0].(*sdk.TextContent).Text
-	assert.True(t, strings.Contains(text, `"blocked"`), "error message should include tool name: %s", text)
-	assert.True(t, strings.Contains(text, "allowed-tools"), "error message should mention allowed-tools: %s", text)
+	assert.Contains(t, text, `"blocked"`, "error message should include tool name: %s", text)
+	assert.Contains(t, text, "allowed-tools", "error message should mention allowed-tools: %s", text)
 }
 
 // TestCallBackendTool_ToolTimeoutEnforcedViaContext verifies that the configured