add test

Author: crimson-gao

alibabacloud-gateway-sls/util/golang/client/log_serializer_test.go

@@ -0,0 +1,71 @@
+package client
+
+import (
+	"testing"
+
+	"github.com/gogo/protobuf/proto"
+)
+
+func TestSerializeLogGroupToPB(t *testing.T) {
+	// Prepare test data
+	logGroupMap := map[string]interface{}{
+		"Topic":  "test-topic",
+		"Source": "test-source",
+		"LogTags": []interface{}{
+			map[string]interface{}{
+				"Key":   "tag-key",
+				"Value": "tag-value",
+			},
+		},
+		"LogItems": []interface{}{
+			map[string]interface{}{
+				"Time":   int(1678888888),
+				"TimeNs": int(123456),
+				"Contents": []interface{}{
+					map[string]interface{}{
+						"Key":   "content-key",
+						"Value": "content-value",
+					},
+				},
+			},
+		},
+	}
+
+	// Execute
+	data, err := serializeLogGroupToPB(logGroupMap)
+	if err != nil {
+		t.Fatalf("serializeLogGroupToPB failed: %v", err)
+	}
+
+	// Validate
+	if len(data) == 0 {
+		t.Fatal("serialized data is empty")
+	}
+
+	var lg LogGroup
+	if err := proto.Unmarshal(data, &lg); err != nil {
+		t.Fatalf("failed to unmarshal serialized data: %v", err)
+	}
+
+	if lg.GetTopic() != "test-topic" {
+		t.Errorf("expected topic 'test-topic', got %s", lg.GetTopic())
+	}
+
+	if lg.GetSource() != "test-source" {
+		t.Errorf("expected source 'test-source', got %s", lg.GetSource())
+	}
+
+	if len(lg.LogTags) != 1 {
+		t.Fatalf("expected 1 tag, got %d", len(lg.LogTags))
+	}
+	if lg.LogTags[0].GetKey() != "tag-key" || lg.LogTags[0].GetValue() != "tag-value" {
+		t.Errorf("unexpected tag content: %v", lg.LogTags[0])
+	}
+
+	if len(lg.Logs) != 1 {
+		t.Fatalf("expected 1 log item, got %d", len(lg.Logs))
+	}
+	if lg.Logs[0].GetTime() != 1678888888 {
+		t.Errorf("expected time 1678888888, got %d", lg.Logs[0].GetTime())
+	}
+}

alibabacloud-gateway-sls/util/golang/client/zstd_test.go

@@ -0,0 +1,51 @@
+package client
+
+import (
+	"bytes"
+	"io/ioutil"
+	"strconv"
+	"testing"
+
+	"github.com/alibabacloud-go/tea/tea"
+)
+
+func TestZstdSimple(t *testing.T) {
+	// 准备测试数据
+	input := "Hello, Zstd!"
+	originalBytes := []byte(input)
+	compressType := "zstd"
+
+	// 1. 压缩
+	compressedBytes, err := Compress(originalBytes, &compressType)
+	if err != nil {
+		t.Fatalf("Compression failed: %v", err)
+	}
+
+	// 确保压缩后的数据不为空
+	if len(compressedBytes) == 0 {
+		t.Fatal("Compressed data is empty")
+	}
+
+	// 2. 解压
+	// ReadAndUncompressBlock 需要原始大小作为字符串参数
+	rawSizeStr := strconv.Itoa(len(originalBytes))
+
+	resultReader, err := ReadAndUncompressBlock(
+		bytes.NewReader(compressedBytes),
+		&compressType,
+		tea.String(rawSizeStr),
+	)
+	if err != nil {
+		t.Fatalf("Decompression failed: %v", err)
+	}
+
+	decodedBytes, err := ioutil.ReadAll(resultReader)
+	if err != nil {
+		t.Fatalf("Failed to read decompressed data: %v", err)
+	}
+
+	// 3. 验证
+	if string(decodedBytes) != input {
+		t.Errorf("Content mismatch.\nExpected: %s\nGot: %s", input, string(decodedBytes))
+	}
+}