66 lines
1.5 KiB
Go
66 lines
1.5 KiB
Go
|
package protocol
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"testing"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
func TestParseUpdate(t *testing.T) {
|
|||
|
|
// 构造测试数据（最小有效Update消息）
|
|||
|
|
data := []byte{
|
|||
|
|
// Header
|
|||
|
|
16, 0, 0, 0, // MessageLength (16 bytes)
|
|||
|
|
1, 0, 0, 0, // RequestID
|
|||
|
|
0, 0, 0, 0, // ResponseTo
|
|||
|
|
209, 7, 0, 0, // OpCode (OP_UPDATE)
|
|||
|
|
|
|||
|
|
// UpdateFlags (Upsert=1)
|
|||
|
|
1, 0, 0, 0,
|
|||
|
|
|
|||
|
|
// DatabaseName "test\x00collection\x00"
|
|||
|
|
't', 'e', 's', 't', 0,
|
|||
|
|
|
|||
|
|
// CollectionName "coll\x00"
|
|||
|
|
'c', 'o', 'l', 'l', 0,
|
|||
|
|
|
|||
|
|
// Query文档（最小BSON文档）
|
|||
|
|
5, 0, 0, 0, 0, // BSON长度 + 空文档
|
|||
|
|
|
|||
|
|
// Update文档
|
|||
|
|
5, 0, 0, 0, 0, // BSON长度 + 空文档
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
msg, err := ParseMessage(data)
|
|||
|
|
if err != nil {
|
|||
|
|
t.Fatalf("ParseMessage failed: %v", err)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
if msg.OpCode != OP_UPDATE {
|
|||
|
|
t.Errorf("Expected OP_UPDATE, got %d", msg.OpCode)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
updateMsg, ok := msg.Body.(*UpdateMessage)
|
|||
|
|
if !ok {
|
|||
|
|
t.Fatalf("Expected UpdateMessage, got %T", msg.Body)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
if updateMsg.Flags != Upsert {
|
|||
|
|
t.Errorf("Expected Upsert flag, got %v", updateMsg.Flags)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
if updateMsg.DatabaseName != "test" {
|
|||
|
|
t.Errorf("Expected database 'test', got '%s'", updateMsg.DatabaseName)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
if updateMsg.CollName != "coll" {
|
|||
|
|
t.Errorf("Expected collection 'coll', got '%s'", updateMsg.CollName)
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func TestParseInvalidUpdate(t *testing.T) {
|
|||
|
|
// 测试短数据包
|
|||
|
|
shortData := []byte{1, 0, 0, 0} // 长度不足4字节
|
|||
|
|
_, err := parseUpdate(shortData) // 使用空标识符丢弃未使用的返回值
|
|||
|
|
if err == nil {
|
|||
|
|
t.Error("Expected error for short data")
|
|||
|
|
}
|
|||
|
|
}
|