109 lines
3.0 KiB
Go
109 lines
3.0 KiB
Go
package actors_test
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
cbor "github.com/ipfs/go-ipld-cbor"
|
|
"github.com/stretchr/testify/assert"
|
|
"go.opencensus.io/trace"
|
|
|
|
"github.com/filecoin-project/go-lotus/chain/actors"
|
|
"github.com/filecoin-project/go-lotus/chain/address"
|
|
"github.com/filecoin-project/go-lotus/chain/types"
|
|
"github.com/filecoin-project/go-lotus/chain/vm"
|
|
"github.com/filecoin-project/go-lotus/tracing"
|
|
)
|
|
|
|
func TestMultiSigCreate(t *testing.T) {
|
|
var creatorAddr, sig1Addr, sig2Addr, outsideAddr address.Address
|
|
opts := []HarnessOpt{
|
|
HarnessAddr(&creatorAddr, 10000),
|
|
HarnessAddr(&sig1Addr, 10000),
|
|
HarnessAddr(&sig2Addr, 10000),
|
|
HarnessAddr(&outsideAddr, 10000),
|
|
}
|
|
|
|
h := NewHarness2(t, opts...)
|
|
ret, _ := h.CreateActor(t, creatorAddr, actors.MultisigActorCodeCid,
|
|
actors.MultiSigConstructorParams{
|
|
Signers: []address.Address{creatorAddr, sig1Addr, sig2Addr},
|
|
Required: 2,
|
|
})
|
|
ApplyOK(t, ret)
|
|
}
|
|
|
|
func ApplyOK(t testing.TB, ret *vm.ApplyRet) {
|
|
t.Helper()
|
|
if ret.ExitCode != 0 {
|
|
t.Fatalf("exit code should be 0, got %d, actorErr: %+v", ret.ExitCode, ret.ActorErr)
|
|
}
|
|
}
|
|
|
|
func TestMultiSigOps(t *testing.T) {
|
|
je := tracing.SetupJaegerTracing("test")
|
|
defer je.Flush()
|
|
ctx, span := trace.StartSpan(context.Background(), "/test-"+t.Name())
|
|
defer span.End()
|
|
|
|
var creatorAddr, sig1Addr, sig2Addr, outsideAddr address.Address
|
|
var multSigAddr address.Address
|
|
opts := []HarnessOpt{
|
|
HarnessCtx(ctx),
|
|
HarnessAddr(&creatorAddr, 10000),
|
|
HarnessAddr(&sig1Addr, 10000),
|
|
HarnessAddr(&sig2Addr, 10000),
|
|
HarnessAddr(&outsideAddr, 1000),
|
|
HarnessActor(&multSigAddr, &creatorAddr, actors.MultisigActorCodeCid,
|
|
func() interface{} {
|
|
return actors.MultiSigConstructorParams{
|
|
Signers: []address.Address{creatorAddr, sig1Addr, sig2Addr},
|
|
Required: 2,
|
|
}
|
|
}),
|
|
}
|
|
|
|
curVal := types.NewInt(2000)
|
|
h := NewHarness2(t, opts...)
|
|
{
|
|
// Send funds into the multisig
|
|
ret, state := h.SendFunds(t, creatorAddr, multSigAddr, curVal)
|
|
ApplyOK(t, ret)
|
|
ms, err := state.GetActor(multSigAddr)
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, curVal, ms.Balance)
|
|
}
|
|
|
|
{
|
|
// Transfer funds outside of multsig
|
|
sendVal := types.NewInt(100)
|
|
ret, _ := h.Invoke(t, creatorAddr, multSigAddr, actors.MultiSigMethods.Propose,
|
|
actors.MultiSigProposeParams{
|
|
To: outsideAddr,
|
|
Value: sendVal,
|
|
})
|
|
ApplyOK(t, ret)
|
|
var txIDParam actors.MultiSigTxID
|
|
err := cbor.DecodeInto(ret.Return, &txIDParam.TxID)
|
|
assert.NoError(t, err, "decoding txid")
|
|
|
|
ret, _ = h.Invoke(t, outsideAddr, multSigAddr, actors.MultiSigMethods.Approve,
|
|
txIDParam)
|
|
assert.Equal(t, uint8(1), ret.ExitCode, "outsideAddr should not approve")
|
|
|
|
ret, state := h.Invoke(t, sig1Addr, multSigAddr, actors.MultiSigMethods.Approve,
|
|
txIDParam)
|
|
outAct, err := state.GetActor(outsideAddr)
|
|
ApplyOK(t, ret)
|
|
curVal = types.BigSub(curVal, sendVal)
|
|
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, types.NewInt(1099), outAct.Balance)
|
|
|
|
msAct, err := state.GetActor(multSigAddr)
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, curVal, msAct.Balance)
|
|
}
|
|
|
|
}
|