129 lines
3.5 KiB
Go
129 lines
3.5 KiB
Go
package testutil
|
|
|
|
import (
|
|
"platform/pkg/remote"
|
|
"sync"
|
|
"testing"
|
|
)
|
|
|
|
// MockCloudClient 是CloudClient接口的测试实现
|
|
type MockCloudClient struct {
|
|
// 存储预期结果的字段
|
|
EdgesMock func(param remote.CloudEdgesReq) (*remote.CloudEdgesResp, error)
|
|
ConnectMock func(param remote.CloudConnectReq) error
|
|
DisconnectMock func(param remote.CloudDisconnectReq) (int, error)
|
|
AutoQueryMock func() (remote.CloudConnectResp, error)
|
|
|
|
// 记录调用历史
|
|
EdgesCalls []remote.CloudEdgesReq
|
|
ConnectCalls []remote.CloudConnectReq
|
|
DisconnectCalls []remote.CloudDisconnectReq
|
|
AutoQueryCalls int
|
|
|
|
// 用于并发安全
|
|
mu sync.Mutex
|
|
}
|
|
|
|
// 确保MockCloudClient实现了CloudClient接口
|
|
var _ remote.CloudClient = (*MockCloudClient)(nil)
|
|
|
|
func (m *MockCloudClient) CloudEdges(param remote.CloudEdgesReq) (*remote.CloudEdgesResp, error) {
|
|
m.mu.Lock()
|
|
defer m.mu.Unlock()
|
|
m.EdgesCalls = append(m.EdgesCalls, param)
|
|
if m.EdgesMock != nil {
|
|
return m.EdgesMock(param)
|
|
}
|
|
return &remote.CloudEdgesResp{}, nil
|
|
}
|
|
|
|
func (m *MockCloudClient) CloudConnect(param remote.CloudConnectReq) error {
|
|
m.mu.Lock()
|
|
defer m.mu.Unlock()
|
|
m.ConnectCalls = append(m.ConnectCalls, param)
|
|
if m.ConnectMock != nil {
|
|
return m.ConnectMock(param)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (m *MockCloudClient) CloudDisconnect(param remote.CloudDisconnectReq) (int, error) {
|
|
m.mu.Lock()
|
|
defer m.mu.Unlock()
|
|
m.DisconnectCalls = append(m.DisconnectCalls, param)
|
|
if m.DisconnectMock != nil {
|
|
return m.DisconnectMock(param)
|
|
}
|
|
return 0, nil
|
|
}
|
|
|
|
func (m *MockCloudClient) CloudAutoQuery() (remote.CloudConnectResp, error) {
|
|
m.mu.Lock()
|
|
defer m.mu.Unlock()
|
|
m.AutoQueryCalls++
|
|
if m.AutoQueryMock != nil {
|
|
return m.AutoQueryMock()
|
|
}
|
|
return remote.CloudConnectResp{}, nil
|
|
}
|
|
|
|
// MockGatewayClient 是GatewayClient接口的测试实现
|
|
type MockGatewayClient struct {
|
|
// 存储预期结果的字段
|
|
PortConfigsMock func(params []remote.PortConfigsReq) error
|
|
PortActiveMock func(param ...remote.PortActiveReq) (map[string]remote.PortData, error)
|
|
|
|
// 记录调用历史
|
|
PortConfigsCalls [][]remote.PortConfigsReq
|
|
PortActiveCalls [][]remote.PortActiveReq
|
|
|
|
// 用于并发安全
|
|
mu sync.Mutex
|
|
}
|
|
|
|
// 确保MockGatewayClient实现了GatewayClient接口
|
|
var _ remote.GatewayClient = (*MockGatewayClient)(nil)
|
|
|
|
func (m *MockGatewayClient) GatewayPortConfigs(params []remote.PortConfigsReq) error {
|
|
m.mu.Lock()
|
|
defer m.mu.Unlock()
|
|
m.PortConfigsCalls = append(m.PortConfigsCalls, params)
|
|
if m.PortConfigsMock != nil {
|
|
return m.PortConfigsMock(params)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (m *MockGatewayClient) GatewayPortActive(param ...remote.PortActiveReq) (map[string]remote.PortData, error) {
|
|
m.mu.Lock()
|
|
defer m.mu.Unlock()
|
|
m.PortActiveCalls = append(m.PortActiveCalls, param)
|
|
if m.PortActiveMock != nil {
|
|
return m.PortActiveMock(param...)
|
|
}
|
|
return map[string]remote.PortData{}, nil
|
|
}
|
|
|
|
// SetupCloudClientMock 替换全局CloudClient为测试实现并在测试完成后恢复
|
|
func SetupCloudClientMock(t *testing.T) *MockCloudClient {
|
|
mock := &MockCloudClient{}
|
|
remote.Cloud = mock
|
|
|
|
return mock
|
|
}
|
|
|
|
// SetupGatewayClientMock 创建一个MockGatewayClient并提供替换函数
|
|
func SetupGatewayClientMock(t *testing.T) *MockGatewayClient {
|
|
mock := &MockGatewayClient{}
|
|
remote.GatewayInitializer = func(url, username, password string) remote.GatewayClient {
|
|
return mock
|
|
}
|
|
return mock
|
|
}
|
|
|
|
// NewMockGatewayClient 创建一个新的MockGatewayClient
|
|
// 保留此函数以保持向后兼容性
|
|
func NewMockGatewayClient() *MockGatewayClient {
|
|
return &MockGatewayClient{}
|
|
}
|