goaidb/protocol/bson_test.go

package protocol

import (
	"strings"
	"testing"
)

func TestParseBSON_EmptyDocument(t *testing.T) {
	// 构造最小有效BSON文档（仅包含长度和结束符）
	data := []byte{
		5, 0, 0, 0, // 文档长度=5字节
		0x00, // 结束符
	}

	result, err := ParseBSON(data)
	if err != nil {
		t.Fatalf("ParseBSON failed: %v", err)
	}

	if len(result) != 0 {
		t.Errorf("Expected empty document, got %d elements", len(result))
	}
}

func TestParseBSON_Int32(t *testing.T) {
	// 构造包含Int32字段的BSON文档
	data := []byte{
		9, 0, 0, 0, // 文档长度=9字节
		0x10,                     // Int32类型
		't', 'e', 's', 't', 0x00, // 键名"test"
		0x12, 0x34, 0x00, 0x00, // 值=0x00003412 (小端序)
		0x00, // 结束符
	}

	result, err := ParseBSON(data)
	if err != nil {
		t.Fatalf("ParseBSON failed: %v", err)
	}

	value, ok := result["test"]
	if !ok {
		t.Error("Expected key 'test' not found")
		return
	}

	if intValue, ok := value.(int32); !ok || intValue != 0x00003412 {
		t.Errorf("Expected Int32 0x00003412, got %v (%T)", value, value)
	}
}

func TestParseBSON_String(t *testing.T) {
	// 构造包含字符串字段的BSON文档
	data := []byte{
		16, 0, 0, 0, // 文档长度=22字节
		0x02,                     // String类型
		'n', 'a', 'm', 'e', 0x00, // 键名"name"
		5, 0, 0, 0, // 字符串长度=7字节（含终止符）
		'h', 'e', 'l', 'l', 'o', 0x00, // 字符串内容
		0x00, // 结束符
	}

	result, err := ParseBSON(data)
	if err != nil {
		t.Fatalf("ParseBSON failed: %v", err)
	}

	value, ok := result["name"]
	if !ok {
		t.Error("Expected key 'name' not found")
		return
	}

	if strValue, ok := value.(string); !ok || strValue != "hello" {
		t.Errorf("Expected string 'hello', got %q (%T)", value, value)
	}
}

func TestParseBSON_UpdateOperators(t *testing.T) {
	// 测试$set操作符解析
	setData := []byte{
		0x1f, 0, 0, 0, // 文档长度=21字节
		0x03,                     // 嵌入文档类型
		'$', 's', 'e', 't', 0x00, // $set操作符键名
		0x14, 0, 0, 0, // 子文档长度
		0x02,                     // 字符串类型
		'n', 'a', 'm', 'e', 0x00, // 键名"name"
		5, 0, 0, 0, // 字符串长度=6字节（含终止符）
		'j', 'o', 'h', 'n', 0x00, // 字符串内容
		0x00, // 结束符
		0x00, // 主文档结束符
	}

	result, err := ParseBSON(setData)
	if err != nil {
		t.Fatalf("ParseBSON failed: %v", err)
	}

	value, ok := result["$set"]
	if !ok {
		t.Error("Expected key '$set' not found")
		return
	}

	if subDoc, ok := value.(map[string]interface{}); !ok || subDoc["name"] != "john" {
		t.Errorf("Expected $set{name: 'john'}, got %v", value)
	}

	// 测试$inc操作符解析
	incData := []byte{
		0x19, 0, 0, 0, // 文档长度=21字节
		0x03,                     // 嵌入文档类型
		'$', 'i', 'n', 'c', 0x00, // $inc操作符键名
		0x0e, 0, 0, 0, // 子文档长度
		0x10,                // Int32类型
		'a', 'g', 'e', 0x00, // 键名"age"
		22, 0, 0, 0, // 值=22
		0x00, // 结束符
		0x00, // 主文档结束符
	}

	result, err = ParseBSON(incData)
	if err != nil {
		t.Fatalf("ParseBSON failed: %v", err)
	}

	value, ok = result["$inc"]
	if !ok {
		t.Error("Expected key '$inc' not found")
		return
	}

	if subDoc, ok := value.(map[string]interface{}); !ok || subDoc["age"] != int32(22) {
		t.Errorf("Expected $inc{age: 22}, got %v", value)
	}
}

func TestParseBSON_ErrorCases(t *testing.T) {
	// 测试数据过短的情况
	shortData := []byte{4, 0, 0, 0} // 长度为4字节的文档（仅包含长度字段）
	_, err := ParseBSON(shortData)
	if err == nil {
		t.Error("Expected error for short data")
	}

	// 测试无效元素类型
	invalidTypeData := []byte{
		6, 0, 0, 0, // 文档长度=6字节
		0xFF,                // 无效元素类型
		'k', 'e', 'y', 0x00, // 键名"key"
		0x00, // 结束符
	}
	_, err = ParseBSON(invalidTypeData)
	if err == nil {
		t.Error("Expected error for invalid element type")
	} else if !containsError(err, "unsupported BSON element") {
		t.Errorf("Expected unsupported element type error, got %v", err)
	}
}

// Helper function to check if error message contains expected text
func containsError(err error, expected string) bool {
	return err != nil && len(err.Error()) >= len(expected) && strings.Contains(err.Error(), expected)
}