dlss
/
FedP2P
4
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Use require in some places in TestUTPRawConn

This commit is contained in:
Matt Joiner 2017-09-01 15:09:10 +10:00
parent 110764480e
commit 158dab71c9
1 changed files with 6 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
client_test.go
View File

@ -167,9 +167,7 @@ func TestReducedDialTimeout(t *testing.T) {
func TestUTPRawConn(t *testing.T) {
l, err := NewUtpSocket("udp", "")
if err != nil {
t.Fatal(err)
}
require.NoError(t, err)
defer l.Close()
go func() {
for {
@ -180,17 +178,14 @@ func TestUTPRawConn(t *testing.T) {
}
}()
// Connect a UTP peer to see if the RawConn will still work.
s, _ := NewUtpSocket("udp", "")
s, err := NewUtpSocket("udp", "")
require.NoError(t, err)
defer s.Close()
utpPeer, err := s.Dial(fmt.Sprintf("localhost:%d", missinggo.AddrPort(l.Addr())))
if err != nil {
t.Fatalf("error dialing utp listener: %s", err)
}
require.NoError(t, err)
defer utpPeer.Close()
peer, err := net.ListenPacket("udp", ":0")
if err != nil {
t.Fatal(err)
}
require.NoError(t, err)
defer peer.Close()
msgsReceived := 0
@ -216,9 +211,7 @@ func TestUTPRawConn(t *testing.T) {
}
}()
udpAddr, err := net.ResolveUDPAddr("udp", fmt.Sprintf("localhost:%d", missinggo.AddrPort(l.Addr())))
if err != nil {
t.Fatal(err)
}
require.NoError(t, err)
for i := 0; i < N; i++ {
_, err := peer.WriteTo([]byte(fmt.Sprintf("%d", i)), udpAddr)
if err != nil {