package cvt

import (
	"testing"

	"github.com/stretchr/testify/require"
	dataflowng "github.com/tidbcloud/dataflow-service-ng/idl/pbgen/proto/consoleapi"
	"github.com/tidbcloud/dataflow-service/idl/pbgen/proto/dataflow"
)

func TestChangefeedStatusPreservesWarning(t *testing.T) {
	require.Equal(t, "warning", connectorStatus(dataflow.Connector_WARNING))
	require.Equal(t, "error", connectorStatus(dataflow.Connector_ERROR))
	require.Equal(t, "warning", ngChangefeedStatus(dataflowng.Changefeed_WARNING))
	require.Equal(t, "error", ngChangefeedStatus(dataflowng.Changefeed_ERROR))
}