160 lines
3.2 KiB
Go
160 lines
3.2 KiB
Go
package actors_test
|
|
|
|
import (
|
|
"context"
|
|
"encoding/binary"
|
|
"fmt"
|
|
"testing"
|
|
|
|
"github.com/filecoin-project/lotus/build"
|
|
"github.com/filecoin-project/lotus/chain/gen/genesis"
|
|
|
|
"github.com/filecoin-project/go-address"
|
|
dstore "github.com/ipfs/go-datastore"
|
|
bstore "github.com/ipfs/go-ipfs-blockstore"
|
|
|
|
. "github.com/filecoin-project/lotus/chain/actors"
|
|
"github.com/filecoin-project/lotus/chain/store"
|
|
"github.com/filecoin-project/lotus/chain/types"
|
|
"github.com/filecoin-project/lotus/chain/vm"
|
|
)
|
|
|
|
func blsaddr(n uint64) address.Address {
|
|
buf := make([]byte, 48)
|
|
binary.PutUvarint(buf, n)
|
|
|
|
addr, err := address.NewBLSAddress(buf)
|
|
if err != nil {
|
|
panic(err) // ok
|
|
}
|
|
|
|
return addr
|
|
}
|
|
|
|
func setupVMTestEnv(t *testing.T) (*vm.VM, []address.Address, bstore.Blockstore) {
|
|
bs := bstore.NewBlockstore(dstore.NewMapDatastore())
|
|
|
|
from := blsaddr(0)
|
|
maddr := blsaddr(1)
|
|
|
|
actors := map[address.Address]types.BigInt{
|
|
from: types.NewInt(1000000),
|
|
maddr: types.NewInt(0),
|
|
}
|
|
st, err := genesis.MakeInitialStateTree(bs, actors)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
stateroot, err := st.Flush(context.TODO())
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
cs := store.NewChainStore(bs, nil, nil)
|
|
|
|
// TODO: should probabaly mock out the randomness bit, nil works for now
|
|
vm, err := vm.NewVM(stateroot, 1, nil, maddr, cs.Blockstore(), cs.VMSys())
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
return vm, []address.Address{from, maddr}, bs
|
|
}
|
|
|
|
func TestVMInvokeMethod(t *testing.T) {
|
|
vm, addrs, _ := setupVMTestEnv(t)
|
|
from := addrs[0]
|
|
|
|
var err error
|
|
cenc, err := SerializeParams(&StorageMinerConstructorParams{Owner: from, Worker: from})
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
execparams := &ExecParams{
|
|
Code: StorageMinerCodeCid,
|
|
Params: cenc,
|
|
}
|
|
enc, err := SerializeParams(execparams)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
msg := &types.Message{
|
|
To: InitAddress,
|
|
From: from,
|
|
Method: IAMethods.Exec,
|
|
Params: enc,
|
|
GasPrice: types.NewInt(1),
|
|
GasLimit: types.NewInt(10000),
|
|
Value: types.NewInt(0),
|
|
}
|
|
|
|
ret, err := vm.ApplyMessage(context.TODO(), msg)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if ret.ExitCode != 0 {
|
|
t.Fatal("invocation failed")
|
|
}
|
|
|
|
outaddr, err := address.NewFromBytes(ret.Return)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if outaddr.String() != "t0102" {
|
|
t.Fatal("hold up")
|
|
}
|
|
}
|
|
|
|
func TestStorageMarketActorCreateMiner(t *testing.T) {
|
|
vm, addrs, bs := setupVMTestEnv(t)
|
|
from := addrs[0]
|
|
maddr := addrs[1]
|
|
|
|
cheatStorageMarketTotal(t, vm, bs)
|
|
|
|
params := &StorageMinerConstructorParams{
|
|
Owner: maddr,
|
|
Worker: maddr,
|
|
SectorSize: build.SectorSizes[0],
|
|
PeerID: "fakepeerid",
|
|
}
|
|
var err error
|
|
enc, err := SerializeParams(params)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
msg := &types.Message{
|
|
To: StoragePowerAddress,
|
|
From: from,
|
|
Method: SPAMethods.CreateStorageMiner,
|
|
Params: enc,
|
|
GasPrice: types.NewInt(1),
|
|
GasLimit: types.NewInt(10000),
|
|
Value: types.NewInt(50000),
|
|
}
|
|
|
|
ret, err := vm.ApplyMessage(context.TODO(), msg)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if ret.ExitCode != 0 {
|
|
fmt.Println(ret.ActorErr)
|
|
t.Fatal("invocation failed: ", ret.ExitCode)
|
|
}
|
|
|
|
outaddr, err := address.NewFromBytes(ret.Return)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if outaddr.String() != "t0102" {
|
|
t.Fatal("hold up")
|
|
}
|
|
}
|