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

重构代码结构，迁移Baiyin SDK相关逻辑至globals包，并添加支付宝客户端初始化

Browse Source
This commit is contained in:
luorijun
2025-04-16 14:01:30 +08:00
parent f1456d01ea
commit 083fabb308
12 changed files with 200 additions and 76 deletions
Download Patch File Download Diff File Expand all files Collapse all files

62
pkg/testutil/remote.go
View File

@@ -1,7 +1,7 @@
package testutil
import (
"platform/pkg/sdks/baiyin"
g "platform/web/globals"
"sync"
"testing"
)
@@ -9,15 +9,15 @@ import (
// MockCloudClient 是CloudClient接口的测试实现
type MockCloudClient struct {
// 存储预期结果的字段
EdgesMock func(param baiyin.CloudEdgesReq) (*baiyin.CloudEdgesResp, error)
ConnectMock func(param baiyin.CloudConnectReq) error
DisconnectMock func(param baiyin.CloudDisconnectReq) (int, error)
AutoQueryMock func() (baiyin.CloudConnectResp, error)
EdgesMock func(param g.CloudEdgesReq) (*g.CloudEdgesResp, error)
ConnectMock func(param g.CloudConnectReq) error
DisconnectMock func(param g.CloudDisconnectReq) (int, error)
AutoQueryMock func() (g.CloudConnectResp, error)
// 记录调用历史
EdgesCalls []baiyin.CloudEdgesReq
ConnectCalls []baiyin.CloudConnectReq
DisconnectCalls []baiyin.CloudDisconnectReq
EdgesCalls []g.CloudEdgesReq
ConnectCalls []g.CloudConnectReq
DisconnectCalls []g.CloudDisconnectReq
AutoQueryCalls int
// 用于并发安全
@@ -25,19 +25,19 @@ type MockCloudClient struct {
}
// 确保MockCloudClient实现了CloudClient接口
var _ baiyin.CloudClient = (*MockCloudClient)(nil)
var _ g.CloudClient = (*MockCloudClient)(nil)
func (m *MockCloudClient) CloudEdges(param baiyin.CloudEdgesReq) (*baiyin.CloudEdgesResp, error) {
func (m *MockCloudClient) CloudEdges(param g.CloudEdgesReq) (*g.CloudEdgesResp, error) {
m.mu.Lock()
defer m.mu.Unlock()
m.EdgesCalls = append(m.EdgesCalls, param)
if m.EdgesMock != nil {
return m.EdgesMock(param)
}
return &baiyin.CloudEdgesResp{}, nil
return &g.CloudEdgesResp{}, nil
}
func (m *MockCloudClient) CloudConnect(param baiyin.CloudConnectReq) error {
func (m *MockCloudClient) CloudConnect(param g.CloudConnectReq) error {
m.mu.Lock()
defer m.mu.Unlock()
m.ConnectCalls = append(m.ConnectCalls, param)
@@ -47,7 +47,7 @@ func (m *MockCloudClient) CloudConnect(param baiyin.CloudConnectReq) error {
return nil
}
func (m *MockCloudClient) CloudDisconnect(param baiyin.CloudDisconnectReq) (int, error) {
func (m *MockCloudClient) CloudDisconnect(param g.CloudDisconnectReq) (int, error) {
m.mu.Lock()
defer m.mu.Unlock()
m.DisconnectCalls = append(m.DisconnectCalls, param)
@@ -57,33 +57,33 @@ func (m *MockCloudClient) CloudDisconnect(param baiyin.CloudDisconnectReq) (int,
return 0, nil
}
func (m *MockCloudClient) CloudAutoQuery() (baiyin.CloudConnectResp, error) {
func (m *MockCloudClient) CloudAutoQuery() (g.CloudConnectResp, error) {
m.mu.Lock()
defer m.mu.Unlock()
m.AutoQueryCalls++
if m.AutoQueryMock != nil {
return m.AutoQueryMock()
}
return baiyin.CloudConnectResp{}, nil
return g.CloudConnectResp{}, nil
}
// SetupCloudClientMock 替换全局CloudClient为测试实现并在测试完成后恢复
func SetupCloudClientMock(t *testing.T) *MockCloudClient {
mock := &MockCloudClient{
EdgesMock: func(param baiyin.CloudEdgesReq) (*baiyin.CloudEdgesResp, error) {
EdgesMock: func(param g.CloudEdgesReq) (*g.CloudEdgesResp, error) {
panic("not implemented")
},
ConnectMock: func(param baiyin.CloudConnectReq) error {
ConnectMock: func(param g.CloudConnectReq) error {
panic("not implemented")
},
DisconnectMock: func(param baiyin.CloudDisconnectReq) (int, error) {
DisconnectMock: func(param g.CloudDisconnectReq) (int, error) {
panic("not implemented")
},
AutoQueryMock: func() (baiyin.CloudConnectResp, error) {
AutoQueryMock: func() (g.CloudConnectResp, error) {
panic("not implemented")
},
}
baiyin.Cloud = mock
g.Cloud = mock
return mock
}
@@ -94,9 +94,9 @@ type MockGatewayClient struct {
}
// 确保MockGatewayClient实现了GatewayClient接口
var _ baiyin.GatewayClient = (*MockGatewayClient)(nil)
var _ g.GatewayClient = (*MockGatewayClient)(nil)
func (m *MockGatewayClient) GatewayPortConfigs(params []baiyin.PortConfigsReq) error {
func (m *MockGatewayClient) GatewayPortConfigs(params []g.PortConfigsReq) error {
testGatewayBase.mu.Lock()
defer testGatewayBase.mu.Unlock()
testGatewayBase.PortConfigsCalls = append(testGatewayBase.PortConfigsCalls, params)
@@ -106,42 +106,42 @@ func (m *MockGatewayClient) GatewayPortConfigs(params []baiyin.PortConfigsReq) e
return nil
}
func (m *MockGatewayClient) GatewayPortActive(param ...baiyin.PortActiveReq) (map[string]baiyin.PortData, error) {
func (m *MockGatewayClient) GatewayPortActive(param ...g.PortActiveReq) (map[string]g.PortData, error) {
testGatewayBase.mu.Lock()
defer testGatewayBase.mu.Unlock()
testGatewayBase.PortActiveCalls = append(testGatewayBase.PortActiveCalls, param)
if testGatewayBase.PortActiveMock != nil {
return testGatewayBase.PortActiveMock(m, param...)
}
return map[string]baiyin.PortData{}, nil
return map[string]g.PortData{}, nil
}
type GatewayClientIns struct {
// 存储预期结果的字段
PortConfigsMock func(c *MockGatewayClient, params []baiyin.PortConfigsReq) error
PortActiveMock func(c *MockGatewayClient, param ...baiyin.PortActiveReq) (map[string]baiyin.PortData, error)
PortConfigsMock func(c *MockGatewayClient, params []g.PortConfigsReq) error
PortActiveMock func(c *MockGatewayClient, param ...g.PortActiveReq) (map[string]g.PortData, error)
// 记录调用历史
PortConfigsCalls [][]baiyin.PortConfigsReq
PortActiveCalls [][]baiyin.PortActiveReq
PortConfigsCalls [][]g.PortConfigsReq
PortActiveCalls [][]g.PortActiveReq
// 用于并发安全
mu sync.Mutex
}
var testGatewayBase = &GatewayClientIns{
PortConfigsMock: func(c *MockGatewayClient, params []baiyin.PortConfigsReq) error {
PortConfigsMock: func(c *MockGatewayClient, params []g.PortConfigsReq) error {
panic("not implemented")
},
PortActiveMock: func(c *MockGatewayClient, param ...baiyin.PortActiveReq) (map[string]baiyin.PortData, error) {
PortActiveMock: func(c *MockGatewayClient, param ...g.PortActiveReq) (map[string]g.PortData, error) {
panic("not implemented")
},
}
// SetupGatewayClientMock 创建一个MockGatewayClient并提供替换函数
func SetupGatewayClientMock(t *testing.T) *GatewayClientIns {
baiyin.GatewayInitializer = func(url, username, password string) baiyin.GatewayClient {
g.GatewayInitializer = func(url, username, password string) g.GatewayClient {
return &MockGatewayClient{
Host: url,
}