Files
tendermint/internal/mempool/ids_test.go
Sergio Mena d3548eb706 Completed the existing FinalizeBlock PR and rebased to master (#7798)
* Rebased and git-squashed the commits in PR #6546

migrate abci to finalizeBlock

work on abci, proxy and mempool

abciresponse, blok events, indexer, some tests

fix some tests

fix errors

fix errors in abci

fix tests amd errors

* Fixes after rebasing PR#6546

* Restored height to RequestFinalizeBlock & other

* Fixed more UTs

* Fixed kvstore

* More UT fixes

* last TC fixed

* make format

* Update internal/consensus/mempool_test.go

Co-authored-by: William Banfield <4561443+williambanfield@users.noreply.github.com>

* Addressed @williambanfield's comments

* Fixed UTs

* Addressed last comments from @williambanfield

* make format

Co-authored-by: marbar3778 <marbar3778@yahoo.com>
Co-authored-by: William Banfield <4561443+williambanfield@users.noreply.github.com>
2022-02-14 23:41:28 +01:00

90 lines
2.4 KiB
Go

package mempool
import (
"testing"
"github.com/stretchr/testify/require"
"github.com/tendermint/tendermint/types"
)
func TestMempoolIDsBasic(t *testing.T) {
ids := NewMempoolIDs()
peerID, err := types.NewNodeID("0011223344556677889900112233445566778899")
require.NoError(t, err)
require.EqualValues(t, 0, ids.GetForPeer(peerID))
ids.ReserveForPeer(peerID)
require.EqualValues(t, 1, ids.GetForPeer(peerID))
ids.Reclaim(peerID)
require.EqualValues(t, 0, ids.GetForPeer(peerID))
ids.ReserveForPeer(peerID)
require.EqualValues(t, 1, ids.GetForPeer(peerID))
}
func TestMempoolIDsPeerDupReserve(t *testing.T) {
ids := NewMempoolIDs()
peerID, err := types.NewNodeID("0011223344556677889900112233445566778899")
require.NoError(t, err)
require.EqualValues(t, 0, ids.GetForPeer(peerID))
ids.ReserveForPeer(peerID)
require.EqualValues(t, 1, ids.GetForPeer(peerID))
ids.ReserveForPeer(peerID)
require.EqualValues(t, 1, ids.GetForPeer(peerID))
}
func TestMempoolIDs2Peers(t *testing.T) {
ids := NewMempoolIDs()
peer1ID, _ := types.NewNodeID("0011223344556677889900112233445566778899")
require.EqualValues(t, 0, ids.GetForPeer(peer1ID))
ids.ReserveForPeer(peer1ID)
require.EqualValues(t, 1, ids.GetForPeer(peer1ID))
ids.Reclaim(peer1ID)
require.EqualValues(t, 0, ids.GetForPeer(peer1ID))
peer2ID, _ := types.NewNodeID("1011223344556677889900112233445566778899")
ids.ReserveForPeer(peer2ID)
require.EqualValues(t, 1, ids.GetForPeer(peer2ID))
ids.ReserveForPeer(peer1ID)
require.EqualValues(t, 2, ids.GetForPeer(peer1ID))
}
func TestMempoolIDsNextExistID(t *testing.T) {
ids := NewMempoolIDs()
peer1ID, _ := types.NewNodeID("0011223344556677889900112233445566778899")
ids.ReserveForPeer(peer1ID)
require.EqualValues(t, 1, ids.GetForPeer(peer1ID))
peer2ID, _ := types.NewNodeID("1011223344556677889900112233445566778899")
ids.ReserveForPeer(peer2ID)
require.EqualValues(t, 2, ids.GetForPeer(peer2ID))
peer3ID, _ := types.NewNodeID("2011223344556677889900112233445566778899")
ids.ReserveForPeer(peer3ID)
require.EqualValues(t, 3, ids.GetForPeer(peer3ID))
ids.Reclaim(peer1ID)
require.EqualValues(t, 0, ids.GetForPeer(peer1ID))
ids.Reclaim(peer3ID)
require.EqualValues(t, 0, ids.GetForPeer(peer3ID))
ids.ReserveForPeer(peer1ID)
require.EqualValues(t, 1, ids.GetForPeer(peer1ID))
ids.ReserveForPeer(peer3ID)
require.EqualValues(t, 3, ids.GetForPeer(peer3ID))
}