core/vdf: unit tests for routines leakage

core/vdf/delayer_test.go

@@ -23,6 +23,7 @@ import (
 	"time"
 
 	"github.com/ethereum/go-ethereum/common"
+	"github.com/fortytw2/leaktest"
 )
 
 func TestOutput(t *testing.T) {
@@ -251,3 +252,23 @@ func TestRequestGo(t *testing.T) {
 	NoCLI()
 	TestRequest(t)
 }
+
+func TestLeakage(t *testing.T) {
+	defer leaktest.Check(t)()
+	delayer := NewDelayer(32)
+	for i := uint64(1); i < iteration0/1000; i++ {
+		output := delayer.Get(input0, i)
+		if output != nil {
+			if delayer.Verify(input0, output, i) {
+				t.Log("success", "output", common.Bytes2Hex(output))
+			} else {
+				t.Error("failed", "output", common.Bytes2Hex(output))
+			}
+		}
+	}
+}
+
+func TestLeakageGo(t *testing.T) {
+	NoCLI()
+	TestLeakage(t)
+}

core/vdf/engine_test.go

@@ -84,3 +84,24 @@ func TestInterruptedGeneratorGo(t *testing.T) {
 	NoCLI()
 	TestInterruptedGenerator(t)
 }
+
+func TestLeak(t *testing.T) {
+	for i := uint64(1); i < iteration0/300; i++ {
+		output, err := Instance().Generate(input0, i, 127, nil)
+		if err != nil {
+			t.Error("error", "err", err)
+		}
+		if output != nil {
+			if Instance().Verify(input0, output, i, 127) {
+				t.Log("success", "output", common.Bytes2Hex(output))
+			} else {
+				t.Error("failed", "output", common.Bytes2Hex(output))
+			}
+		}
+	}
+}
+
+func TestLeakGo(t *testing.T) {
+	NoCLI()
+	TestLeak(t)
+}