fixup! fixup! feat: stmgr: add env to disable premigrations

filecoin-project · Feb 16, 2023 · 0e18e96 · 0e18e96
1 parent b379b47
commit 0e18e96
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/chain/stmgr/forks_test.go b/chain/stmgr/forks_test.go
@@ -543,9 +543,12 @@ func TestDisablePreMigration(t *testing.T) {
 	cg, err := gen.NewGenerator()
 	require.NoError(t, err)
 
-	os.Setenv(EnvDisablePreMigrations, "1")
+	err = os.Setenv(EnvDisablePreMigrations, "1")
+	require.NoError(t, err)
+
 	defer func() {
-		os.Unsetenv(EnvDisablePreMigrations)
+		err := os.Unsetenv(EnvDisablePreMigrations)
+		require.NoError(t, err)
 	}()
 
 	counter := make(chan struct{}, 10)
@@ -569,7 +572,6 @@ func TestDisablePreMigration(t *testing.T) {
 				PreMigration: func(ctx context.Context, _ *StateManager, _ MigrationCache,
 					_ cid.Cid, _ abi.ChainEpoch, _ *types.TipSet) error {
 					panic("should be skipped")
-					return nil
 				},
 			}}},
 		},