From 0af6c4fc3a966b8ffe9e1c5020062fd631df68f5 Mon Sep 17 00:00:00 2001
From: lauener <kilian.lauener@epfl.ch>
Date: Tue, 26 Mar 2024 08:30:55 +0100
Subject: [PATCH] Add test for xof Reset()

---
 xof/xof_test.go | 24 ++++++++++++++++++++++++
 1 file changed, 24 insertions(+)

diff --git a/xof/xof_test.go b/xof/xof_test.go
index e1e844ee1..01009e8f4 100644
--- 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)