one-api/providers/ali/ali_test.go
Buer ef041e28a1
♻️ refactor: provider refactor (#41)
* ♻️ refactor: provider refactor
* 完善百度/讯飞的函数调用,现在可以在`lobe-chat`中正常调用函数了
2024-01-19 02:47:10 +08:00

25 lines
515 B
Go

package ali_test
import (
"net/http"
"one-api/common"
"one-api/common/test"
"one-api/model"
)
func setupAliTestServer() (baseUrl string, server *test.ServerTest, teardown func()) {
server = test.NewTestServer()
ts := server.TestServer(func(w http.ResponseWriter, r *http.Request) bool {
return test.OpenAICheck(w, r)
})
ts.Start()
teardown = ts.Close
baseUrl = ts.URL
return
}
func getAliChannel(baseUrl string) model.Channel {
return test.GetChannel(common.ChannelTypeAli, baseUrl, "", "", "")
}