@@ -124,14 +124,17 @@ impl<'a> RequestParams<'a> for &'a (dyn RequestExt + 'a) {
124
124
125
125
#[ cfg( test) ]
126
126
mod tests {
127
+ extern crate conduit_test;
128
+
127
129
use std:: io;
128
130
use std:: net:: SocketAddr ;
129
131
130
132
use { RequestParams , RouteBuilder } ;
131
133
134
+ use self :: conduit_test:: ResponseExt ;
132
135
use conduit:: {
133
- vec_to_body , Extensions , Handler , HeaderMap , Host , Method , Response , Scheme , StatusCode ,
134
- TypeMap , Version ,
136
+ Body , Extensions , Handler , HeaderMap , Host , Method , Response , Scheme , StatusCode , TypeMap ,
137
+ Version ,
135
138
} ;
136
139
137
140
struct RequestSentinel {
@@ -196,24 +199,20 @@ mod tests {
196
199
fn basic_get ( ) {
197
200
let router = test_router ( ) ;
198
201
let mut req = RequestSentinel :: new ( Method :: GET , "/posts/1" ) ;
199
- let mut res = router. call ( & mut req) . ok ( ) . expect ( "No response" ) ;
202
+ let res = router. call ( & mut req) . expect ( "No response" ) ;
200
203
201
204
assert_eq ! ( res. status( ) , StatusCode :: OK ) ;
202
- let mut s = Vec :: new ( ) ;
203
- res. body_mut ( ) . write_body ( & mut s) . unwrap ( ) ;
204
- assert_eq ! ( s, b"1, GET" ) ;
205
+ assert_eq ! ( res. into_cow( ) , "1, GET" . as_bytes( ) ) ;
205
206
}
206
207
207
208
#[ test]
208
209
fn basic_post ( ) {
209
210
let router = test_router ( ) ;
210
211
let mut req = RequestSentinel :: new ( Method :: POST , "/posts/10" ) ;
211
- let mut res = router. call ( & mut req) . ok ( ) . expect ( "No response" ) ;
212
+ let res = router. call ( & mut req) . expect ( "No response" ) ;
212
213
213
214
assert_eq ! ( res. status( ) , StatusCode :: OK ) ;
214
- let mut s = Vec :: new ( ) ;
215
- res. body_mut ( ) . write_body ( & mut s) . unwrap ( ) ;
216
- assert_eq ! ( s, b"10, POST" ) ;
215
+ assert_eq ! ( res. into_cow( ) , "10, POST" . as_bytes( ) ) ;
217
216
}
218
217
219
218
#[ test]
@@ -236,6 +235,6 @@ mod tests {
236
235
res. push ( format ! ( "{:?}" , req. method( ) ) ) ;
237
236
238
237
let bytes = res. join ( ", " ) . into_bytes ( ) ;
239
- Response :: builder ( ) . body ( vec_to_body ( bytes) )
238
+ Response :: builder ( ) . body ( Body :: from_vec ( bytes) )
240
239
}
241
240
}
0 commit comments