Refactor state tests (#2920)

* refactor state tests * refactor state tests
0xPolygonHermez · Dec 14, 2023 · 5e1c431 · 5e1c431
1 parent a2e6787
commit 5e1c431
Show file tree

Hide file tree

Showing 7 changed files with 246 additions and 166 deletions.
diff --git a/db/migrations/state/0013.sql b/db/migrations/state/0013.sql
@@ -31,8 +31,8 @@ ALTER TABLE state.transaction
 DROP INDEX IF EXISTS state.idx_transaction_l2_hash;
 
 ALTER TABLE state.batch
-    DROP COLUMN wip;
+    DROP COLUMN IF EXISTS wip;
 
 ALTER TABLE state.virtual_batch
-    DROP COLUMN timestamp_batch_etrog;
+    DROP COLUMN IF EXISTS timestamp_batch_etrog;
 
diff --git a/state/stateV2_test.go b/state/stateV2_test.go
diff --git a/state/test/common.go b/state/test/common.go
@@ -0,0 +1,12 @@
+package test
+
+import (
+	"github.com/0xPolygonHermez/zkevm-node/db"
+	"github.com/0xPolygonHermez/zkevm-node/test/dbutils"
+)
+
+func InitOrResetDB(cfg db.Config) {
+	if err := dbutils.InitOrResetState(cfg); err != nil {
+		panic(err)
+	}
+}
diff --git a/state/genesis_test.go → state/test/dragonfruit/genesis_test.go b/state/genesis_test.go → state/test/dragonfruit/genesis_test.go