refactor: 重构trustlog-sdk目录结构到trustlog/go-trustlog
- 将所有trustlog-sdk文件移动到trustlog/go-trustlog/目录 - 更新README中所有import路径从trustlog-sdk改为go-trustlog - 更新cookiecutter配置文件中的项目名称 - 更新根目录.lefthook.yml以引用新位置的配置 - 添加go.sum文件到版本控制 - 删除过时的示例文件 这次重构与trustlog-server保持一致的目录结构, 为未来支持多语言SDK(Python、Java等)预留空间。
This commit is contained in:
166
api/adapter/tcp_protocol_test.go
Normal file
166
api/adapter/tcp_protocol_test.go
Normal file
@@ -0,0 +1,166 @@
|
||||
package adapter
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"testing"
|
||||
)
|
||||
|
||||
func TestEncodeTCPMessage(t *testing.T) {
|
||||
tests := []struct {
|
||||
name string
|
||||
msg *TCPMessage
|
||||
wantErr bool
|
||||
}{
|
||||
{
|
||||
name: "valid data message",
|
||||
msg: &TCPMessage{
|
||||
Type: MessageTypeData,
|
||||
Topic: "test-topic",
|
||||
UUID: "test-uuid-1234",
|
||||
Payload: []byte("test payload"),
|
||||
},
|
||||
wantErr: false,
|
||||
},
|
||||
{
|
||||
name: "valid ack message",
|
||||
msg: &TCPMessage{
|
||||
Type: MessageTypeAck,
|
||||
Topic: "",
|
||||
UUID: "test-uuid-5678",
|
||||
Payload: nil,
|
||||
},
|
||||
wantErr: false,
|
||||
},
|
||||
{
|
||||
name: "nil message",
|
||||
msg: nil,
|
||||
wantErr: true,
|
||||
},
|
||||
}
|
||||
|
||||
for _, tt := range tests {
|
||||
t.Run(tt.name, func(t *testing.T) {
|
||||
data, err := EncodeTCPMessage(tt.msg)
|
||||
if (err != nil) != tt.wantErr {
|
||||
t.Errorf("EncodeTCPMessage() error = %v, wantErr %v", err, tt.wantErr)
|
||||
return
|
||||
}
|
||||
|
||||
if !tt.wantErr && data == nil {
|
||||
t.Error("EncodeTCPMessage() returned nil data")
|
||||
}
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
func TestDecodeTCPMessage(t *testing.T) {
|
||||
// 创建一个测试消息
|
||||
original := &TCPMessage{
|
||||
Type: MessageTypeData,
|
||||
Topic: "test-topic",
|
||||
UUID: "test-uuid-1234",
|
||||
Payload: []byte("test payload data"),
|
||||
}
|
||||
|
||||
// 编码
|
||||
encoded, err := EncodeTCPMessage(original)
|
||||
if err != nil {
|
||||
t.Fatalf("Failed to encode message: %v", err)
|
||||
}
|
||||
|
||||
// 解码
|
||||
reader := bytes.NewReader(encoded)
|
||||
decoded, err := DecodeTCPMessage(reader)
|
||||
if err != nil {
|
||||
t.Fatalf("Failed to decode message: %v", err)
|
||||
}
|
||||
|
||||
// 验证
|
||||
if decoded.Type != original.Type {
|
||||
t.Errorf("Type mismatch: got %v, want %v", decoded.Type, original.Type)
|
||||
}
|
||||
if decoded.Topic != original.Topic {
|
||||
t.Errorf("Topic mismatch: got %v, want %v", decoded.Topic, original.Topic)
|
||||
}
|
||||
if decoded.UUID != original.UUID {
|
||||
t.Errorf("UUID mismatch: got %v, want %v", decoded.UUID, original.UUID)
|
||||
}
|
||||
if !bytes.Equal(decoded.Payload, original.Payload) {
|
||||
t.Errorf("Payload mismatch: got %v, want %v", decoded.Payload, original.Payload)
|
||||
}
|
||||
}
|
||||
|
||||
func TestEncodeDecodeRoundTrip(t *testing.T) {
|
||||
testCases := []struct {
|
||||
name string
|
||||
msg *TCPMessage
|
||||
}{
|
||||
{
|
||||
name: "data message with payload",
|
||||
msg: &TCPMessage{
|
||||
Type: MessageTypeData,
|
||||
Topic: "persistent://public/default/test",
|
||||
UUID: "550e8400-e29b-41d4-a716-446655440000",
|
||||
Payload: []byte("Hello, World!"),
|
||||
},
|
||||
},
|
||||
{
|
||||
name: "ack message",
|
||||
msg: &TCPMessage{
|
||||
Type: MessageTypeAck,
|
||||
Topic: "",
|
||||
UUID: "test-uuid",
|
||||
Payload: nil,
|
||||
},
|
||||
},
|
||||
{
|
||||
name: "nack message",
|
||||
msg: &TCPMessage{
|
||||
Type: MessageTypeNack,
|
||||
Topic: "",
|
||||
UUID: "another-uuid",
|
||||
Payload: []byte{},
|
||||
},
|
||||
},
|
||||
{
|
||||
name: "message with large payload",
|
||||
msg: &TCPMessage{
|
||||
Type: MessageTypeData,
|
||||
Topic: "test",
|
||||
UUID: "uuid",
|
||||
Payload: make([]byte, 10000),
|
||||
},
|
||||
},
|
||||
}
|
||||
|
||||
for _, tc := range testCases {
|
||||
t.Run(tc.name, func(t *testing.T) {
|
||||
// 编码
|
||||
encoded, err := EncodeTCPMessage(tc.msg)
|
||||
if err != nil {
|
||||
t.Fatalf("Encode failed: %v", err)
|
||||
}
|
||||
|
||||
// 解码
|
||||
reader := bytes.NewReader(encoded)
|
||||
decoded, err := DecodeTCPMessage(reader)
|
||||
if err != nil {
|
||||
t.Fatalf("Decode failed: %v", err)
|
||||
}
|
||||
|
||||
// 验证所有字段
|
||||
if decoded.Type != tc.msg.Type {
|
||||
t.Errorf("Type: got %v, want %v", decoded.Type, tc.msg.Type)
|
||||
}
|
||||
if decoded.Topic != tc.msg.Topic {
|
||||
t.Errorf("Topic: got %v, want %v", decoded.Topic, tc.msg.Topic)
|
||||
}
|
||||
if decoded.UUID != tc.msg.UUID {
|
||||
t.Errorf("UUID: got %v, want %v", decoded.UUID, tc.msg.UUID)
|
||||
}
|
||||
if !bytes.Equal(decoded.Payload, tc.msg.Payload) {
|
||||
t.Errorf("Payload: got %v, want %v", decoded.Payload, tc.msg.Payload)
|
||||
}
|
||||
})
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user