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.
230 lines
4.4 KiB
230 lines
4.4 KiB
package v1
|
|
|
|
import (
|
|
"fmt"
|
|
"math/rand"
|
|
"sort"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
"github.com/tendermint/tendermint/internal/mempool"
|
|
)
|
|
|
|
func TestTxStore_GetTxBySender(t *testing.T) {
|
|
txs := NewTxStore()
|
|
wtx := &WrappedTx{
|
|
tx: []byte("test_tx"),
|
|
sender: "foo",
|
|
priority: 1,
|
|
timestamp: time.Now(),
|
|
}
|
|
|
|
res := txs.GetTxBySender(wtx.sender)
|
|
require.Nil(t, res)
|
|
|
|
txs.SetTx(wtx)
|
|
|
|
res = txs.GetTxBySender(wtx.sender)
|
|
require.NotNil(t, res)
|
|
require.Equal(t, wtx, res)
|
|
}
|
|
|
|
func TestTxStore_GetTxByHash(t *testing.T) {
|
|
txs := NewTxStore()
|
|
wtx := &WrappedTx{
|
|
tx: []byte("test_tx"),
|
|
sender: "foo",
|
|
priority: 1,
|
|
timestamp: time.Now(),
|
|
}
|
|
|
|
key := mempool.TxKey(wtx.tx)
|
|
res := txs.GetTxByHash(key)
|
|
require.Nil(t, res)
|
|
|
|
txs.SetTx(wtx)
|
|
|
|
res = txs.GetTxByHash(key)
|
|
require.NotNil(t, res)
|
|
require.Equal(t, wtx, res)
|
|
}
|
|
|
|
func TestTxStore_SetTx(t *testing.T) {
|
|
txs := NewTxStore()
|
|
wtx := &WrappedTx{
|
|
tx: []byte("test_tx"),
|
|
priority: 1,
|
|
timestamp: time.Now(),
|
|
}
|
|
|
|
key := mempool.TxKey(wtx.tx)
|
|
txs.SetTx(wtx)
|
|
|
|
res := txs.GetTxByHash(key)
|
|
require.NotNil(t, res)
|
|
require.Equal(t, wtx, res)
|
|
|
|
wtx.sender = "foo"
|
|
txs.SetTx(wtx)
|
|
|
|
res = txs.GetTxByHash(key)
|
|
require.NotNil(t, res)
|
|
require.Equal(t, wtx, res)
|
|
}
|
|
|
|
func TestTxStore_GetOrSetPeerByTxHash(t *testing.T) {
|
|
txs := NewTxStore()
|
|
wtx := &WrappedTx{
|
|
tx: []byte("test_tx"),
|
|
priority: 1,
|
|
timestamp: time.Now(),
|
|
}
|
|
|
|
key := mempool.TxKey(wtx.tx)
|
|
txs.SetTx(wtx)
|
|
|
|
res, ok := txs.GetOrSetPeerByTxHash(mempool.TxKey([]byte("test_tx_2")), 15)
|
|
require.Nil(t, res)
|
|
require.False(t, ok)
|
|
|
|
res, ok = txs.GetOrSetPeerByTxHash(key, 15)
|
|
require.NotNil(t, res)
|
|
require.False(t, ok)
|
|
|
|
res, ok = txs.GetOrSetPeerByTxHash(key, 15)
|
|
require.NotNil(t, res)
|
|
require.True(t, ok)
|
|
|
|
require.True(t, txs.TxHasPeer(key, 15))
|
|
require.False(t, txs.TxHasPeer(key, 16))
|
|
}
|
|
|
|
func TestTxStore_RemoveTx(t *testing.T) {
|
|
txs := NewTxStore()
|
|
wtx := &WrappedTx{
|
|
tx: []byte("test_tx"),
|
|
priority: 1,
|
|
timestamp: time.Now(),
|
|
}
|
|
|
|
txs.SetTx(wtx)
|
|
|
|
key := mempool.TxKey(wtx.tx)
|
|
res := txs.GetTxByHash(key)
|
|
require.NotNil(t, res)
|
|
|
|
txs.RemoveTx(res)
|
|
|
|
res = txs.GetTxByHash(key)
|
|
require.Nil(t, res)
|
|
}
|
|
|
|
func TestTxStore_Size(t *testing.T) {
|
|
txStore := NewTxStore()
|
|
numTxs := 1000
|
|
|
|
for i := 0; i < numTxs; i++ {
|
|
txStore.SetTx(&WrappedTx{
|
|
tx: []byte(fmt.Sprintf("test_tx_%d", i)),
|
|
priority: int64(i),
|
|
timestamp: time.Now(),
|
|
})
|
|
}
|
|
|
|
require.Equal(t, numTxs, txStore.Size())
|
|
}
|
|
|
|
func TestWrappedTxList_Reset(t *testing.T) {
|
|
list := NewWrappedTxList(func(wtx1, wtx2 *WrappedTx) bool {
|
|
return wtx1.height >= wtx2.height
|
|
})
|
|
|
|
require.Zero(t, list.Size())
|
|
|
|
for i := 0; i < 100; i++ {
|
|
list.Insert(&WrappedTx{height: int64(i)})
|
|
}
|
|
|
|
require.Equal(t, 100, list.Size())
|
|
|
|
list.Reset()
|
|
require.Zero(t, list.Size())
|
|
}
|
|
|
|
func TestWrappedTxList_Insert(t *testing.T) {
|
|
list := NewWrappedTxList(func(wtx1, wtx2 *WrappedTx) bool {
|
|
return wtx1.height >= wtx2.height
|
|
})
|
|
|
|
rng := rand.New(rand.NewSource(time.Now().UnixNano()))
|
|
|
|
var expected []int
|
|
for i := 0; i < 100; i++ {
|
|
height := rng.Int63n(10000)
|
|
expected = append(expected, int(height))
|
|
list.Insert(&WrappedTx{height: height})
|
|
|
|
if i%10 == 0 {
|
|
list.Insert(&WrappedTx{height: height})
|
|
expected = append(expected, int(height))
|
|
}
|
|
}
|
|
|
|
got := make([]int, list.Size())
|
|
for i, wtx := range list.txs {
|
|
got[i] = int(wtx.height)
|
|
}
|
|
|
|
sort.Ints(expected)
|
|
require.Equal(t, expected, got)
|
|
}
|
|
|
|
func TestWrappedTxList_Remove(t *testing.T) {
|
|
list := NewWrappedTxList(func(wtx1, wtx2 *WrappedTx) bool {
|
|
return wtx1.height >= wtx2.height
|
|
})
|
|
|
|
rng := rand.New(rand.NewSource(time.Now().UnixNano()))
|
|
|
|
var txs []*WrappedTx
|
|
for i := 0; i < 100; i++ {
|
|
height := rng.Int63n(10000)
|
|
tx := &WrappedTx{height: height}
|
|
|
|
txs = append(txs, tx)
|
|
list.Insert(tx)
|
|
|
|
if i%10 == 0 {
|
|
tx = &WrappedTx{height: height}
|
|
list.Insert(tx)
|
|
txs = append(txs, tx)
|
|
}
|
|
}
|
|
|
|
// remove a tx that does not exist
|
|
list.Remove(&WrappedTx{height: 20000})
|
|
|
|
// remove a tx that exists (by height) but not referenced
|
|
list.Remove(&WrappedTx{height: txs[0].height})
|
|
|
|
// remove a few existing txs
|
|
for i := 0; i < 25; i++ {
|
|
j := rng.Intn(len(txs))
|
|
list.Remove(txs[j])
|
|
txs = append(txs[:j], txs[j+1:]...)
|
|
}
|
|
|
|
expected := make([]int, len(txs))
|
|
for i, tx := range txs {
|
|
expected[i] = int(tx.height)
|
|
}
|
|
|
|
got := make([]int, list.Size())
|
|
for i, wtx := range list.txs {
|
|
got[i] = int(wtx.height)
|
|
}
|
|
|
|
sort.Ints(expected)
|
|
require.Equal(t, expected, got)
|
|
}
|