github.com/number571/tendermint@v0.34.11-gost/internal/mempool/ids_test.go (about)

     1  package mempool
     2  
     3  import (
     4  	"testing"
     5  
     6  	"github.com/number571/tendermint/types"
     7  	"github.com/stretchr/testify/require"
     8  )
     9  
    10  func TestMempoolIDsBasic(t *testing.T) {
    11  	ids := NewMempoolIDs()
    12  
    13  	peerID, err := types.NewNodeID("0011223344556677889900112233445566778899")
    14  	require.NoError(t, err)
    15  
    16  	ids.ReserveForPeer(peerID)
    17  	require.EqualValues(t, 1, ids.GetForPeer(peerID))
    18  	ids.Reclaim(peerID)
    19  
    20  	ids.ReserveForPeer(peerID)
    21  	require.EqualValues(t, 2, ids.GetForPeer(peerID))
    22  	ids.Reclaim(peerID)
    23  }