88 lines
1.6 KiB
Go
88 lines
1.6 KiB
Go
package network
|
|
|
|
import (
|
|
"testing"
|
|
|
|
ma "github.com/multiformats/go-multiaddr"
|
|
)
|
|
|
|
func TestListen(T *testing.T) {
|
|
var notifee NotifyBundle
|
|
addr, err := ma.NewMultiaddr("/ip4/127.0.0.1/udp/1234")
|
|
if err != nil {
|
|
T.Fatal("unexpected multiaddr error")
|
|
}
|
|
notifee.Listen(nil, addr)
|
|
|
|
called := false
|
|
notifee.ListenF = func(Network, ma.Multiaddr) {
|
|
called = true
|
|
}
|
|
if called {
|
|
T.Fatal("called should be false")
|
|
}
|
|
|
|
notifee.Listen(nil, addr)
|
|
if !called {
|
|
T.Fatal("Listen should have been called")
|
|
}
|
|
}
|
|
|
|
func TestListenClose(T *testing.T) {
|
|
var notifee NotifyBundle
|
|
addr, err := ma.NewMultiaddr("/ip4/127.0.0.1/udp/1234")
|
|
if err != nil {
|
|
T.Fatal("unexpected multiaddr error")
|
|
}
|
|
notifee.ListenClose(nil, addr)
|
|
|
|
called := false
|
|
notifee.ListenCloseF = func(Network, ma.Multiaddr) {
|
|
called = true
|
|
}
|
|
if called {
|
|
T.Fatal("called should be false")
|
|
}
|
|
|
|
notifee.ListenClose(nil, addr)
|
|
if !called {
|
|
T.Fatal("ListenClose should have been called")
|
|
}
|
|
}
|
|
|
|
func TestConnected(T *testing.T) {
|
|
var notifee NotifyBundle
|
|
notifee.Connected(nil, nil)
|
|
|
|
called := false
|
|
notifee.ConnectedF = func(Network, Conn) {
|
|
called = true
|
|
}
|
|
if called {
|
|
T.Fatal("called should be false")
|
|
}
|
|
|
|
notifee.Connected(nil, nil)
|
|
if !called {
|
|
T.Fatal("Connected should have been called")
|
|
}
|
|
}
|
|
|
|
func TestDisconnected(T *testing.T) {
|
|
var notifee NotifyBundle
|
|
notifee.Disconnected(nil, nil)
|
|
|
|
called := false
|
|
notifee.DisconnectedF = func(Network, Conn) {
|
|
called = true
|
|
}
|
|
if called {
|
|
T.Fatal("called should be false")
|
|
}
|
|
|
|
notifee.Disconnected(nil, nil)
|
|
if !called {
|
|
T.Fatal("Disconnected should have been called")
|
|
}
|
|
}
|