mirror of https://github.com/status-im/go-waku.git
57 lines
1.3 KiB
Go
57 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"
|
||
|
|
"github.com/waku-org/go-waku/waku/v2/utils"
|
||
|
|
)
|
||
|
|
|
||
|
|
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, utils.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, utils.ErrNoPeersAvailable)
|
||
|
|
_, err = slots.getPeers(protocol1).getRandom()
|
||
|
|
require.Equal(t, err, utils.ErrNoPeersAvailable)
|
||
|
|
}
|