package file_download

import (
	"testing"

	"github.com/stretchr/testify/require"

	"github.com/pingcap-inc/clinic/pkg/ngm/file_download/api"
)

func TestRewriteDownloadPaths(t *testing.T) {
	data := &api.ListData{Records: []api.ListFileRecord{{
		Files: []api.FileMetaInfo{{
			DownloadPath: "/ngm/api/v1/oom_record/files?provider=aws&key=object",
			DownloadURL:  "/ngm/api/v1/oom_record/files?provider=aws&key=object",
		}},
	}}}

	rewriteDownloadPaths(data, "/api/v1/orgs/123/ngm/clusters/456/oom_record/files")

	require.Equal(t,
		"/api/v1/orgs/123/ngm/clusters/456/oom_record/files?provider=aws&key=object",
		data.Records[0].Files[0].DownloadPath,
	)
	require.Equal(t, data.Records[0].Files[0].DownloadPath, data.Records[0].Files[0].DownloadURL)
}