fix tests - update storer stub

multiversx · Mar 26, 2024 · a722602 · a722602
1 parent c289b13
commit a722602
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/testscommon/pruningStorerStub.go b/testscommon/pruningStorerStub.go
@@ -6,6 +6,7 @@ type PruningStorerStub struct {
 	PutCalled   func(key []byte, data []byte) error
 	PruneCalled func(index uint64) error
 	DumpCalled  func() error
+	CloseCalled func() error
 }
 
 // Get -
@@ -44,6 +45,15 @@ func (p *PruningStorerStub) Dump() error {
 	return nil
 }
 
+// Close -
+func (p *PruningStorerStub) Close() error {
+	if p.CloseCalled != nil {
+		return p.CloseCalled()
+	}
+
+	return nil
+}
+
 // IsInterfaceNil -
 func (p *PruningStorerStub) IsInterfaceNil() bool {
 	return p == nil