157 lines
4.2 KiB
Go
157 lines
4.2 KiB
Go
package main
|
|
|
|
import (
|
|
"os"
|
|
|
|
"github.com/hashicorp/go-multierror"
|
|
"github.com/urfave/cli/v2"
|
|
|
|
"github.com/filecoin-project/go-address"
|
|
|
|
abi_spec "github.com/filecoin-project/specs-actors/actors/abi"
|
|
big_spec "github.com/filecoin-project/specs-actors/actors/abi/big"
|
|
exitcode_spec "github.com/filecoin-project/specs-actors/actors/runtime/exitcode"
|
|
|
|
"github.com/filecoin-project/oni/tvx/chain"
|
|
"github.com/filecoin-project/oni/tvx/drivers"
|
|
)
|
|
|
|
var suiteMessagesCmd = &cli.Command{
|
|
Name: "suite-messages",
|
|
Description: "generate test vectors from the messages test suite adapted from github.com/filecoin-project/chain-validation",
|
|
Action: suiteMessages,
|
|
}
|
|
|
|
func suiteMessages(c *cli.Context) error {
|
|
var err *multierror.Error
|
|
err = multierror.Append(MessageTest_AccountActorCreation())
|
|
err = multierror.Append(MessageTest_InitActorSequentialIDAddressCreate())
|
|
err = multierror.Append(MessageTest_MultiSigActor())
|
|
err = multierror.Append(MessageTest_ValueTransferSimple())
|
|
err = multierror.Append(MessageTest_ValueTransferAdvance())
|
|
return err.ErrorOrNil()
|
|
}
|
|
|
|
func MessageTest_AccountActorCreation() error {
|
|
testCases := []struct {
|
|
desc string
|
|
|
|
existingActorType address.Protocol
|
|
existingActorBal abi_spec.TokenAmount
|
|
|
|
newActorAddr address.Address
|
|
newActorInitBal abi_spec.TokenAmount
|
|
|
|
expExitCode exitcode_spec.ExitCode
|
|
}{
|
|
{
|
|
"success create SECP256K1 account actor",
|
|
address.SECP256K1,
|
|
abi_spec.NewTokenAmount(10_000_000_000),
|
|
|
|
chain.MustNewSECP256K1Addr("publickeyfoo"),
|
|
abi_spec.NewTokenAmount(10_000),
|
|
|
|
exitcode_spec.Ok,
|
|
},
|
|
{
|
|
"success create BLS account actor",
|
|
address.SECP256K1,
|
|
abi_spec.NewTokenAmount(10_000_000_000),
|
|
|
|
chain.MustNewBLSAddr(1),
|
|
abi_spec.NewTokenAmount(10_000),
|
|
|
|
exitcode_spec.Ok,
|
|
},
|
|
{
|
|
"fail create SECP256K1 account actor insufficient balance",
|
|
address.SECP256K1,
|
|
abi_spec.NewTokenAmount(9_999),
|
|
|
|
chain.MustNewSECP256K1Addr("publickeybar"),
|
|
abi_spec.NewTokenAmount(10_000),
|
|
|
|
exitcode_spec.SysErrSenderStateInvalid,
|
|
},
|
|
{
|
|
"fail create BLS account actor insufficient balance",
|
|
address.SECP256K1,
|
|
abi_spec.NewTokenAmount(9_999),
|
|
|
|
chain.MustNewBLSAddr(1),
|
|
abi_spec.NewTokenAmount(10_000),
|
|
|
|
exitcode_spec.SysErrSenderStateInvalid,
|
|
},
|
|
// TODO add edge case tests that have insufficient balance after gas fees
|
|
}
|
|
for _, tc := range testCases {
|
|
err := func() error {
|
|
td := drivers.NewTestDriver()
|
|
td.Vector.Meta.Desc = tc.desc
|
|
|
|
existingAccountAddr, _ := td.NewAccountActor(tc.existingActorType, tc.existingActorBal)
|
|
|
|
td.UpdatePreStateRoot()
|
|
|
|
msg := td.MessageProducer.Transfer(existingAccountAddr, tc.newActorAddr, chain.Value(tc.newActorInitBal), chain.Nonce(0))
|
|
result := td.ApplyFailure(
|
|
msg,
|
|
tc.expExitCode,
|
|
)
|
|
|
|
// new actor balance will only exist if message was applied successfully.
|
|
if tc.expExitCode.IsSuccess() {
|
|
td.AssertBalance(tc.newActorAddr, tc.newActorInitBal)
|
|
td.AssertBalance(existingAccountAddr, big_spec.Sub(big_spec.Sub(tc.existingActorBal, result.Receipt.GasUsed.Big()), tc.newActorInitBal))
|
|
}
|
|
|
|
td.MustSerialize(os.Stdout)
|
|
|
|
return nil
|
|
}()
|
|
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func MessageTest_InitActorSequentialIDAddressCreate() error {
|
|
td := drivers.NewTestDriver()
|
|
|
|
var initialBal = abi_spec.NewTokenAmount(200_000_000_000)
|
|
var toSend = abi_spec.NewTokenAmount(10_000)
|
|
|
|
sender, _ := td.NewAccountActor(drivers.SECP, initialBal)
|
|
|
|
receiver, receiverID := td.NewAccountActor(drivers.SECP, initialBal)
|
|
|
|
firstPaychAddr := chain.MustNewIDAddr(chain.MustIDFromAddress(receiverID) + 1)
|
|
secondPaychAddr := chain.MustNewIDAddr(chain.MustIDFromAddress(receiverID) + 2)
|
|
|
|
firstInitRet := td.ComputeInitActorExecReturn(sender, 0, 0, firstPaychAddr)
|
|
secondInitRet := td.ComputeInitActorExecReturn(sender, 1, 0, secondPaychAddr)
|
|
|
|
td.UpdatePreStateRoot()
|
|
|
|
msg1 := td.MessageProducer.CreatePaymentChannelActor(sender, receiver, chain.Value(toSend), chain.Nonce(0))
|
|
td.ApplyExpect(
|
|
msg1,
|
|
chain.MustSerialize(&firstInitRet),
|
|
)
|
|
|
|
msg2 := td.MessageProducer.CreatePaymentChannelActor(sender, receiver, chain.Value(toSend), chain.Nonce(1))
|
|
td.ApplyExpect(
|
|
msg2,
|
|
chain.MustSerialize(&secondInitRet),
|
|
)
|
|
|
|
td.MustSerialize(os.Stdout)
|
|
|
|
return nil
|
|
}
|