fix lint

0xPolygonHermez · Sep 6, 2024 · 74fbd80 · 74fbd80
1 parent ef4c6c4
commit 74fbd80
Show file tree

Hide file tree

Showing 3 changed files with 17 additions and 3 deletions.
diff --git a/state/storage/sqlstorage/forkid_test.go b/state/storage/sqlstorage/forkid_test.go
@@ -88,7 +88,8 @@ func TestUpdateForkID(t *testing.T) {
 	err = storage.AddForkID(ctx, forkInterval, dbTx)
 	require.NoError(t, err)
 	forkInterval.ToBatchNumber = 3
-	storage.UpdateForkID(ctx, forkInterval, dbTx)
+	err = storage.UpdateForkID(ctx, forkInterval, dbTx)
+	require.NoError(t, err)
 	readForkIntervals, err := storage.GetForkIDs(ctx, dbTx)
 	require.NoError(t, err)
 	require.Len(t, readForkIntervals, 1)

diff --git a/state/storage/sqlstorage/l1infotree_test.go b/state/storage/sqlstorage/l1infotree_test.go
@@ -168,7 +168,8 @@ func TestL1InfoRootRemovedOnCascade(t *testing.T) {
 		err = storage.AddL1InfoTreeLeaf(ctx, &leaf, dbTx)
 		require.NoError(t, err)
 	}
-	storage.ResetToL1BlockNumber(ctx, 1, dbTx)
+	err = storage.ResetToL1BlockNumber(ctx, 1, dbTx)
+	require.NoError(t, err)
 	leafsRead, err := storage.GetAllL1InfoTreeLeaves(ctx, dbTx)
 	require.NoError(t, err)
 	require.NotNil(t, leafsRead)

diff --git a/state/storage/sqlstorage/sqlstorage.go b/state/storage/sqlstorage/sqlstorage.go
@@ -17,6 +17,10 @@ type SqlStorage struct {
 	db *sql.DB
 }
 
+const (
+	SqliteDriverName = "sqlite3"
+)
+
 func NewSqlStorage(cfg Config, runMigrations bool) (*SqlStorage, error) {
 	log.Infof("Running DB migrations")
 
@@ -33,7 +37,15 @@ func NewSqlStorage(cfg Config, runMigrations bool) (*SqlStorage, error) {
 			return nil, err
 		}
 	}
-	db.Exec("PRAGMA foreign_keys = ON;")
+	if cfg.DriverName == SqliteDriverName {
+		log.Debugf("Enabling foreign keys for sqlite database")
+		_, err = db.Exec("PRAGMA foreign_keys = ON;")
+		if err != nil {
+			err := fmt.Errorf("error enabling foreign keys: %w", err)
+			log.Errorf(err.Error())
+			return nil, err
+		}
+	}
 	return &SqlStorage{db}, nil
 }