EnableLargeSectors: restore previous supported proof types.

itests/deals_512mb_test.go

@@ -15,7 +15,7 @@ func TestStorageDealMissingBlock(t *testing.T) {
 	ctx := context.Background()
 
 	// enable 512MiB proofs so we can conduct larger transfers.
-	kit.EnableLargeSectors()
+	kit.EnableLargeSectors(t)
 	kit.QuietMiningLogs()
 
 	client, miner, ens := kit.EnsembleMinimal(t,

itests/kit/sectors.go

@@ -1,6 +1,8 @@
 package kit
 
 import (
+	"testing"
+
 	"github.com/filecoin-project/go-state-types/abi"
 
 	"github.com/filecoin-project/lotus/chain/actors/policy"
@@ -8,9 +10,14 @@ import (
 
 // EnableLargeSectors enables 512MiB sectors. This is useful in combination with
 // mock proofs, for testing larger transfers.
-func EnableLargeSectors() {
+func EnableLargeSectors(t *testing.T) {
 	policy.SetSupportedProofTypes(
 		abi.RegisteredSealProof_StackedDrg2KiBV1,
-		abi.RegisteredSealProof_StackedDrg512MiBV1, // <==
+		abi.RegisteredSealProof_StackedDrg512MiBV1, // <== here
 	)
+	t.Cleanup(func() { // reset when done.
+		policy.SetSupportedProofTypes(
+			abi.RegisteredSealProof_StackedDrg2KiBV1,
+		)
+	})
 }