fix kurtosis test

0xPolygonHermez · Aug 30, 2024 · cc22e0d · cc22e0d
1 parent 446c2c0
commit cc22e0d
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/zk/datastream/server/datastream_populate.go b/zk/datastream/server/datastream_populate.go
@@ -2,6 +2,7 @@ package server
 
 import (
 	"context"
+	"errors"
 	"fmt"
 	"time"
 
@@ -365,7 +366,7 @@ func (srv *DataStreamServer) WriteGenesisToStream(
 	tx kv.Tx,
 ) error {
 	batchNo, err := reader.GetBatchNoByL2Block(0)
-	if err != nil {
+	if err != nil && !errors.Is(err, hermez_db.ErrorNotStored) {
 		return err
 	}
 

diff --git a/zk/stages/stage_batches.go b/zk/stages/stage_batches.go
@@ -132,7 +132,7 @@ func SpawnStageBatches(
 
 	// get batch for batches progress
 	stageProgressBatchNo, err := hermezDb.GetBatchNoByL2Block(stageProgressBlockNo)
-	if err != nil {
+	if err != nil && !errors.Is(err, hermez_db.ErrorNotStored) {
 		return fmt.Errorf("get batch no by l2 block error: %v", err)
 	}
 
@@ -523,7 +523,7 @@ func UnwindBatchesStage(u *stagedsync.UnwindState, tx kv.RwTx, cfg BatchesCfg, c
 	}
 
 	fromBatchPrev, err := hermezDb.GetBatchNoByL2Block(fromBlock - 1)
-	if err != nil {
+	if err != nil && !errors.Is(err, hermez_db.ErrorNotStored) {
 		return fmt.Errorf("get batch no by l2 block error: %v", err)
 	}
 	fromBatch, err := hermezDb.GetBatchNoByL2Block(fromBlock)