pull out test boilerplate

This commit is contained in:
whyrusleeping 2019-07-12 14:57:49 -07:00
parent c5e9a8a550
commit a5428d5c19

View File

@ -23,7 +23,7 @@ func blsaddr(n uint64) address.Address {
return addr return addr
} }
func TestVMInvokeMethod(t *testing.T) { func setupVMTestEnv(t *testing.T) (*VM, []address.Address) {
bs := bstore.NewBlockstore(dstore.NewMapDatastore()) bs := bstore.NewBlockstore(dstore.NewMapDatastore())
from := blsaddr(0) from := blsaddr(0)
@ -51,6 +51,12 @@ func TestVMInvokeMethod(t *testing.T) {
if err != nil { if err != nil {
t.Fatal(err) t.Fatal(err)
} }
return vm, []address.Address{from, maddr}
}
func TestVMInvokeMethod(t *testing.T) {
vm, addrs := setupVMTestEnv(t)
from := addrs[0]
execparams := &ExecParams{ execparams := &ExecParams{
Code: StorageMinerCodeCid, Code: StorageMinerCodeCid,
@ -91,35 +97,9 @@ func TestVMInvokeMethod(t *testing.T) {
} }
func TestStorageMarketActorCreateMiner(t *testing.T) { func TestStorageMarketActorCreateMiner(t *testing.T) {
////////// BOILERPLATE ////////// vm, addrs := setupVMTestEnv(t)
bs := bstore.NewBlockstore(dstore.NewMapDatastore()) from := addrs[0]
maddr := addrs[1]
from := blsaddr(0)
maddr := blsaddr(1)
actors := map[address.Address]types.BigInt{
from: types.NewInt(1000000),
maddr: types.NewInt(0),
}
st, err := MakeInitialStateTree(bs, actors)
if err != nil {
t.Fatal(err)
}
stateroot, err := st.Flush()
if err != nil {
t.Fatal(err)
}
cs := &ChainStore{
bs: bs,
}
vm, err := NewVM(stateroot, 1, maddr, cs)
if err != nil {
t.Fatal(err)
}
////// END BOILERPLATE //////////
params := &StorageMinerConstructorParams{ params := &StorageMinerConstructorParams{
Worker: maddr, Worker: maddr,