register the puppet actor conditionally.

This commit is contained in:
Raúl Kripalani 2020-08-21 11:51:50 +01:00
parent e38f4219c5
commit 6ed1c9e905

View File

@ -11,10 +11,11 @@ import (
"github.com/filecoin-project/sector-storage/ffiwrapper" "github.com/filecoin-project/sector-storage/ffiwrapper"
"github.com/filecoin-project/test-vectors/chaos"
"github.com/filecoin-project/lotus/chain/types" "github.com/filecoin-project/lotus/chain/types"
"github.com/filecoin-project/lotus/chain/vm" "github.com/filecoin-project/lotus/chain/vm"
"github.com/filecoin-project/lotus/lib/blockstore" "github.com/filecoin-project/lotus/lib/blockstore"
"github.com/filecoin-project/test-vectors/chaos"
) )
var ( var (
@ -49,10 +50,14 @@ func (d *Driver) ExecuteMessage(msg *types.Message, preroot cid.Cid, bs blocksto
return nil, cid.Undef, err return nil, cid.Undef, err
} }
// add support for the puppet and chaos actors.
invoker := vm.NewInvoker() invoker := vm.NewInvoker()
invoker.Register(puppet.PuppetActorCodeID, puppet.Actor{}, puppet.State{})
if chaosOn, ok := d.vector.Selector.Unpack()["chaos_actor"]; ok && chaosOn == "true" { // add support for the puppet and chaos actors.
selector := d.vector.Selector.Unpack()
if puppetOn, ok := selector["puppet_actor"]; ok && puppetOn == "true" {
invoker.Register(puppet.PuppetActorCodeID, puppet.Actor{}, puppet.State{})
}
if chaosOn, ok := selector["chaos_actor"]; ok && chaosOn == "true" {
invoker.Register(chaos.ChaosActorCodeCID, chaos.Actor{}, chaos.State{}) invoker.Register(chaos.ChaosActorCodeCID, chaos.Actor{}, chaos.State{})
} }