tests

Signed-off-by: ChrisJBurns <[email protected]>

Author: ChrisJBurns

pkg/mcp/server/handler_mock_test.go

@@ -133,15 +133,12 @@ func TestHandler_SearchRegistry_WithMocks(t *testing.T) {
 				registryProvider: mockRegistry,
 			}
 
-			// Marshal arguments to JSON
-			argsJSON, _ := json.Marshal(map[string]interface{}{
-				"query": tt.query,
-			})
-
 			request := &mcp.CallToolRequest{
 				Params: &mcp.CallToolParamsRaw{
-					Name:      "search_registry",
-					Arguments: argsJSON,
+					Name: "search_registry",
+					Arguments: json.RawMessage(`{
+						"query": "` + tt.query + `"
+					}`),
 				},
 			}
 
@@ -339,15 +336,12 @@ func TestHandler_StopServer_WithMocks(t *testing.T) {
 				registryProvider: mockRegistry,
 			}
 
-			// Marshal arguments to JSON
-			argsJSON, _ := json.Marshal(map[string]interface{}{
-				"name": tt.serverName,
-			})
-
 			request := &mcp.CallToolRequest{
 				Params: &mcp.CallToolParamsRaw{
-					Name:      "stop_server",
-					Arguments: argsJSON,
+					Name: "stop_server",
+					Arguments: json.RawMessage(`{
+						"name": "` + tt.serverName + `"
+					}`),
 				},
 			}
 
@@ -426,15 +420,12 @@ func TestHandler_RemoveServer_WithMocks(t *testing.T) {
 				registryProvider: mockRegistry,
 			}
 
-			// Marshal arguments to JSON
-			argsJSON, _ := json.Marshal(map[string]interface{}{
-				"name": tt.serverName,
-			})
-
 			request := &mcp.CallToolRequest{
 				Params: &mcp.CallToolParamsRaw{
-					Name:      "remove_server",
-					Arguments: argsJSON,
+					Name: "remove_server",
+					Arguments: json.RawMessage(`{
+						"name": "` + tt.serverName + `"
+					}`),
 				},
 			}
 
@@ -516,15 +507,12 @@ func TestHandler_GetServerLogs_WithMocks(t *testing.T) {
 				registryProvider: mockRegistry,
 			}
 
-			// Marshal arguments to JSON
-			argsJSON, _ := json.Marshal(map[string]interface{}{
-				"name": tt.serverName,
-			})
-
 			request := &mcp.CallToolRequest{
 				Params: &mcp.CallToolParamsRaw{
-					Name:      "get_server_logs",
-					Arguments: argsJSON,
+					Name: "get_server_logs",
+					Arguments: json.RawMessage(`{
+						"name": "` + tt.serverName + `"
+					}`),
 				},
 			}