@@ -15,6 +15,7 @@ import (
1515 "github.com/pion/logging"
1616 "github.com/pion/transport/v3/dpipe"
1717 "github.com/pion/transport/v3/test"
18+ "github.com/stretchr/testify/assert"
1819)
1920
2021func TestSimpleReadWrite (t * testing.T ) {
@@ -25,51 +26,38 @@ func TestSimpleReadWrite(t *testing.T) {
2526
2627 ca , cb := dpipe .Pipe ()
2728 certificate , err := selfsign .GenerateSelfSigned ()
28- if err != nil {
29- t .Fatal (err )
30- }
29+ assert .NoError (t , err )
3130 gotHello := make (chan struct {})
3231
3332 go func () {
3433 server , sErr := testServer (ctx , dtlsnet .PacketConnFromConn (cb ), cb .RemoteAddr (), & Config {
3534 Certificates : []tls.Certificate {certificate },
3635 LoggerFactory : logging .NewDefaultLoggerFactory (),
3736 }, false )
38- if sErr != nil {
39- t .Error (sErr )
37+ assert .NoError (t , sErr )
4038
41- return
42- }
4339 buf := make ([]byte , 1024 )
44- if _ , sErr = server .Read (buf ); sErr != nil { //nolint:contextcheck
45- t . Error ( sErr )
46- }
40+ _ , sErr = server .Read (buf ) //nolint:contextcheck
41+ assert . NoError ( t , sErr )
42+
4743 gotHello <- struct {}{}
48- if sErr = server .Close (); sErr != nil { //nolint:contextcheck
49- t .Error (sErr )
50- }
44+ assert .NoError (t , server .Close ()) //nolint:contextcheck
5145 }()
5246
5347 client , err := testClient (ctx , dtlsnet .PacketConnFromConn (ca ), ca .RemoteAddr (), & Config {
5448 LoggerFactory : logging .NewDefaultLoggerFactory (),
5549 InsecureSkipVerify : true ,
5650 }, false )
57- if err != nil {
58- t .Fatal (err )
59- }
60- if _ , err = client .Write ([]byte ("hello" )); err != nil {
61- t .Error (err )
62- }
51+ assert .NoError (t , err )
52+ _ , err = client .Write ([]byte ("hello" ))
53+ assert .NoError (t , err )
6354 select {
6455 case <- gotHello :
6556 // OK
6657 case <- time .After (time .Second * 5 ):
67- t .Error ("timeout" )
68- }
69-
70- if err = client .Close (); err != nil {
71- t .Error (err )
58+ assert .Fail (t , "timeout" )
7259 }
60+ assert .NoError (t , client .Close ())
7361}
7462
7563func benchmarkConn (b * testing.B , payloadSize int64 ) {
@@ -80,43 +68,36 @@ func benchmarkConn(b *testing.B, payloadSize int64) {
8068
8169 ca , cb := dpipe .Pipe ()
8270 certificate , err := selfsign .GenerateSelfSigned ()
71+ assert .NoError (b , err )
8372 server := make (chan * Conn )
73+
8474 go func () {
8575 s , sErr := testServer (ctx , dtlsnet .PacketConnFromConn (cb ), cb .RemoteAddr (), & Config {
8676 Certificates : []tls.Certificate {certificate },
8777 }, false )
88- if err != nil {
89- b .Error (sErr )
78+ assert .NoError (b , sErr )
9079
91- return
92- }
9380 server <- s
9481 }()
95- if err != nil {
96- b .Fatal (err )
97- }
82+
9883 hw := make ([]byte , payloadSize )
9984 b .ReportAllocs ()
10085 b .SetBytes (int64 (len (hw )))
10186 go func () {
10287 client , cErr := testClient (
10388 ctx , dtlsnet .PacketConnFromConn (ca ), ca .RemoteAddr (), & Config {InsecureSkipVerify : true }, false ,
10489 )
105- if cErr != nil {
106- b .Error (err )
107- }
90+ assert .NoError (b , cErr )
10891 for {
109- if _ , cErr = client .Write (hw ); cErr != nil { //nolint:contextcheck
110- b .Error (err )
111- }
92+ _ , cErr = client .Write (hw ) //nolint:contextcheck
93+ assert .NoError (b , cErr )
11294 }
11395 }()
11496 s := <- server
11597 buf := make ([]byte , 2048 )
11698 for i := 0 ; i < b .N ; i ++ {
117- if _ , err = s .Read (buf ); err != nil {
118- b .Error (err )
119- }
99+ _ , err = s .Read (buf )
100+ assert .NoError (b , err )
120101 }
121102 })
122103}
0 commit comments