Function TestHandleAnalyzeSQL_InvalidSQLPartialResults

pkg/mcp/tools_test.go:695–742

(t *testing.T)

}

func TestHandleAnalyzeSQL_InvalidSQLPartialResults(t *testing.T) {
	ctx := context.Background()
	req := makeReq(map[string]any{"sql": "SELECT FROM"})
	res, err := handleAnalyzeSQL(ctx, req)
	if err != nil {
		t.Fatalf("unexpected protocol error: %v", err)
	}
	data := unmarshalResult(t, res)

	// Validate should report valid=false (it returns a result, not an error).
	validateData, ok := data["validate"].(map[string]any)
	if !ok {
		t.Fatal("expected 'validate' sub-result to be a map")
	}
	valid, ok := validateData["valid"].(bool)
	if !ok || valid {
		t.Errorf("expected validate.valid=false for invalid SQL, got %v", validateData["valid"])
	}

	// Security should still succeed.
	securityData, ok := data["security"].(map[string]any)
	if !ok {
		t.Fatal("expected 'security' sub-result to be a map")
	}
	if _, ok := securityData["is_clean"]; !ok {
		t.Error("expected 'is_clean' key in security sub-result")
	}

	// Lint should still succeed.
	lintData, ok := data["lint"].(map[string]any)
	if !ok {
		t.Fatal("expected 'lint' sub-result to be a map")
	}
	if _, ok := lintData["violation_count"]; !ok {
		t.Error("expected 'violation_count' key in lint sub-result")
	}

	// Parse, metadata, and format should have errors since "SELECT FROM" fails to parse.
	errs, ok := data["errors"].(map[string]any)
	if !ok {
		t.Fatal("expected 'errors' map for failed sub-results")
	}
	for _, key := range []string{"parse", "metadata", "format"} {
		if _, ok := errs[key]; !ok {
			t.Errorf("expected error entry for %q in errors map", key)
		}
	}
}