package crrf

import (
	"testing"

	"github.com/stretchr/testify/require"
	"github.com/tidbcloud/dataflow-service-e2e/internal/tidbcloud/dedicated/consoleapi"
	br_model "github.com/tidbcloud/dataflow-service-e2e/internal/tidbcloud/dedicated/consoleapi/br/model"
)

func TestGetSecondaryRegionFromBackupSetting(t *testing.T) {
	regions := []*consoleapi.RegionInfoResp{
		{Name: "us-west-2"},
		{Name: "us-west-1"},
	}

	got, err := GetSecondaryRegionFromBackupSetting("us-west-2", regions)
	require.NoError(t, err)
	require.Equal(t, "us-west-1", got)
}

func TestBuildCRRFRestoreRequest(t *testing.T) {
	req := BuildCRRFRestoreRequest("restore-name", "us-west-1", 1735689600)
	require.Equal(t, "us-west-1", req.Region)
	require.Empty(t, req.BackupRegion)
	require.Equal(t, string(br_model.DataflowRestoreModeRESTOREMODEPITR), req.RestoreMode)
}