fix: reset epoch timestamp when the reserve is reset (#4816)

ethersphere · Sep 11, 2024 · 38d8d2e · 38d8d2e
1 parent ab76c0b
commit 38d8d2e
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 1 deletion.
diff --git a/pkg/storer/internal/reserve/reserve.go b/pkg/storer/internal/reserve/reserve.go
@@ -518,9 +518,14 @@ func (r *Reserve) Reset(ctx context.Context) error {
 
 	size := r.Size()
 
+	err := r.st.Run(ctx, func(s transaction.Store) error { return s.IndexStore().Delete(&EpochItem{}) })
+	if err != nil {
+		return err
+	}
+
 	bRitems := make([]*BatchRadiusItem, 0, size)
 
-	err := r.st.IndexStore().Iterate(storage.Query{
+	err = r.st.IndexStore().Iterate(storage.Query{
 		Factory: func() storage.Item { return &BatchRadiusItem{} },
 	}, func(res storage.Result) (bool, error) {
 		bRitems = append(bRitems, res.Entry.(*BatchRadiusItem))

diff --git a/pkg/storer/internal/reserve/reserve_test.go b/pkg/storer/internal/reserve/reserve_test.go
@@ -824,6 +824,8 @@ func TestReset(t *testing.T) {
 	}
 	assert.Equal(t, c, total)
 
+	checkStore(t, ts.IndexStore(), &reserve.EpochItem{}, false)
+
 	err = r.Reset(context.Background())
 	if err != nil {
 		t.Fatal(err)
@@ -851,6 +853,8 @@ func TestReset(t *testing.T) {
 	}
 	assert.Equal(t, c, 0)
 
+	checkStore(t, ts.IndexStore(), &reserve.EpochItem{}, true)
+
 	for _, c := range chs {
 		h, err := c.Stamp().Hash()
 		if err != nil {