1
0
Fork 0
mirror of https://github.com/actix/actix-web.git synced 2024-12-17 13:46:36 +00:00
actix-web/actix-http/tests/test_client.rs

157 lines
5.7 KiB
Rust
Raw Permalink Normal View History

2021-06-17 16:57:58 +00:00
use std::convert::Infallible;
2021-12-08 06:01:11 +00:00
use actix_http::{body::BoxBody, HttpMessage, HttpService, Request, Response, StatusCode};
2019-12-12 17:08:38 +00:00
use actix_http_test::test_server;
use actix_service::ServiceFactoryExt;
2021-04-01 14:26:13 +00:00
use actix_utils::future;
use bytes::Bytes;
use derive_more::{Display, Error};
2021-04-01 14:26:13 +00:00
use futures_util::StreamExt as _;
2018-11-14 06:53:30 +00:00
const STR: &str = "Hello World Hello World Hello World Hello World Hello World \
Hello World Hello World Hello World Hello World Hello World \
Hello World Hello World Hello World Hello World Hello World \
Hello World Hello World Hello World Hello World Hello World \
Hello World Hello World Hello World Hello World Hello World \
Hello World Hello World Hello World Hello World Hello World \
Hello World Hello World Hello World Hello World Hello World \
Hello World Hello World Hello World Hello World Hello World \
Hello World Hello World Hello World Hello World Hello World \
Hello World Hello World Hello World Hello World Hello World \
Hello World Hello World Hello World Hello World Hello World \
Hello World Hello World Hello World Hello World Hello World \
Hello World Hello World Hello World Hello World Hello World \
Hello World Hello World Hello World Hello World Hello World \
Hello World Hello World Hello World Hello World Hello World \
Hello World Hello World Hello World Hello World Hello World \
Hello World Hello World Hello World Hello World Hello World \
Hello World Hello World Hello World Hello World Hello World \
Hello World Hello World Hello World Hello World Hello World \
Hello World Hello World Hello World Hello World Hello World \
Hello World Hello World Hello World Hello World Hello World";
2019-11-26 05:25:50 +00:00
#[actix_rt::test]
async fn test_h1_v2() {
2019-12-12 17:08:38 +00:00
let srv = test_server(move || {
2019-12-02 11:33:11 +00:00
HttpService::build()
2021-06-17 16:57:58 +00:00
.finish(|_| future::ok::<_, Infallible>(Response::ok().set_body(STR)))
2019-12-02 11:33:11 +00:00
.tcp()
2020-03-08 07:42:45 +00:00
})
.await;
2018-11-14 06:53:30 +00:00
2019-11-26 05:25:50 +00:00
let response = srv.get("/").send().await.unwrap();
assert!(response.status().is_success());
2018-11-14 06:53:30 +00:00
2021-01-15 02:11:10 +00:00
let request = srv.get("/").insert_header(("x-test", "111")).send();
2019-11-26 05:25:50 +00:00
let mut response = request.await.unwrap();
assert!(response.status().is_success());
2018-11-14 06:53:30 +00:00
2019-11-26 05:25:50 +00:00
// read response
let bytes = response.body().await.unwrap();
assert_eq!(bytes, Bytes::from_static(STR.as_ref()));
2018-11-14 06:53:30 +00:00
2019-11-26 05:25:50 +00:00
let mut response = srv.post("/").send().await.unwrap();
assert!(response.status().is_success());
2019-11-19 12:54:19 +00:00
2019-11-26 05:25:50 +00:00
// read response
let bytes = response.body().await.unwrap();
assert_eq!(bytes, Bytes::from_static(STR.as_ref()));
2018-11-14 06:53:30 +00:00
}
2019-11-26 05:25:50 +00:00
#[actix_rt::test]
async fn test_connection_close() {
2019-12-12 17:08:38 +00:00
let srv = test_server(move || {
2019-11-26 05:25:50 +00:00
HttpService::build()
2021-06-17 16:57:58 +00:00
.finish(|_| future::ok::<_, Infallible>(Response::ok().set_body(STR)))
2019-12-02 11:33:11 +00:00
.tcp()
2019-11-26 05:25:50 +00:00
.map(|_| ())
2020-03-08 07:42:45 +00:00
})
.await;
2019-11-19 12:54:19 +00:00
2019-11-26 05:25:50 +00:00
let response = srv.get("/").force_close().send().await.unwrap();
assert!(response.status().is_success());
2018-11-14 06:53:30 +00:00
}
2019-11-26 05:25:50 +00:00
#[actix_rt::test]
async fn test_with_query_parameter() {
2019-12-12 17:08:38 +00:00
let srv = test_server(move || {
2019-11-26 05:25:50 +00:00
HttpService::build()
2021-06-17 16:57:58 +00:00
.finish(|req: Request| async move {
2019-11-26 05:25:50 +00:00
if req.uri().query().unwrap().contains("qp=") {
2021-06-17 16:57:58 +00:00
Ok::<_, Infallible>(Response::ok())
2019-11-26 05:25:50 +00:00
} else {
2021-06-17 16:57:58 +00:00
Ok(Response::bad_request())
2019-11-26 05:25:50 +00:00
}
})
2019-12-02 11:33:11 +00:00
.tcp()
2019-11-26 05:25:50 +00:00
.map(|_| ())
2020-03-08 07:42:45 +00:00
})
.await;
2018-11-14 06:53:30 +00:00
2019-11-26 05:25:50 +00:00
let request = srv.request(http::Method::GET, srv.url("/?qp=5"));
let response = request.send().await.unwrap();
assert!(response.status().is_success());
2018-11-14 06:53:30 +00:00
}
#[derive(Debug, Display, Error)]
#[display(fmt = "expect failed")]
struct ExpectFailed;
2021-12-04 19:40:47 +00:00
impl From<ExpectFailed> for Response<BoxBody> {
2021-06-17 16:57:58 +00:00
fn from(_: ExpectFailed) -> Self {
Response::new(StatusCode::EXPECTATION_FAILED)
}
}
#[actix_rt::test]
async fn test_h1_expect() {
let srv = test_server(move || {
HttpService::build()
.expect(|req: Request| async {
if req.headers().contains_key("AUTH") {
Ok(req)
} else {
Err(ExpectFailed)
}
})
.h1(|req: Request| async move {
let (_, mut body) = req.into_parts();
let mut buf = Vec::new();
while let Some(Ok(chunk)) = body.next().await {
buf.extend_from_slice(&chunk);
}
let str = std::str::from_utf8(&buf).unwrap();
assert_eq!(str, "expect body");
2021-06-17 16:57:58 +00:00
Ok::<_, Infallible>(Response::ok())
})
.tcp()
})
.await;
// test expect without payload.
let request = srv
.request(http::Method::GET, srv.url("/"))
.insert_header(("Expect", "100-continue"));
let response = request.send().await;
assert!(response.is_err());
// test expect would fail to continue
let request = srv
.request(http::Method::GET, srv.url("/"))
.insert_header(("Expect", "100-continue"));
let response = request.send_body("expect body").await.unwrap();
assert_eq!(response.status(), StatusCode::EXPECTATION_FAILED);
// test expect would continue
let request = srv
.request(http::Method::GET, srv.url("/"))
.insert_header(("Expect", "100-continue"))
.insert_header(("AUTH", "996"));
let response = request.send_body("expect body").await.unwrap();
assert!(response.status().is_success());
}