new-api/relay/relay_adaptor_tianyiyun_test.go

package relay

import (
	"strconv"
	"testing"

	"github.com/QuantumNous/new-api/constant"
	tianyiyun "github.com/QuantumNous/new-api/relay/channel/task/doubao_tianyiyun"
	"github.com/stretchr/testify/require"
)

func TestGetTaskAdaptorReturnsTianyiYunSeedanceAdaptor(t *testing.T) {
	adaptor := GetTaskAdaptor(constant.TaskPlatform(strconv.Itoa(constant.ChannelTypeDoubaoVideoCompatibleTianyiYun)))

	require.IsType(t, &tianyiyun.TaskAdaptor{}, adaptor)
}