mirror of
https://source.quilibrium.com/quilibrium/ceremonyclient.git
synced 2024-11-15 04:35:17 +00:00
113 lines
3.5 KiB
Go
113 lines
3.5 KiB
Go
package pstoremanager_test
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/libp2p/go-libp2p/core/event"
|
|
"github.com/libp2p/go-libp2p/core/network"
|
|
"github.com/libp2p/go-libp2p/core/peer"
|
|
"github.com/libp2p/go-libp2p/p2p/host/eventbus"
|
|
"github.com/libp2p/go-libp2p/p2p/host/pstoremanager"
|
|
swarmt "github.com/libp2p/go-libp2p/p2p/net/swarm/testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
"go.uber.org/mock/gomock"
|
|
)
|
|
|
|
//go:generate sh -c "go run go.uber.org/mock/mockgen -package pstoremanager_test -destination mock_peerstore_test.go github.com/libp2p/go-libp2p/core/peerstore Peerstore"
|
|
|
|
func TestGracePeriod(t *testing.T) {
|
|
t.Parallel()
|
|
ctrl := gomock.NewController(t)
|
|
defer ctrl.Finish()
|
|
eventBus := eventbus.NewBus()
|
|
pstore := NewMockPeerstore(ctrl)
|
|
const gracePeriod = 250 * time.Millisecond
|
|
man, err := pstoremanager.NewPeerstoreManager(pstore, eventBus, swarmt.GenSwarm(t), pstoremanager.WithGracePeriod(gracePeriod))
|
|
require.NoError(t, err)
|
|
defer man.Close()
|
|
man.Start()
|
|
|
|
emitter, err := eventBus.Emitter(new(event.EvtPeerConnectednessChanged))
|
|
require.NoError(t, err)
|
|
start := time.Now()
|
|
removed := make(chan struct{})
|
|
pstore.EXPECT().RemovePeer(peer.ID("foobar")).DoAndReturn(func(p peer.ID) {
|
|
defer close(removed)
|
|
// make sure the call happened after the grace period
|
|
require.GreaterOrEqual(t, time.Since(start), gracePeriod)
|
|
require.LessOrEqual(t, time.Since(start), 3*gracePeriod)
|
|
})
|
|
require.NoError(t, emitter.Emit(event.EvtPeerConnectednessChanged{
|
|
Peer: "foobar",
|
|
Connectedness: network.NotConnected,
|
|
}))
|
|
<-removed
|
|
}
|
|
|
|
func TestReconnect(t *testing.T) {
|
|
t.Parallel()
|
|
ctrl := gomock.NewController(t)
|
|
eventBus := eventbus.NewBus()
|
|
pstore := NewMockPeerstore(ctrl)
|
|
const gracePeriod = 200 * time.Millisecond
|
|
man, err := pstoremanager.NewPeerstoreManager(pstore, eventBus, swarmt.GenSwarm(t), pstoremanager.WithGracePeriod(gracePeriod))
|
|
require.NoError(t, err)
|
|
defer man.Close()
|
|
man.Start()
|
|
|
|
emitter, err := eventBus.Emitter(new(event.EvtPeerConnectednessChanged))
|
|
require.NoError(t, err)
|
|
require.NoError(t, emitter.Emit(event.EvtPeerConnectednessChanged{
|
|
Peer: "foobar",
|
|
Connectedness: network.NotConnected,
|
|
}))
|
|
require.NoError(t, emitter.Emit(event.EvtPeerConnectednessChanged{
|
|
Peer: "foobar",
|
|
Connectedness: network.Connected,
|
|
}))
|
|
time.Sleep(gracePeriod * 3 / 2)
|
|
// There should have been no calls to RemovePeer.
|
|
ctrl.Finish()
|
|
}
|
|
|
|
func TestClose(t *testing.T) {
|
|
ctrl := gomock.NewController(t)
|
|
defer ctrl.Finish()
|
|
eventBus := eventbus.NewBus()
|
|
pstore := NewMockPeerstore(ctrl)
|
|
const gracePeriod = time.Hour
|
|
man, err := pstoremanager.NewPeerstoreManager(pstore, eventBus, swarmt.GenSwarm(t), pstoremanager.WithGracePeriod(gracePeriod))
|
|
require.NoError(t, err)
|
|
man.Start()
|
|
|
|
emitter, err := eventBus.Emitter(new(event.EvtPeerConnectednessChanged))
|
|
require.NoError(t, err)
|
|
|
|
sub, err := eventBus.Subscribe(&event.EvtPeerConnectednessChanged{})
|
|
require.NoError(t, err)
|
|
|
|
require.NoError(t, emitter.Emit(event.EvtPeerConnectednessChanged{
|
|
Peer: "foobar",
|
|
Connectedness: network.NotConnected,
|
|
}))
|
|
|
|
// make sure the event is sent before we close
|
|
select {
|
|
case <-sub.Out():
|
|
time.Sleep(100 * time.Millisecond) // make sure this event is also picked up by the pstoremanager
|
|
case <-time.After(5 * time.Second):
|
|
t.Fatalf("Hit timeout")
|
|
}
|
|
|
|
done := make(chan struct{})
|
|
pstore.EXPECT().RemovePeer(peer.ID("foobar")).Do(func(peer.ID) { close(done) })
|
|
require.NoError(t, man.Close())
|
|
select {
|
|
case <-done:
|
|
case <-time.After(5 * time.Second):
|
|
t.Fatalf("Hit timeout")
|
|
}
|
|
}
|