47 lines
1.3 KiB
Go
47 lines
1.3 KiB
Go
package sub2api
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/QuantumNous/new-api/constant"
|
|
relaycommon "github.com/QuantumNous/new-api/relay/common"
|
|
relayconstant "github.com/QuantumNous/new-api/relay/constant"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestGetRequestURLAlphaSearch(t *testing.T) {
|
|
adaptor := &Adaptor{}
|
|
info := &relaycommon.RelayInfo{
|
|
ChannelMeta: &relaycommon.ChannelMeta{
|
|
ChannelType: constant.ChannelTypeSub2API,
|
|
ChannelBaseUrl: "https://sub2api.example",
|
|
},
|
|
RequestURLPath: "/v1/alpha/search",
|
|
RelayMode: relayconstant.RelayModeAlphaSearch,
|
|
}
|
|
|
|
url, err := adaptor.GetRequestURL(info)
|
|
require.NoError(t, err)
|
|
assert.Equal(t, "https://sub2api.example/v1/alpha/search", url)
|
|
}
|
|
|
|
func TestAdaptorInheritsNewAPIResponsesCompactSupport(t *testing.T) {
|
|
adaptor := &Adaptor{}
|
|
info := &relaycommon.RelayInfo{
|
|
ChannelMeta: &relaycommon.ChannelMeta{
|
|
ChannelType: constant.ChannelTypeSub2API,
|
|
ChannelBaseUrl: "https://sub2api.example",
|
|
},
|
|
RequestURLPath: "/v1/responses/compact",
|
|
RelayMode: relayconstant.RelayModeResponsesCompact,
|
|
}
|
|
|
|
url, err := adaptor.GetRequestURL(info)
|
|
|
|
require.NoError(t, err)
|
|
assert.Equal(t, "https://sub2api.example/v1/responses/compact", url)
|
|
assert.Equal(t, "sub2api", adaptor.GetChannelName())
|
|
assert.Empty(t, adaptor.GetModelList())
|
|
}
|