Function makeQueryResponse

backend/api/mcp/tool_query_test.go:143–165

makeQueryResponse builds a mock SQL Query API response.

(columns, columnTypes []string, rows [][]any, latency string)

// makeQueryResponse builds a mock SQL Query API response.
func makeQueryResponse(columns, columnTypes []string, rows [][]any, latency string) map[string]any {
	apiRows := make([]map[string]any, 0, len(rows))
	for _, row := range rows {
		values := make([]map[string]any, 0, len(row))
		for _, v := range row {
			values = append(values, map[string]any{"stringValue": v})
		}
		apiRows = append(apiRows, map[string]any{"values": values})
	}
	return map[string]any{
		"results": []map[string]any{
			{
				"columnNames":     columns,
				"columnTypeNames": columnTypes,
				"rows":            apiRows,
				"rowsCount":       fmt.Sprintf("%d", len(rows)),
				"latency":         latency,
				"error":           "",
				"statement":       "SELECT ...",
			},
		},
	}
}

// mockQueryServer returns an HTTP handler that routes to ListDatabases or Query
// based on the request URL path.