132 lines
4.2 KiB
Go
132 lines
4.2 KiB
Go
package itests
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"github.com/filecoin-project/go-state-types/abi"
|
|
"github.com/filecoin-project/go-state-types/exitcode"
|
|
|
|
"github.com/filecoin-project/lotus/api"
|
|
"github.com/filecoin-project/lotus/chain/actors/builtin"
|
|
"github.com/filecoin-project/lotus/chain/types"
|
|
"github.com/filecoin-project/lotus/chain/types/ethtypes"
|
|
"github.com/filecoin-project/lotus/chain/wallet/key"
|
|
"github.com/filecoin-project/lotus/itests/kit"
|
|
)
|
|
|
|
// TestEthAccountAbstraction goes over the account abstraction workflow:
|
|
// - an embryo is created when it receives a message
|
|
// - the embryo turns into an EOA when it sends a message
|
|
func TestEthAccountAbstraction(t *testing.T) {
|
|
kit.QuietMiningLogs()
|
|
|
|
blockTime := 100 * time.Millisecond
|
|
client, _, ens := kit.EnsembleMinimal(t, kit.MockProofs(), kit.ThroughRPC())
|
|
ens.InterconnectAll().BeginMining(blockTime)
|
|
|
|
ctx, cancel := context.WithTimeout(context.Background(), time.Minute)
|
|
defer cancel()
|
|
|
|
secpKey, err := key.GenerateKey(types.KTDelegated)
|
|
require.NoError(t, err)
|
|
|
|
embryoAddress, err := client.WalletImport(ctx, &secpKey.KeyInfo)
|
|
require.NoError(t, err)
|
|
|
|
fmt.Println(embryoAddress)
|
|
|
|
// create an embryo actor at the target address
|
|
msgCreateEmbryo := &types.Message{
|
|
From: client.DefaultKey.Address,
|
|
To: embryoAddress,
|
|
Value: abi.TokenAmount(types.MustParseFIL("100")),
|
|
}
|
|
smCreateEmbryo, err := client.MpoolPushMessage(ctx, msgCreateEmbryo, nil)
|
|
require.NoError(t, err)
|
|
mLookup, err := client.StateWaitMsg(ctx, smCreateEmbryo.Cid(), 3, api.LookbackNoLimit, true)
|
|
require.NoError(t, err)
|
|
require.Equal(t, exitcode.Ok, mLookup.Receipt.ExitCode)
|
|
|
|
// confirm the embryo is an embryo
|
|
embryoActor, err := client.StateGetActor(ctx, embryoAddress, types.EmptyTSK)
|
|
require.NoError(t, err)
|
|
|
|
require.Equal(t, uint64(0), embryoActor.Nonce)
|
|
require.True(t, builtin.IsEmbryoActor(embryoActor.Code))
|
|
|
|
// send a message from the embryo address
|
|
msgFromEmbryo := &types.Message{
|
|
From: embryoAddress,
|
|
// self-send because an "eth tx payload" can't be to a filecoin address?
|
|
To: embryoAddress,
|
|
}
|
|
msgFromEmbryo, err = client.GasEstimateMessageGas(ctx, msgFromEmbryo, nil, types.EmptyTSK)
|
|
require.NoError(t, err)
|
|
|
|
txArgs, err := ethtypes.NewEthTxArgsFromMessage(msgFromEmbryo)
|
|
require.NoError(t, err)
|
|
|
|
digest, err := txArgs.OriginalRlpMsg()
|
|
require.NoError(t, err)
|
|
|
|
siggy, err := client.WalletSign(ctx, embryoAddress, digest)
|
|
require.NoError(t, err)
|
|
|
|
smFromEmbryoCid, err := client.MpoolPush(ctx, &types.SignedMessage{Message: *msgFromEmbryo, Signature: *siggy})
|
|
require.NoError(t, err)
|
|
|
|
mLookup, err = client.StateWaitMsg(ctx, smFromEmbryoCid, 3, api.LookbackNoLimit, true)
|
|
require.NoError(t, err)
|
|
require.Equal(t, exitcode.Ok, mLookup.Receipt.ExitCode)
|
|
|
|
// confirm ugly Embryo duckling has turned into a beautiful EthAccount swan
|
|
|
|
eoaActor, err := client.StateGetActor(ctx, embryoAddress, types.EmptyTSK)
|
|
require.NoError(t, err)
|
|
|
|
require.False(t, builtin.IsEmbryoActor(eoaActor.Code))
|
|
require.True(t, builtin.IsEthAccountActor(eoaActor.Code))
|
|
require.Equal(t, uint64(1), eoaActor.Nonce)
|
|
|
|
// Send another message, it should succeed without any code CID changes
|
|
|
|
msgFromEmbryo = &types.Message{
|
|
From: embryoAddress,
|
|
To: embryoAddress,
|
|
Nonce: 1,
|
|
}
|
|
|
|
msgFromEmbryo, err = client.GasEstimateMessageGas(ctx, msgFromEmbryo, nil, types.EmptyTSK)
|
|
require.NoError(t, err)
|
|
|
|
txArgs, err = ethtypes.NewEthTxArgsFromMessage(msgFromEmbryo)
|
|
require.NoError(t, err)
|
|
|
|
digest, err = txArgs.OriginalRlpMsg()
|
|
require.NoError(t, err)
|
|
|
|
siggy, err = client.WalletSign(ctx, embryoAddress, digest)
|
|
require.NoError(t, err)
|
|
|
|
smFromEmbryoCid, err = client.MpoolPush(ctx, &types.SignedMessage{Message: *msgFromEmbryo, Signature: *siggy})
|
|
require.NoError(t, err)
|
|
|
|
mLookup, err = client.StateWaitMsg(ctx, smFromEmbryoCid, 3, api.LookbackNoLimit, true)
|
|
require.NoError(t, err)
|
|
require.Equal(t, exitcode.Ok, mLookup.Receipt.ExitCode)
|
|
|
|
// confirm no changes in code CID
|
|
|
|
eoaActor, err = client.StateGetActor(ctx, embryoAddress, types.EmptyTSK)
|
|
require.NoError(t, err)
|
|
require.Equal(t, uint64(2), eoaActor.Nonce)
|
|
|
|
require.False(t, builtin.IsEmbryoActor(eoaActor.Code))
|
|
require.True(t, builtin.IsEthAccountActor(eoaActor.Code))
|
|
}
|