package common

import (
	"testing"

	"github.com/QuantumNous/new-api/constant"
	"github.com/stretchr/testify/require"
)

func TestTianyiYunChannelUsesVolcEngineAPIType(t *testing.T) {
	apiType, ok := ChannelType2APIType(constant.ChannelTypeDoubaoVideoCompatibleTianyiYun)

	require.True(t, ok)
	require.Equal(t, constant.APITypeVolcEngine, apiType)
}

func TestTianyiYunChannelUsesDoubaoVideoEndpointType(t *testing.T) {
	got := GetEndpointTypesByChannelType(constant.ChannelTypeDoubaoVideoCompatibleTianyiYun, "cdance2.0-0611")

	require.Equal(t, []constant.EndpointType{constant.EndpointTypeDoubaoVideo}, got)
}