package vm

import (
	"testing"

	"github.com/stretchr/testify/require"
)

func TestBuildURL(t *testing.T) {
	addrP := &addrParams{
		Provider:  "aws",
		Region:    "us-west-1",
		ClusterID: 123,
	}

	// dev
	vq, err := NewVmQuery(nil, "dev")
	require.NoError(t, err)

	url, err := vq.buildVMURL(addrP, pathQuery)
	require.NoError(t, err)
	// require.Equal(t, "http://www.ds.us-west-1.aws.observability-dev.pingcap.cloud/internal/topsql-vmselect/select/123/prometheus/api/v1/query", url)
	require.Equal(t, "http://www.ds.us-west-1.aws.observability.tidbcloud.com/internal/topsql-vmselect/select/123/prometheus/api/v1/query", url)

	// staging
	vq, err = NewVmQuery(nil, "staging")
	require.NoError(t, err)

	url, err = vq.buildVMURL(addrP, pathQuery)
	require.NoError(t, err)
	// require.Equal(t, "http://www.ds.us-west-1.aws.observability-staging.pingcap.cloud/internal/topsql-vmselect/select/123/prometheus/api/v1/query", url)
	require.Equal(t, "http://www.ds.us-west-1.aws.observability.tidbcloud.com/internal/topsql-vmselect/select/123/prometheus/api/v1/query", url)

	// prod
	vq, err = NewVmQuery(nil, "prod")
	require.NoError(t, err)

	url, err = vq.buildVMURL(addrP, pathQuery)
	require.NoError(t, err)
	require.Equal(t, "http://www.ds.us-west-1.aws.observability.tidbcloud.com/internal/topsql-vmselect/select/123/prometheus/api/v1/query", url)
}