1
0
mirror of https://github.com/actix/actix-website synced 2024-11-30 19:14:36 +01:00
actix-website/examples/testing/src/main.rs

30 lines
849 B
Rust
Raw Normal View History

2019-06-17 21:39:58 +02:00
mod integration_one;
mod integration_three;
mod integration_two;
mod stream_response;
mod websockets;
// <unit-tests>
use actix_web::{http, test, HttpRequest, HttpResponse};
fn index(req: HttpRequest) -> HttpResponse {
if let Some(hdr) = req.headers().get(http::header::CONTENT_TYPE) {
if let Ok(_s) = hdr.to_str() {
return HttpResponse::Ok().into();
}
}
HttpResponse::BadRequest().into()
}
fn main() {
let req =
test::TestRequest::with_header("content-type", "text/plain").to_http_request();
let resp = test::block_on(index(req)).unwrap();
assert_eq!(resp.status(), http::StatusCode::OK);
let req = test::TestRequest::default().to_http_request();
let resp = test::block_on(index(req)).unwrap();
assert_eq!(resp.status(), http::StatusCode::BAD_REQUEST);
}
// </unit-tests>