37 lines
916 B
Go
37 lines
916 B
Go
|
package policy
|
||
|
|
||
|
import (
|
||
|
"testing"
|
||
|
|
||
|
"github.com/stretchr/testify/require"
|
||
|
|
||
|
"github.com/filecoin-project/go-state-types/abi"
|
||
|
miner0 "github.com/filecoin-project/specs-actors/actors/builtin/miner"
|
||
|
)
|
||
|
|
||
|
func TestSupportedProofTypes(t *testing.T) {
|
||
|
var oldTypes []abi.RegisteredSealProof
|
||
|
for t := range miner0.SupportedProofTypes {
|
||
|
oldTypes = append(oldTypes, t)
|
||
|
}
|
||
|
t.Cleanup(func() {
|
||
|
SetSupportedProofTypes(oldTypes...)
|
||
|
})
|
||
|
|
||
|
SetSupportedProofTypes(abi.RegisteredSealProof_StackedDrg2KiBV1)
|
||
|
require.EqualValues(t,
|
||
|
miner0.SupportedProofTypes,
|
||
|
map[abi.RegisteredSealProof]struct{}{
|
||
|
abi.RegisteredSealProof_StackedDrg2KiBV1: {},
|
||
|
},
|
||
|
)
|
||
|
AddSupportedProofTypes(abi.RegisteredSealProof_StackedDrg8MiBV1)
|
||
|
require.EqualValues(t,
|
||
|
miner0.SupportedProofTypes,
|
||
|
map[abi.RegisteredSealProof]struct{}{
|
||
|
abi.RegisteredSealProof_StackedDrg2KiBV1: {},
|
||
|
abi.RegisteredSealProof_StackedDrg8MiBV1: {},
|
||
|
},
|
||
|
)
|
||
|
}
|