lotus/miner/miner_test.go

105 lines
2.0 KiB
Go
Raw Normal View History

package miner
import (
"context"
"testing"
"github.com/filecoin-project/go-address"
"github.com/filecoin-project/lotus/chain/types"
)
func mustIDAddr(i uint64) address.Address {
a, err := address.NewIDAddress(i)
if err != nil {
panic(err)
}
return a
}
func TestMessageFiltering(t *testing.T) {
ctx := context.TODO()
a1 := mustIDAddr(1)
a2 := mustIDAddr(2)
actors := map[address.Address]*types.Actor{
a1: &types.Actor{
Nonce: 3,
Balance: types.NewInt(1200),
},
a2: &types.Actor{
Nonce: 1,
Balance: types.NewInt(1000),
},
}
af := func(ctx context.Context, addr address.Address, tsk types.TipSetKey) (*types.Actor, error) {
return actors[addr], nil
}
msgs := []types.Message{
types.Message{
From: a1,
2019-10-14 04:32:55 +00:00
To: a1,
Nonce: 3,
Value: types.NewInt(500),
GasLimit: types.NewInt(50),
GasPrice: types.NewInt(1),
},
types.Message{
From: a1,
2019-10-14 04:32:55 +00:00
To: a1,
Nonce: 4,
Value: types.NewInt(500),
GasLimit: types.NewInt(50),
GasPrice: types.NewInt(1),
},
types.Message{
From: a2,
2019-10-14 04:32:55 +00:00
To: a1,
Nonce: 1,
Value: types.NewInt(800),
GasLimit: types.NewInt(100),
GasPrice: types.NewInt(1),
},
types.Message{
From: a2,
2019-10-14 04:32:55 +00:00
To: a1,
Nonce: 0,
Value: types.NewInt(800),
GasLimit: types.NewInt(100),
GasPrice: types.NewInt(1),
},
types.Message{
From: a2,
2019-10-14 04:32:55 +00:00
To: a1,
Nonce: 2,
Value: types.NewInt(150),
GasLimit: types.NewInt(100),
GasPrice: types.NewInt(1),
},
}
2020-01-17 06:39:23 +00:00
outmsgs, err := SelectMessages(ctx, af, nil, wrapMsgs(msgs))
if err != nil {
t.Fatal(err)
}
if len(outmsgs) != 3 {
t.Fatal("filtering didnt work as expected")
}
m1 := outmsgs[2].Message
if m1.From != msgs[2].From || m1.Nonce != msgs[2].Nonce {
t.Fatal("filtering bad")
}
}
func wrapMsgs(msgs []types.Message) []*types.SignedMessage {
var out []*types.SignedMessage
for _, m := range msgs {
out = append(out, &types.SignedMessage{Message: m})
}
return out
}