mirror of
https://github.com/status-im/go-waku.git
synced 2025-01-12 23:04:45 +00:00
6955d01498
* Update peer selection options for lightPush * Update peer selection options for filter * migrate peer selection functionality from peer manager Co-authored-by: richΛrd <info@richardramos.me> --------- Co-authored-by: richΛrd <info@richardramos.me>
56 lines
1.3 KiB
Go
56 lines
1.3 KiB
Go
package peermanager
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/libp2p/go-libp2p/core/peer"
|
|
libp2pProtocol "github.com/libp2p/go-libp2p/core/protocol"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestServiceSlot(t *testing.T) {
|
|
slots := NewServiceSlot()
|
|
|
|
protocol := libp2pProtocol.ID("test/protocol")
|
|
|
|
peerID := peer.ID("peerId")
|
|
|
|
//
|
|
slots.getPeers(protocol).add(peerID)
|
|
//
|
|
fetchedPeer, err := slots.getPeers(protocol).getRandom()
|
|
require.NoError(t, err)
|
|
require.Equal(t, peerID, fetchedPeer)
|
|
|
|
//
|
|
slots.getPeers(protocol).remove(peerID)
|
|
//
|
|
_, err = slots.getPeers(protocol).getRandom()
|
|
require.Equal(t, err, ErrNoPeersAvailable)
|
|
}
|
|
|
|
func TestServiceSlotRemovePeerFromAll(t *testing.T) {
|
|
slots := NewServiceSlot()
|
|
|
|
protocol := libp2pProtocol.ID("test/protocol")
|
|
protocol1 := libp2pProtocol.ID("test/protocol1")
|
|
|
|
peerID := peer.ID("peerId")
|
|
|
|
//
|
|
slots.getPeers(protocol).add(peerID)
|
|
slots.getPeers(protocol1).add(peerID)
|
|
//
|
|
fetchedPeer, err := slots.getPeers(protocol1).getRandom()
|
|
require.NoError(t, err)
|
|
require.Equal(t, peerID, fetchedPeer)
|
|
|
|
//
|
|
slots.removePeer(peerID)
|
|
//
|
|
_, err = slots.getPeers(protocol).getRandom()
|
|
require.Equal(t, err, ErrNoPeersAvailable)
|
|
_, err = slots.getPeers(protocol1).getRandom()
|
|
require.Equal(t, err, ErrNoPeersAvailable)
|
|
}
|