From 08bb2e2b5724d6ea1d889041078714ac0213dc51 Mon Sep 17 00:00:00 2001
From: Sun Yimin <emmansun@users.noreply.github.com>
Date: Fri, 15 Jul 2022 17:18:14 +0800
Subject: [PATCH] zuc: add test case

---
 zuc/eia256_test.go | 34 ++++++++++++++++++++++++++--------
 1 file changed, 26 insertions(+), 8 deletions(-)

diff --git a/zuc/eia256_test.go b/zuc/eia256_test.go
index 555ae50..b6327e1 100644
--- a/zuc/eia256_test.go
+++ b/zuc/eia256_test.go
@@ -230,15 +230,33 @@ func TestEIA256_Sum32(t *testing.T) {
 	}
 }
 
-func TestEIA256_Finish32(t *testing.T) {
-	expected := "f4f20d7c"
-	h, err := NewHash256(zucEIA256Tests[2].key, zucEIA256Tests[2].iv, 4)
-	if err != nil {
-		t.Fatal(err)
+func TestEIA256_Finish(t *testing.T) {
+	expected := []struct {
+		expected string
+		macLen   int
+	}{
+		{
+			"9dd592c4",
+			4,
+		},
+		{
+			"1f6f71e386a2ce01",
+			8,
+		},
+		{
+			"bf5339cfd87bba97d70ef4f5973af8bb",
+			16,
+		},
 	}
-	mac := h.Finish([]byte("emmansunshangmi1emmansun shangmiemmansun shangmi 1234"), 8*53)
-	if hex.EncodeToString(mac) != expected {
-		t.Errorf("expected=%s, result=%s\n", expected, hex.EncodeToString(mac))
+	for _, exp := range expected {
+		h, err := NewHash256(zucEIA256Tests[2].key, zucEIA256Tests[2].iv, exp.macLen)
+		if err != nil {
+			t.Fatal(err)
+		}
+		mac := h.Finish([]byte("emmansunshangmi1emmansun shangmiemmansun shangmi 12345"), 8*53+4)
+		if hex.EncodeToString(mac) != exp.expected {
+			t.Errorf("expected=%s, result=%s\n", exp.expected, hex.EncodeToString(mac))
+		}
 	}
 }