chore: apply task changes

internal/gateway/crypto_test.go

@@ -0,0 +1,103 @@
+package gateway
+
+import (
+	"crypto/aes"
+	"encoding/base64"
+	"strings"
+	"testing"
+)
+
+func TestAESEncrypt(t *testing.T) {
+	tests := []struct {
+		name      string
+		data      []byte
+		appSecret string
+		wantErr   bool
+	}{
+		{
+			name:      "正常加密",
+			data:      []byte(`{"params":{"cardNo":"898608070422D0010269"}}`),
+			appSecret: "BZeQttaZQt0i73moF",
+			wantErr:   false,
+		},
+		{
+			name:      "空数据加密",
+			data:      []byte(""),
+			appSecret: "BZeQttaZQt0i73moF",
+			wantErr:   false,
+		},
+	}
+
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			encrypted, err := aesEncrypt(tt.data, tt.appSecret)
+			if (err != nil) != tt.wantErr {
+				t.Errorf("aesEncrypt() error = %v, wantErr %v", err, tt.wantErr)
+				return
+			}
+			if !tt.wantErr && encrypted == "" {
+				t.Error("aesEncrypt() 返回空字符串")
+			}
+			// 验证 Base64 格式
+			if !tt.wantErr {
+				_, err := base64.StdEncoding.DecodeString(encrypted)
+				if err != nil {
+					t.Errorf("aesEncrypt() 返回的不是有效的 Base64: %v", err)
+				}
+			}
+		})
+	}
+}
+
+func TestGenerateSign(t *testing.T) {
+	appID := "60bgt1X8i7AvXqkd"
+	encryptedData := "test_encrypted_data"
+	timestamp := int64(1704067200)
+	appSecret := "BZeQttaZQt0i73moF"
+
+	sign := generateSign(appID, encryptedData, timestamp, appSecret)
+
+	// 验证签名格式（32 位大写十六进制）
+	if len(sign) != 32 {
+		t.Errorf("签名长度错误: got %d, want 32", len(sign))
+	}
+
+	if sign != strings.ToUpper(sign) {
+		t.Error("签名应为大写")
+	}
+
+	// 验证签名可重现
+	sign2 := generateSign(appID, encryptedData, timestamp, appSecret)
+	if sign != sign2 {
+		t.Error("相同参数应生成相同签名")
+	}
+}
+
+func TestNewECBEncrypterPanic(t *testing.T) {
+	defer func() {
+		if recover() == nil {
+			t.Fatal("newECBEncrypter 期望触发 panic，但未触发")
+		}
+	}()
+
+	newECBEncrypter(nil)
+}
+
+func TestECBEncrypterCryptBlocksPanic(t *testing.T) {
+	block, err := aes.NewCipher(make([]byte, aesBlockSize))
+	if err != nil {
+		t.Fatalf("创建 AES cipher 失败: %v", err)
+	}
+
+	encrypter := newECBEncrypter(block)
+	defer func() {
+		if recover() == nil {
+			t.Fatal("CryptBlocks 期望触发 panic，但未触发")
+		}
+	}()
+
+	// 传入非完整块长度，触发 panic
+	src := []byte("short")
+	dst := make([]byte, len(src))
+	encrypter.CryptBlocks(dst, src)
+}