package common

import (
	"testing"

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

func TestChinaMobileSeedanceChannelMappings(t *testing.T) {
	apiType, ok := ChannelType2APIType(constant.ChannelTypeChinaMobileSeedance)
	require.True(t, ok)
	require.Equal(t, constant.APITypeVolcEngine, apiType)

	endpoints := GetEndpointTypesByChannelType(constant.ChannelTypeChinaMobileSeedance, "doubao-seedance-2.0")
	require.Contains(t, endpoints, constant.EndpointTypeDoubaoVideo)
}