github.com/badrootd/celestia-core@v0.0.0-20240305091328-aa4207a4b25d/mempool/cat/peers_test.go (about)

     1  package cat
     2  
     3  import (
     4  	"testing"
     5  
     6  	"github.com/stretchr/testify/require"
     7  
     8  	"github.com/badrootd/celestia-core/p2p"
     9  	"github.com/badrootd/celestia-core/p2p/mocks"
    10  )
    11  
    12  func TestPeerLifecycle(t *testing.T) {
    13  	ids := newMempoolIDs()
    14  	peer1 := &mocks.Peer{}
    15  	peerID := p2p.ID("peer1")
    16  	peer1.On("ID").Return(peerID)
    17  
    18  	require.Nil(t, ids.GetPeer(1))
    19  	require.Zero(t, ids.GetIDForPeer(peerID))
    20  	require.Len(t, ids.GetAll(), 0)
    21  	ids.ReserveForPeer(peer1)
    22  
    23  	id := ids.GetIDForPeer(peerID)
    24  	require.Equal(t, uint16(1), id)
    25  	require.Equal(t, peer1, ids.GetPeer(id))
    26  	require.Len(t, ids.GetAll(), 1)
    27  
    28  	// duplicate peer should panic
    29  	require.Panics(t, func() {
    30  		ids.ReserveForPeer(peer1)
    31  	})
    32  
    33  	require.Equal(t, ids.Reclaim(peerID), id)
    34  	require.Nil(t, ids.GetPeer(id))
    35  	require.Zero(t, ids.GetIDForPeer(peerID))
    36  	require.Len(t, ids.GetAll(), 0)
    37  }