package handlers

import (
	"testing"

	"github.com/pingcap-inc/clinic/clinic_data_service/pkg/dto"
	"github.com/stretchr/testify/require"
)

func TestOverrideClusterIDWithPath(t *testing.T) {
	req := dto.DataProxyQueryRequest{ClusterId: "999"}

	err := overrideClusterIDWithPath(&req, "123")

	require.NoError(t, err)
	require.Equal(t, "123", req.ClusterId)
}

func TestOverrideClusterIDWithPathRejectsInvalidPathID(t *testing.T) {
	req := dto.DataProxyQueryRequest{ClusterId: "999"}

	err := overrideClusterIDWithPath(&req, "cluster-123")

	require.Error(t, err)
	require.Equal(t, "999", req.ClusterId)
}