package state import ( "context" "fmt" "testing" "github.com/filecoin-project/specs-actors/actors/builtin" address "github.com/filecoin-project/go-address" "github.com/filecoin-project/lotus/chain/types" "github.com/ipfs/go-cid" cbor "github.com/ipfs/go-ipld-cbor" ) func BenchmarkStateTreeSet(b *testing.B) { cst := cbor.NewMemCborStore() st, err := NewStateTree(cst) if err != nil { b.Fatal(err) } b.ResetTimer() b.ReportAllocs() for i := 0; i < b.N; i++ { a, err := address.NewIDAddress(uint64(i)) if err != nil { b.Fatal(err) } err = st.SetActor(a, &types.Actor{ Balance: types.NewInt(1258812523), Code: builtin.StorageMinerActorCodeID, Head: builtin.AccountActorCodeID, Nonce: uint64(i), }) if err != nil { b.Fatal(err) } } } func BenchmarkStateTreeSetFlush(b *testing.B) { cst := cbor.NewMemCborStore() st, err := NewStateTree(cst) if err != nil { b.Fatal(err) } b.ResetTimer() b.ReportAllocs() for i := 0; i < b.N; i++ { a, err := address.NewIDAddress(uint64(i)) if err != nil { b.Fatal(err) } err = st.SetActor(a, &types.Actor{ Balance: types.NewInt(1258812523), Code: builtin.StorageMinerActorCodeID, Head: builtin.AccountActorCodeID, Nonce: uint64(i), }) if err != nil { b.Fatal(err) } if _, err := st.Flush(context.TODO()); err != nil { b.Fatal(err) } } } func BenchmarkStateTree10kGetActor(b *testing.B) { cst := cbor.NewMemCborStore() st, err := NewStateTree(cst) if err != nil { b.Fatal(err) } for i := 0; i < 10000; i++ { a, err := address.NewIDAddress(uint64(i)) if err != nil { b.Fatal(err) } err = st.SetActor(a, &types.Actor{ Balance: types.NewInt(1258812523 + uint64(i)), Code: builtin.StorageMinerActorCodeID, Head: builtin.AccountActorCodeID, Nonce: uint64(i), }) if err != nil { b.Fatal(err) } } if _, err := st.Flush(context.TODO()); err != nil { b.Fatal(err) } b.ResetTimer() b.ReportAllocs() for i := 0; i < b.N; i++ { a, err := address.NewIDAddress(uint64(i % 10000)) if err != nil { b.Fatal(err) } _, err = st.GetActor(a) if err != nil { b.Fatal(err) } } } func TestSetCache(t *testing.T) { cst := cbor.NewMemCborStore() st, err := NewStateTree(cst) if err != nil { t.Fatal(err) } a, err := address.NewIDAddress(uint64(222)) if err != nil { t.Fatal(err) } act := &types.Actor{ Balance: types.NewInt(0), Code: builtin.StorageMinerActorCodeID, Head: builtin.AccountActorCodeID, Nonce: 0, } err = st.SetActor(a, act) if err != nil { t.Fatal(err) } act.Nonce = 1 outact, err := st.GetActor(a) if err != nil { t.Fatal(err) } if outact.Nonce == 1 { t.Error("nonce should not have updated") } } func TestSnapshots(t *testing.T) { ctx := context.Background() cst := cbor.NewMemCborStore() st, err := NewStateTree(cst) if err != nil { t.Fatal(err) } var addrs []address.Address //for _, a := range []string{"t15ocrptbu4i5qucjvvwecihd7fqqgzb27pz5l5zy", "t1dpyvgavvl3f4ujlk6odedss54z6rt5gyuknsuva", "t1feiejbkcvozy7iltt2pxzuoq4d2kpbsusugan7a", "t3rgjfqybjx7bahuhfv7nwfg3tlm4i4zyvldfirjvzm5z5xwjoqbj3rfi2mpmlxpqwxxxafgpkjilqzpg7cefa"} { for _, a := range []string{"t0100", "t0101", "t0102", "t0103"} { addr, err := address.NewFromString(a) if err != nil { t.Fatal(err) } addrs = append(addrs, addr) } if err := st.Snapshot(ctx); err != nil { t.Fatal(err) } if err := st.SetActor(addrs[0], &types.Actor{Code: builtin.AccountActorCodeID, Head: builtin.AccountActorCodeID, Balance: types.NewInt(55)}); err != nil { t.Fatal(err) } { // sub call that will fail if err := st.Snapshot(ctx); err != nil { t.Fatal(err) } if err := st.SetActor(addrs[1], &types.Actor{Code: builtin.AccountActorCodeID, Head: builtin.AccountActorCodeID, Balance: types.NewInt(77)}); err != nil { t.Fatal(err) } if err := st.Revert(); err != nil { t.Fatal(err) } st.ClearSnapshot() } // more operations in top level call... if err := st.SetActor(addrs[2], &types.Actor{Code: builtin.AccountActorCodeID, Head: builtin.AccountActorCodeID, Balance: types.NewInt(123)}); err != nil { t.Fatal(err) } { // sub call that succeeds if err := st.Snapshot(ctx); err != nil { t.Fatal(err) } if err := st.SetActor(addrs[3], &types.Actor{Code: builtin.AccountActorCodeID, Head: builtin.AccountActorCodeID, Balance: types.NewInt(5)}); err != nil { t.Fatal(err) } st.ClearSnapshot() } st.ClearSnapshot() if _, err := st.Flush(ctx); err != nil { t.Fatal(err) } assertHas(t, st, addrs[0]) assertNotHas(t, st, addrs[1]) assertHas(t, st, addrs[2]) assertHas(t, st, addrs[3]) } func assertHas(t *testing.T, st *StateTree, addr address.Address) { _, err := st.GetActor(addr) if err != nil { t.Fatal(err) } } func assertNotHas(t *testing.T, st *StateTree, addr address.Address) { _, err := st.GetActor(addr) if err == nil { t.Fatal("shouldnt have found actor", addr) } } func TestStateTreeConsistency(t *testing.T) { cst := cbor.NewMemCborStore() st, err := NewStateTree(cst) if err != nil { t.Fatal(err) } var addrs []address.Address for i := 100; i < 150; i++ { a, err := address.NewIDAddress(uint64(i)) if err != nil { t.Fatal(err) } addrs = append(addrs, a) } randomCid, err := cid.Decode("bafy2bzacecu7n7wbtogznrtuuvf73dsz7wasgyneqasksdblxupnyovmtwxxu") if err != nil { t.Fatal(err) } for i, a := range addrs { err := st.SetActor(a, &types.Actor{ Code: randomCid, Head: randomCid, Balance: types.NewInt(uint64(10000 + i)), Nonce: uint64(1000 - i), }) if err != nil { t.Fatalf("while setting actor: %+v", err) } } root, err := st.Flush(context.TODO()) if err != nil { t.Fatal(err) } fmt.Println("root is: ", root) if root.String() != "bafy2bzaceadyjnrv3sbjvowfl3jr4pdn5p2bf3exjjie2f3shg4oy5sub7h34" { t.Fatal("MISMATCH!") } }