lanhu/platform
2
0
Fork 0
Code Issues Actions Packages Releases 13 Activity

更新提取接口参数，完善流程与测试代码逻辑

Browse Source
This commit is contained in:
luorijun
2025-04-14 16:00:04 +08:00
parent 8c268fd7a9
commit e9749db50c
10 changed files with 140 additions and 115 deletions
Download Patch File Download Diff File Expand all files Collapse all files

85
web/services/channel_test.go
View File

@@ -288,26 +288,6 @@ func Test_channelService_CreateChannel(t *testing.T) {
ctx := context.WithValue(context.Background(), requestid.ConfigDefault.ContextKey, "test-request-id")
var adminAuth = &AuthContext{Payload: Payload{Id: 100, Type: PayloadAdmin}}
var userAuth = &AuthContext{Payload: Payload{Id: 101, Type: PayloadUser}}
var user = &models.User{
ID: 101,
Phone: "12312341234",
}
db.Create(user)
var whitelists = []*models.Whitelist{
{ID: 1, UserID: 101, Host: "123.123.123.123"},
{ID: 2, UserID: 101, Host: "456.456.456.456"},
{ID: 3, UserID: 101, Host: "789.789.789.789"},
}
db.Create(whitelists)
var proxy = &models.Proxy{
ID: 1,
Version: 1,
Name: "test-proxy",
Host: "111.111.111.111",
Type: 1,
Secret: "test:secret",
}
db.Create(proxy)
mc.AutoQueryMock = func() (remote.CloudConnectResp, error) {
return remote.CloudConnectResp{
"test-proxy": []remote.AutoConfig{
@@ -315,18 +295,48 @@ func Test_channelService_CreateChannel(t *testing.T) {
},
}, nil
}
var clearDb = func() {
db.Exec("delete from resource where true")
var resource = &models.Resource{
var user *models.User
var whitelists []*models.Whitelist
var proxy *models.Proxy
var resource *models.Resource
var resourcePss *models.ResourcePss
var resetDb = func() {
user = &models.User{
ID: 101,
Phone: "12312341234",
}
db.Exec("delete from user where true")
db.Create(user)
whitelists = []*models.Whitelist{
{ID: 1, UserID: 101, Host: "123.123.123.123"},
{ID: 2, UserID: 101, Host: "456.456.456.456"},
{ID: 3, UserID: 101, Host: "789.789.789.789"},
}
db.Exec("delete from whitelist where true")
db.Create(whitelists)
proxy = &models.Proxy{
ID: 1,
Version: 1,
Name: "test-proxy",
Host: "111.111.111.111",
Type: 1,
Secret: "test:secret",
}
db.Exec("delete from proxy where true")
db.Create(proxy)
resource = &models.Resource{
ID: 1,
UserID: 101,
Active: true,
}
db.Exec("delete from resource where true")
db.Create(resource)
db.Exec("delete from resource_pss where true")
var resourcePss = &models.ResourcePss{
resourcePss = &models.ResourcePss{
ID: 1,
ResourceID: 1,
Type: 1,
@@ -334,10 +344,12 @@ func Test_channelService_CreateChannel(t *testing.T) {
Expire: common.LocalDateTime(time.Now().AddDate(1, 0, 0)),
DailyLimit: 10000,
}
db.Exec("delete from resource_pss where true")
db.Create(resourcePss)
db.Exec("delete from channel where true")
}
tests := []struct {
name string
args args
@@ -359,7 +371,7 @@ func Test_channelService_CreateChannel(t *testing.T) {
},
setup: func() {
mr.FlushAll()
clearDb()
resetDb()
mc.ConnectMock = func(param remote.CloudConnectReq) error {
if param.Uuid != proxy.Name {
@@ -509,7 +521,7 @@ func Test_channelService_CreateChannel(t *testing.T) {
},
setup: func() {
mr.FlushAll()
clearDb()
resetDb()
mc.ConnectMock = func(param remote.CloudConnectReq) error {
if param.Uuid != proxy.Name {
@@ -653,7 +665,7 @@ func Test_channelService_CreateChannel(t *testing.T) {
},
setup: func() {
mr.FlushAll()
clearDb()
resetDb()
mc.ConnectMock = func(param remote.CloudConnectReq) error {
if param.Uuid != proxy.Name {
@@ -802,7 +814,7 @@ func Test_channelService_CreateChannel(t *testing.T) {
},
setup: func() {
mr.FlushAll()
clearDb()
resetDb()
},
wantErr: true,
wantErrContains: "无权限访问",
@@ -819,7 +831,7 @@ func Test_channelService_CreateChannel(t *testing.T) {
},
setup: func() {
mr.FlushAll()
clearDb()
resetDb()
resource2 := &models.Resource{
ID: 2,
@@ -852,7 +864,7 @@ func Test_channelService_CreateChannel(t *testing.T) {
},
setup: func() {
mr.FlushAll()
clearDb()
resetDb()
// 创建一个配额几乎用完的资源包
resource2 := models.Resource{
@@ -886,7 +898,14 @@ func Test_channelService_CreateChannel(t *testing.T) {
},
setup: func() {
mr.FlushAll()
clearDb()
resetDb()
mc.AutoQueryMock = func() (remote.CloudConnectResp, error) {
return remote.CloudConnectResp{
"test-proxy": []remote.AutoConfig{
{Count: 20000},
},
}, nil
}
// 创建大量占用端口的通道
var channels = make([]models.Channel, 10000)
var expr = time.Now().Add(time.Hour)
@@ -908,8 +927,6 @@ func Test_channelService_CreateChannel(t *testing.T) {
for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) {
mr.FlushAll()
clearDb()
if tt.setup != nil {
tt.setup()
}