mirror of https://github.com/status-im/go-waku.git
123 lines
3.2 KiB
Go
123 lines
3.2 KiB
Go
package node
|
|
|
|
import (
|
|
"context"
|
|
"net"
|
|
"sync"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestConnectionStatusChanges(t *testing.T) {
|
|
ctx, cancel := context.WithCancel(context.Background())
|
|
defer cancel()
|
|
|
|
connStatusChan := make(chan ConnStatus)
|
|
|
|
// Node1: Only Relay
|
|
hostAddr1, err := net.ResolveTCPAddr("tcp", "0.0.0.0:0")
|
|
require.NoError(t, err)
|
|
node1, err := New(ctx,
|
|
WithHostAddress([]*net.TCPAddr{hostAddr1}),
|
|
WithWakuRelay(),
|
|
WithConnStatusChan(connStatusChan),
|
|
)
|
|
require.NoError(t, err)
|
|
err = node1.Start()
|
|
require.NoError(t, err)
|
|
|
|
// Node2: Relay
|
|
hostAddr2, err := net.ResolveTCPAddr("tcp", "0.0.0.0:0")
|
|
require.NoError(t, err)
|
|
node2, err := New(ctx,
|
|
WithHostAddress([]*net.TCPAddr{hostAddr2}),
|
|
WithWakuRelay(),
|
|
)
|
|
err = node2.Start()
|
|
require.NoError(t, err)
|
|
|
|
// Node3: Relay + Store
|
|
hostAddr3, err := net.ResolveTCPAddr("tcp", "0.0.0.0:0")
|
|
require.NoError(t, err)
|
|
node3, err := New(ctx,
|
|
WithHostAddress([]*net.TCPAddr{hostAddr3}),
|
|
WithWakuRelay(),
|
|
WithWakuStore(false, false),
|
|
)
|
|
require.NoError(t, err)
|
|
err = node3.Start()
|
|
require.NoError(t, err)
|
|
|
|
node1.DialPeer(ctx, node2.ListenAddresses()[0].String())
|
|
node1.DialPeer(ctx, node3.ListenAddresses()[0].String())
|
|
|
|
var wg sync.WaitGroup
|
|
|
|
wg.Add(1)
|
|
go func() {
|
|
defer wg.Done()
|
|
|
|
connStatus := <-connStatusChan
|
|
_, ok := connStatus.Peers[node2.Host().ID()]
|
|
require.True(t, connStatus.IsOnline)
|
|
require.True(t, ok)
|
|
require.False(t, connStatus.HasHistory)
|
|
|
|
connStatus = <-connStatusChan
|
|
_, ok = connStatus.Peers[node3.Host().ID()]
|
|
require.True(t, connStatus.IsOnline)
|
|
require.True(t, ok)
|
|
require.True(t, connStatus.HasHistory)
|
|
}()
|
|
|
|
wg.Wait()
|
|
|
|
wg.Add(1)
|
|
go func() {
|
|
defer wg.Done()
|
|
|
|
connStatus := <-connStatusChan
|
|
_, ok := connStatus.Peers[node3.Host().ID()]
|
|
require.True(t, connStatus.IsOnline)
|
|
require.False(t, ok) // Peer3 should have been disconnected
|
|
require.False(t, connStatus.HasHistory) // No history, because there are no peers connected with store protocol
|
|
require.Len(t, node1.Host().Network().Peers(), 1) // No peers connected
|
|
}()
|
|
|
|
node3.Stop()
|
|
wg.Wait()
|
|
|
|
wg.Add(1)
|
|
go func() {
|
|
defer wg.Done()
|
|
|
|
connStatus := <-connStatusChan
|
|
_, ok := connStatus.Peers[node3.Host().ID()]
|
|
require.False(t, connStatus.IsOnline) // Peers are not connected. Should be offline
|
|
require.False(t, ok) // Peer2 should have been disconnected
|
|
require.False(t, connStatus.HasHistory) // No history, because there are no peers connected with store protocol
|
|
require.Len(t, node1.Host().Network().Peers(), 0) // No peers connected
|
|
}()
|
|
|
|
err = node1.ClosePeerById(node2.Host().ID())
|
|
require.NoError(t, err)
|
|
wg.Wait()
|
|
|
|
wg.Add(1)
|
|
go func() {
|
|
defer wg.Done()
|
|
|
|
connStatus := <-connStatusChan
|
|
_, ok := connStatus.Peers[node2.Host().ID()]
|
|
require.True(t, connStatus.IsOnline) // Peers2 is connected. Should be online
|
|
require.True(t, ok) // Peer2 should have been connected
|
|
require.False(t, connStatus.HasHistory) // No history because peer2 only has relay
|
|
require.Len(t, node1.Host().Network().Peers(), 1)
|
|
}()
|
|
|
|
err = node1.DialPeerByID(ctx, node2.Host().ID())
|
|
require.NoError(t, err)
|
|
wg.Wait()
|
|
}
|