完善测试用例

hmacx/hmacx_test.go

@@ -0,0 +1,75 @@
+package hmacx
+
+import (
+	"testing"
+)
+
+func TestHMACSHA256(t *testing.T) {
+	message := "hello world"
+	key := "secret_key"
+
+	// 计算HMAC-SHA256
+	hmac1 := HMACSHA256(message, key)
+
+	// 再次计算相同的消息和密钥
+	hmac2 := HMACSHA256(message, key)
+
+	// 相同的消息和密钥应该产生相同的HMAC
+	if hmac1 != hmac2 {
+		t.Errorf("相同输入产生了不同的HMAC: %s vs %s", hmac1, hmac2)
+	}
+
+	// 不同的消息应该产生不同的HMAC
+	hmac3 := HMACSHA256("different message", key)
+	if hmac1 == hmac3 {
+		t.Errorf("不同消息产生了相同的HMAC")
+	}
+
+	// 不同的密钥应该产生不同的HMAC
+	hmac4 := HMACSHA256(message, "different_key")
+	if hmac1 == hmac4 {
+		t.Errorf("不同密钥产生了相同的HMAC")
+	}
+}
+
+func TestHMACSHA512(t *testing.T) {
+	message := "hello world"
+	key := "secret_key"
+
+	// 计算HMAC-SHA512
+	hmac1 := HMACSHA512(message, key)
+
+	// 再次计算相同的消息和密钥
+	hmac2 := HMACSHA512(message, key)
+
+	// 相同的消息和密钥应该产生相同的HMAC
+	if hmac1 != hmac2 {
+		t.Errorf("相同输入产生了不同的HMAC: %s vs %s", hmac1, hmac2)
+	}
+
+	// 不同的消息应该产生不同的HMAC
+	hmac3 := HMACSHA512("different message", key)
+	if hmac1 == hmac3 {
+		t.Errorf("不同消息产生了相同的HMAC")
+	}
+
+	// 不同的密钥应该产生不同的HMAC
+	hmac4 := HMACSHA512(message, "different_key")
+	if hmac1 == hmac4 {
+		t.Errorf("不同密钥产生了相同的HMAC")
+	}
+}
+
+func TestHMACSHA256VsSHA512(t *testing.T) {
+	message := "hello world"
+	key := "secret_key"
+
+	// 计算HMAC-SHA256和HMAC-SHA512
+	hmac256 := HMACSHA256(message, key)
+	hmac512 := HMACSHA512(message, key)
+
+	// SHA256和SHA512应该产生不同的HMAC
+	if hmac256 == hmac512 {
+		t.Errorf("SHA256和SHA512产生了相同的HMAC")
+	}
+}