@@ -121,7 +121,6 @@ mod tests {
121
121
task:: { Context , Poll } ,
122
122
} ;
123
123
124
- use async_std:: task;
125
124
use futures:: { channel:: mpsc, prelude:: * } ;
126
125
127
126
use super :: RwStreamSink ;
@@ -165,58 +164,52 @@ mod tests {
165
164
}
166
165
}
167
166
168
- #[ test]
169
- fn basic_reading ( ) {
167
+ #[ tokio :: test]
168
+ async fn basic_reading ( ) {
170
169
let ( tx1, _) = mpsc:: channel :: < Vec < u8 > > ( 10 ) ;
171
170
let ( mut tx2, rx2) = mpsc:: channel ( 10 ) ;
172
171
173
172
let mut wrapper = RwStreamSink :: new ( Wrapper ( rx2. map ( Ok ) , tx1) ) ;
174
173
175
- task:: block_on ( async move {
176
- tx2. send ( Vec :: from ( "hel" ) ) . await . unwrap ( ) ;
177
- tx2. send ( Vec :: from ( "lo wor" ) ) . await . unwrap ( ) ;
178
- tx2. send ( Vec :: from ( "ld" ) ) . await . unwrap ( ) ;
179
- tx2. close ( ) . await . unwrap ( ) ;
174
+ tx2. send ( Vec :: from ( "hel" ) ) . await . unwrap ( ) ;
175
+ tx2. send ( Vec :: from ( "lo wor" ) ) . await . unwrap ( ) ;
176
+ tx2. send ( Vec :: from ( "ld" ) ) . await . unwrap ( ) ;
177
+ tx2. close ( ) . await . unwrap ( ) ;
180
178
181
- let mut data = Vec :: new ( ) ;
182
- wrapper. read_to_end ( & mut data) . await . unwrap ( ) ;
183
- assert_eq ! ( data, b"hello world" ) ;
184
- } )
179
+ let mut data = Vec :: new ( ) ;
180
+ wrapper. read_to_end ( & mut data) . await . unwrap ( ) ;
181
+ assert_eq ! ( data, b"hello world" ) ;
185
182
}
186
183
187
- #[ test]
188
- fn skip_empty_stream_items ( ) {
184
+ #[ tokio :: test]
185
+ async fn skip_empty_stream_items ( ) {
189
186
let data: Vec < & [ u8 ] > = vec ! [ b"" , b"foo" , b"" , b"bar" , b"" , b"baz" , b"" ] ;
190
187
let mut rws = RwStreamSink :: new ( stream:: iter ( data) . map ( Ok ) ) ;
191
188
let mut buf = [ 0 ; 9 ] ;
192
- task:: block_on ( async move {
193
- assert_eq ! ( 3 , rws. read( & mut buf) . await . unwrap( ) ) ;
194
- assert_eq ! ( 3 , rws. read( & mut buf[ 3 ..] ) . await . unwrap( ) ) ;
195
- assert_eq ! ( 3 , rws. read( & mut buf[ 6 ..] ) . await . unwrap( ) ) ;
196
- assert_eq ! ( 0 , rws. read( & mut buf) . await . unwrap( ) ) ;
197
- assert_eq ! ( b"foobarbaz" , & buf[ ..] )
198
- } )
189
+ assert_eq ! ( 3 , rws. read( & mut buf) . await . unwrap( ) ) ;
190
+ assert_eq ! ( 3 , rws. read( & mut buf[ 3 ..] ) . await . unwrap( ) ) ;
191
+ assert_eq ! ( 3 , rws. read( & mut buf[ 6 ..] ) . await . unwrap( ) ) ;
192
+ assert_eq ! ( 0 , rws. read( & mut buf) . await . unwrap( ) ) ;
193
+ assert_eq ! ( b"foobarbaz" , & buf[ ..] )
199
194
}
200
195
201
- #[ test]
202
- fn partial_read ( ) {
196
+ #[ tokio :: test]
197
+ async fn partial_read ( ) {
203
198
let data: Vec < & [ u8 ] > = vec ! [ b"hell" , b"o world" ] ;
204
199
let mut rws = RwStreamSink :: new ( stream:: iter ( data) . map ( Ok ) ) ;
205
200
let mut buf = [ 0 ; 3 ] ;
206
- task:: block_on ( async move {
207
- assert_eq ! ( 3 , rws. read( & mut buf) . await . unwrap( ) ) ;
208
- assert_eq ! ( b"hel" , & buf[ ..3 ] ) ;
209
- assert_eq ! ( 0 , rws. read( & mut buf[ ..0 ] ) . await . unwrap( ) ) ;
210
- assert_eq ! ( 1 , rws. read( & mut buf) . await . unwrap( ) ) ;
211
- assert_eq ! ( b"l" , & buf[ ..1 ] ) ;
212
- assert_eq ! ( 3 , rws. read( & mut buf) . await . unwrap( ) ) ;
213
- assert_eq ! ( b"o w" , & buf[ ..3 ] ) ;
214
- assert_eq ! ( 0 , rws. read( & mut buf[ ..0 ] ) . await . unwrap( ) ) ;
215
- assert_eq ! ( 3 , rws. read( & mut buf) . await . unwrap( ) ) ;
216
- assert_eq ! ( b"orl" , & buf[ ..3 ] ) ;
217
- assert_eq ! ( 1 , rws. read( & mut buf) . await . unwrap( ) ) ;
218
- assert_eq ! ( b"d" , & buf[ ..1 ] ) ;
219
- assert_eq ! ( 0 , rws. read( & mut buf) . await . unwrap( ) ) ;
220
- } )
201
+ assert_eq ! ( 3 , rws. read( & mut buf) . await . unwrap( ) ) ;
202
+ assert_eq ! ( b"hel" , & buf[ ..3 ] ) ;
203
+ assert_eq ! ( 0 , rws. read( & mut buf[ ..0 ] ) . await . unwrap( ) ) ;
204
+ assert_eq ! ( 1 , rws. read( & mut buf) . await . unwrap( ) ) ;
205
+ assert_eq ! ( b"l" , & buf[ ..1 ] ) ;
206
+ assert_eq ! ( 3 , rws. read( & mut buf) . await . unwrap( ) ) ;
207
+ assert_eq ! ( b"o w" , & buf[ ..3 ] ) ;
208
+ assert_eq ! ( 0 , rws. read( & mut buf[ ..0 ] ) . await . unwrap( ) ) ;
209
+ assert_eq ! ( 3 , rws. read( & mut buf) . await . unwrap( ) ) ;
210
+ assert_eq ! ( b"orl" , & buf[ ..3 ] ) ;
211
+ assert_eq ! ( 1 , rws. read( & mut buf) . await . unwrap( ) ) ;
212
+ assert_eq ! ( b"d" , & buf[ ..1 ] ) ;
213
+ assert_eq ! ( 0 , rws. read( & mut buf) . await . unwrap( ) ) ;
221
214
}
222
215
}
0 commit comments