Function TestHandleLintSQL

pkg/mcp/tools_test.go:319–352

--------------------------------------------------------------------------- handleLintSQL ---------------------------------------------------------------------------

(t *testing.T)

// ---------------------------------------------------------------------------

func TestHandleLintSQL(t *testing.T) {
	ctx := context.Background()

	t.Run("valid SQL returns lint result", func(t *testing.T) {
		req := makeReq(map[string]any{"sql": "SELECT id FROM users"})
		res, err := handleLintSQL(ctx, req)
		if err != nil {
			t.Fatalf("unexpected error: %v", err)
		}
		data := unmarshalResult(t, res)
		if _, ok := data["violation_count"]; !ok {
			t.Error("expected 'violation_count' key in result")
		}
		if _, ok := data["violations"]; !ok {
			t.Error("expected 'violations' key in result")
		}
	})

	t.Run("empty sql returns protocol error", func(t *testing.T) {
		req := makeReq(map[string]any{"sql": ""})
		_, err := handleLintSQL(ctx, req)
		if err == nil {
			t.Fatal("expected error for empty sql, got nil")
		}
	})

	t.Run("missing sql param returns protocol error", func(t *testing.T) {
		req := makeReq(map[string]any{})
		_, err := handleLintSQL(ctx, req)
		if err == nil {
			t.Fatal("expected error for missing sql param, got nil")
		}
	})
}

// ---------------------------------------------------------------------------
// handleAnalyzeSQL