@@ -169,7 +169,7 @@ mod tests {
169
169
use net:: test:: { next_test_ip4, next_test_ip6} ;
170
170
use sync:: mpsc:: channel;
171
171
use sys_common:: AsInner ;
172
- use time:: Duration ;
172
+ use time:: { Instant , Duration } ;
173
173
use thread;
174
174
175
175
fn each_ip ( f : & mut FnMut ( SocketAddr , SocketAddr ) ) {
@@ -370,22 +370,22 @@ mod tests {
370
370
fn test_read_timeout ( ) {
371
371
let addr = next_test_ip4 ( ) ;
372
372
373
- let mut stream = t ! ( UdpSocket :: bind( & addr) ) ;
373
+ let stream = t ! ( UdpSocket :: bind( & addr) ) ;
374
374
t ! ( stream. set_read_timeout( Some ( Duration :: from_millis( 1000 ) ) ) ) ;
375
375
376
376
let mut buf = [ 0 ; 10 ] ;
377
- let wait = Duration :: span ( || {
378
- let kind = stream . recv_from ( & mut buf ) . err ( ) . expect ( "expected error" ) . kind ( ) ;
379
- assert ! ( kind == ErrorKind :: WouldBlock || kind == ErrorKind :: TimedOut ) ;
380
- } ) ;
381
- assert ! ( wait > Duration :: from_millis( 400 ) ) ;
377
+
378
+ let start = Instant :: now ( ) ;
379
+ let kind = stream . recv_from ( & mut buf ) . err ( ) . expect ( "expected error" ) . kind ( ) ;
380
+ assert ! ( kind == ErrorKind :: WouldBlock || kind == ErrorKind :: TimedOut ) ;
381
+ assert ! ( start . elapsed ( ) > Duration :: from_millis( 400 ) ) ;
382
382
}
383
383
384
384
#[ test]
385
385
fn test_read_with_timeout ( ) {
386
386
let addr = next_test_ip4 ( ) ;
387
387
388
- let mut stream = t ! ( UdpSocket :: bind( & addr) ) ;
388
+ let stream = t ! ( UdpSocket :: bind( & addr) ) ;
389
389
t ! ( stream. set_read_timeout( Some ( Duration :: from_millis( 1000 ) ) ) ) ;
390
390
391
391
t ! ( stream. send_to( b"hello world" , & addr) ) ;
@@ -394,10 +394,9 @@ mod tests {
394
394
t ! ( stream. recv_from( & mut buf) ) ;
395
395
assert_eq ! ( b"hello world" , & buf[ ..] ) ;
396
396
397
- let wait = Duration :: span ( || {
398
- let kind = stream. recv_from ( & mut buf) . err ( ) . expect ( "expected error" ) . kind ( ) ;
399
- assert ! ( kind == ErrorKind :: WouldBlock || kind == ErrorKind :: TimedOut ) ;
400
- } ) ;
401
- assert ! ( wait > Duration :: from_millis( 400 ) ) ;
397
+ let start = Instant :: now ( ) ;
398
+ let kind = stream. recv_from ( & mut buf) . err ( ) . expect ( "expected error" ) . kind ( ) ;
399
+ assert ! ( kind == ErrorKind :: WouldBlock || kind == ErrorKind :: TimedOut ) ;
400
+ assert ! ( start. elapsed( ) > Duration :: from_millis( 400 ) ) ;
402
401
}
403
402
}
0 commit comments