You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
118 lines
2.6 KiB
118 lines
2.6 KiB
package mempool
|
|
|
|
import (
|
|
"encoding/binary"
|
|
"sync"
|
|
"testing"
|
|
|
|
"github.com/tendermint/tendermint/proxy"
|
|
"github.com/tendermint/tendermint/types"
|
|
"github.com/tendermint/tmsp/example/golang"
|
|
tmsp "github.com/tendermint/tmsp/types"
|
|
)
|
|
|
|
func TestSerialReap(t *testing.T) {
|
|
|
|
app := example.NewCounterApplication(true)
|
|
app.SetOption("serial", "on")
|
|
mtx := new(sync.Mutex)
|
|
appConnMem := proxy.NewLocalAppConn(mtx, app)
|
|
appConnCon := proxy.NewLocalAppConn(mtx, app)
|
|
mempool := NewMempool(appConnMem)
|
|
|
|
appendTxsRange := func(start, end int) {
|
|
// Append some txs.
|
|
for i := start; i < end; i++ {
|
|
|
|
// This will succeed
|
|
txBytes := make([]byte, 32)
|
|
binary.LittleEndian.PutUint64(txBytes, uint64(i))
|
|
err := mempool.CheckTx(txBytes)
|
|
if err != nil {
|
|
t.Fatal("Error after CheckTx: %v", err)
|
|
}
|
|
|
|
// This will fail because not serial (incrementing)
|
|
// However, error should still be nil.
|
|
// It just won't show up on Reap().
|
|
err = mempool.CheckTx(txBytes)
|
|
if err != nil {
|
|
t.Fatal("Error after CheckTx: %v", err)
|
|
}
|
|
|
|
}
|
|
}
|
|
|
|
reapCheck := func(exp int) {
|
|
txs, err := mempool.Reap()
|
|
if err != nil {
|
|
t.Error("Error in mempool.Reap()", err)
|
|
}
|
|
if len(txs) != exp {
|
|
t.Fatalf("Expected to reap %v txs but got %v", exp, len(txs))
|
|
}
|
|
}
|
|
|
|
updateRange := func(start, end int) {
|
|
txs := make([]types.Tx, 0)
|
|
for i := start; i < end; i++ {
|
|
txBytes := make([]byte, 32)
|
|
binary.LittleEndian.PutUint64(txBytes, uint64(i))
|
|
txs = append(txs, txBytes)
|
|
}
|
|
err := mempool.Update(0, txs)
|
|
if err != nil {
|
|
t.Error("Error in mempool.Update()", err)
|
|
}
|
|
}
|
|
|
|
commitRange := func(start, end int) {
|
|
// Append some txs.
|
|
for i := start; i < end; i++ {
|
|
txBytes := make([]byte, 32)
|
|
binary.LittleEndian.PutUint64(txBytes, uint64(i))
|
|
_, retCode := appConnCon.AppendTx(txBytes)
|
|
if retCode != tmsp.RetCodeOK {
|
|
t.Error("Error committing tx", retCode)
|
|
}
|
|
}
|
|
_, retCode := appConnCon.GetHash()
|
|
if retCode != tmsp.RetCodeOK {
|
|
t.Error("Error committing range", retCode)
|
|
}
|
|
}
|
|
|
|
//----------------------------------------
|
|
|
|
// Append some txs.
|
|
appendTxsRange(0, 100)
|
|
|
|
// Reap the txs.
|
|
reapCheck(100)
|
|
|
|
// Reap again. We should get the same amount
|
|
reapCheck(100)
|
|
|
|
// Append 0 to 999, we should reap 900 new txs
|
|
// because 100 were already counted.
|
|
appendTxsRange(0, 1000)
|
|
|
|
// Reap the txs.
|
|
reapCheck(1000)
|
|
|
|
// Reap again. We should get the same amount
|
|
reapCheck(1000)
|
|
|
|
// Commit from the conensus AppConn
|
|
commitRange(0, 500)
|
|
updateRange(0, 500)
|
|
|
|
// We should have 500 left.
|
|
reapCheck(500)
|
|
|
|
// Append 100 invalid txs and 100 valid txs
|
|
appendTxsRange(900, 1100)
|
|
|
|
// We should have 600 now.
|
|
reapCheck(600)
|
|
}
|