116 lines
2.9 KiB
Go
116 lines
2.9 KiB
Go
|
package identify
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/libp2p/go-libp2p-core/network"
|
||
|
|
"github.com/libp2p/go-libp2p-core/peer"
|
||
|
|
|
||
|
|
blhost "github.com/libp2p/go-libp2p-blankhost"
|
||
|
|
swarmt "github.com/libp2p/go-libp2p-swarm/testing"
|
||
|
|
pb "github.com/libp2p/go-libp2p/p2p/protocol/identify/pb"
|
||
|
|
|
||
|
|
"github.com/stretchr/testify/require"
|
||
|
|
)
|
||
|
|
|
||
|
|
func doeval(t *testing.T, ph *peerHandler, f func()) {
|
||
|
|
done := make(chan struct{}, 1)
|
||
|
|
ph.evalTestCh <- func() {
|
||
|
|
f()
|
||
|
|
done <- struct{}{}
|
||
|
|
}
|
||
|
|
<-done
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestMakeApplyDelta(t *testing.T) {
|
||
|
|
isTesting = true
|
||
|
|
ctx, cancel := context.WithCancel(context.Background())
|
||
|
|
defer cancel()
|
||
|
|
|
||
|
|
h1 := blhost.NewBlankHost(swarmt.GenSwarm(t, ctx))
|
||
|
|
defer h1.Close()
|
||
|
|
ids1 := NewIDService(h1)
|
||
|
|
ph := newPeerHandler(h1.ID(), ids1, &pb.Identify{})
|
||
|
|
ph.start()
|
||
|
|
defer ph.close()
|
||
|
|
|
||
|
|
m1 := ph.mkDelta()
|
||
|
|
require.NotNil(t, m1)
|
||
|
|
// all the Id protocols must have been added
|
||
|
|
require.NotEmpty(t, m1.AddedProtocols)
|
||
|
|
doeval(t, ph, func() {
|
||
|
|
ph.applyDelta(m1)
|
||
|
|
})
|
||
|
|
|
||
|
|
h1.SetStreamHandler("p1", func(network.Stream) {})
|
||
|
|
m2 := ph.mkDelta()
|
||
|
|
require.Len(t, m2.AddedProtocols, 1)
|
||
|
|
require.Contains(t, m2.AddedProtocols, "p1")
|
||
|
|
require.Empty(t, m2.RmProtocols)
|
||
|
|
doeval(t, ph, func() {
|
||
|
|
ph.applyDelta(m2)
|
||
|
|
})
|
||
|
|
|
||
|
|
h1.SetStreamHandler("p2", func(network.Stream) {})
|
||
|
|
h1.SetStreamHandler("p3", func(stream network.Stream) {})
|
||
|
|
m3 := ph.mkDelta()
|
||
|
|
require.Len(t, m3.AddedProtocols, 2)
|
||
|
|
require.Contains(t, m3.AddedProtocols, "p2")
|
||
|
|
require.Contains(t, m3.AddedProtocols, "p3")
|
||
|
|
require.Empty(t, m3.RmProtocols)
|
||
|
|
doeval(t, ph, func() {
|
||
|
|
ph.applyDelta(m3)
|
||
|
|
})
|
||
|
|
|
||
|
|
h1.RemoveStreamHandler("p3")
|
||
|
|
m4 := ph.mkDelta()
|
||
|
|
require.Empty(t, m4.AddedProtocols)
|
||
|
|
require.Len(t, m4.RmProtocols, 1)
|
||
|
|
require.Contains(t, m4.RmProtocols, "p3")
|
||
|
|
doeval(t, ph, func() {
|
||
|
|
ph.applyDelta(m4)
|
||
|
|
})
|
||
|
|
|
||
|
|
h1.RemoveStreamHandler("p2")
|
||
|
|
h1.RemoveStreamHandler("p1")
|
||
|
|
m5 := ph.mkDelta()
|
||
|
|
require.Empty(t, m5.AddedProtocols)
|
||
|
|
require.Len(t, m5.RmProtocols, 2)
|
||
|
|
require.Contains(t, m5.RmProtocols, "p2")
|
||
|
|
require.Contains(t, m5.RmProtocols, "p1")
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestHandlerClose(t *testing.T) {
|
||
|
|
isTesting = true
|
||
|
|
ctx, cancel := context.WithCancel(context.Background())
|
||
|
|
defer cancel()
|
||
|
|
|
||
|
|
h1 := blhost.NewBlankHost(swarmt.GenSwarm(t, ctx))
|
||
|
|
defer h1.Close()
|
||
|
|
ids1 := NewIDService(h1)
|
||
|
|
ph := newPeerHandler(h1.ID(), ids1, nil)
|
||
|
|
ph.start()
|
||
|
|
|
||
|
|
require.NoError(t, ph.close())
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestPeerSupportsProto(t *testing.T) {
|
||
|
|
ctx, cancel := context.WithCancel(context.Background())
|
||
|
|
defer cancel()
|
||
|
|
|
||
|
|
h1 := blhost.NewBlankHost(swarmt.GenSwarm(t, ctx))
|
||
|
|
defer h1.Close()
|
||
|
|
ids1 := NewIDService(h1)
|
||
|
|
|
||
|
|
rp := peer.ID("test")
|
||
|
|
ph := newPeerHandler(rp, ids1, nil)
|
||
|
|
require.NoError(t, h1.Peerstore().AddProtocols(rp, "test"))
|
||
|
|
require.True(t, ph.peerSupportsProtos([]string{"test"}))
|
||
|
|
require.False(t, ph.peerSupportsProtos([]string{"random"}))
|
||
|
|
|
||
|
|
// remove support for protocol and check
|
||
|
|
require.NoError(t, h1.Peerstore().RemoveProtocols(rp, "test"))
|
||
|
|
require.False(t, ph.peerSupportsProtos([]string{"test"}))
|
||
|
|
}
|