Function TestHandleParseSQL

pkg/mcp/tools_test.go:199–236

--------------------------------------------------------------------------- handleParseSQL ---------------------------------------------------------------------------

(t *testing.T)

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

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

	t.Run("valid SQL returns statement info", func(t *testing.T) {
		req := makeReq(map[string]any{"sql": "SELECT id FROM users; SELECT name FROM orders"})
		res, err := handleParseSQL(ctx, req)
		if err != nil {
			t.Fatalf("unexpected error: %v", err)
		}
		data := unmarshalResult(t, res)
		if _, ok := data["statement_count"]; !ok {
			t.Error("expected 'statement_count' key in result")
		}
		if _, ok := data["statement_types"]; !ok {
			t.Error("expected 'statement_types' key in result")
		}
		count, ok := data["statement_count"].(float64)
		if !ok || count < 1 {
			t.Errorf("expected at least 1 statement, got %v", count)
		}
	})

	t.Run("empty sql returns protocol error", func(t *testing.T) {
		req := makeReq(map[string]any{"sql": ""})
		_, err := handleParseSQL(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 := handleParseSQL(ctx, req)
		if err == nil {
			t.Fatal("expected error for missing sql param, got nil")
		}
	})
}

// ---------------------------------------------------------------------------
// handleExtractMetadata