Add test for xof Reset()

dedis · Mar 26, 2024 · 0af6c4f · 0af6c4f
1 parent de86258
commit 0af6c4f
Showing 1 changed file with 24 additions and 0 deletions.
diff --git a/xof/xof_test.go b/xof/xof_test.go
@@ -214,6 +214,30 @@ func testReseed(t *testing.T, s kyber.XOFFactory) {
 	}
 }
 
+func TestReset(t *testing.T) {
+	for _, s := range impls {
+		testReset(t, s)
+	}
+}
+
+func testReset(t *testing.T, s kyber.XOFFactory) {
+	t.Logf("implementation %T", s)
+	seed := []byte("seed")
+	nbrBytes := 1024
+	x := s.XOF(seed)
+
+	beforeResetBytes := make([]byte, nbrBytes)
+	afterResetBytes := make([]byte, nbrBytes)
+	x.Read(beforeResetBytes)
+
+	x.Reset()
+	x.Read(afterResetBytes)
+
+	if !bytes.Equal(beforeResetBytes, afterResetBytes) {
+		t.Fatal("reset doesn't restore initial states")
+	}
+}
+
 func TestEncDecMismatch(t *testing.T) {
 	for _, i := range impls {
 		testEncDecMismatch(t, i)