1
0
mirror of https://github.com/fafhrd91/actix-web synced 2024-11-27 17:52:56 +01:00

Run rustfmt

This commit is contained in:
Yuki Okushi 2020-05-21 17:56:53 +09:00
parent fe89ba7027
commit 6dd78d9355
No known key found for this signature in database
GPG Key ID: B0986C85C0E2DAA1
6 changed files with 45 additions and 40 deletions

View File

@ -474,13 +474,15 @@ where
mod tests { mod tests {
use actix_service::Service; use actix_service::Service;
use bytes::Bytes; use bytes::Bytes;
use futures_util::future::{ok, err}; use futures_util::future::{err, ok};
use super::*; use super::*;
use crate::http::{header, HeaderValue, Method, StatusCode}; use crate::http::{header, HeaderValue, Method, StatusCode};
use crate::middleware::DefaultHeaders; use crate::middleware::DefaultHeaders;
use crate::service::ServiceRequest; use crate::service::ServiceRequest;
use crate::test::{call_service, init_service, try_init_service, read_body, TestRequest}; use crate::test::{
call_service, init_service, read_body, try_init_service, TestRequest,
};
use crate::{web, HttpRequest, HttpResponse}; use crate::{web, HttpRequest, HttpResponse};
#[actix_rt::test] #[actix_rt::test]

View File

@ -273,13 +273,15 @@ impl<B> PinnedDrop for StreamLog<B> {
} }
} }
impl<B: MessageBody> MessageBody for StreamLog<B> { impl<B: MessageBody> MessageBody for StreamLog<B> {
fn size(&self) -> BodySize { fn size(&self) -> BodySize {
self.body.size() self.body.size()
} }
fn poll_next(self: Pin<&mut Self>, cx: &mut Context<'_>) -> Poll<Option<Result<Bytes, Error>>> { fn poll_next(
self: Pin<&mut Self>,
cx: &mut Context<'_>,
) -> Poll<Option<Result<Bytes, Error>>> {
let this = self.project(); let this = self.project();
match this.body.poll_next(cx) { match this.body.poll_next(cx) {
Poll::Ready(Some(Ok(chunk))) => { Poll::Ready(Some(Ok(chunk))) => {
@ -324,11 +326,13 @@ impl Format {
if let Some(key) = cap.get(2) { if let Some(key) = cap.get(2) {
results.push(match cap.get(3).unwrap().as_str() { results.push(match cap.get(3).unwrap().as_str() {
"a" => if key.as_str() == "r" { "a" => {
FormatText::RealIPRemoteAddr if key.as_str() == "r" {
} else { FormatText::RealIPRemoteAddr
unreachable!() } else {
}, unreachable!()
}
}
"i" => FormatText::RequestHeader( "i" => FormatText::RequestHeader(
HeaderName::try_from(key.as_str()).unwrap(), HeaderName::try_from(key.as_str()).unwrap(),
), ),
@ -481,7 +485,8 @@ impl FormatText {
*self = s; *self = s;
} }
FormatText::RealIPRemoteAddr => { FormatText::RealIPRemoteAddr => {
let s = if let Some(remote) = req.connection_info().realip_remote_addr() { let s = if let Some(remote) = req.connection_info().realip_remote_addr()
{
FormatText::Str(remote.to_string()) FormatText::Str(remote.to_string())
} else { } else {
FormatText::Str("-".to_string()) FormatText::Str("-".to_string())
@ -630,7 +635,9 @@ mod tests {
let req = TestRequest::with_header( let req = TestRequest::with_header(
header::FORWARDED, header::FORWARDED,
header::HeaderValue::from_static("for=192.0.2.60;proto=http;by=203.0.113.43"), header::HeaderValue::from_static(
"for=192.0.2.60;proto=http;by=203.0.113.43",
),
) )
.to_srv_request(); .to_srv_request();

View File

@ -177,7 +177,7 @@ mod tests {
assert!(res.status().is_success()); assert!(res.status().is_success());
} }
#[actix_rt::test] #[actix_rt::test]
async fn should_normalize_nothing_notrail() { async fn should_normalize_nothing_notrail() {
const URI: &str = "/v1/something"; const URI: &str = "/v1/something";

View File

@ -362,31 +362,23 @@ mod tests {
.service( .service(
web::resource("/test") web::resource("/test")
.route(web::get().to(|| HttpResponse::Ok())) .route(web::get().to(|| HttpResponse::Ok()))
.route(web::put().to(|| { .route(web::put().to(|| async {
async { Err::<HttpResponse, _>(error::ErrorBadRequest("err"))
Err::<HttpResponse, _>(error::ErrorBadRequest("err"))
}
})) }))
.route(web::post().to(|| { .route(web::post().to(|| async {
async { delay_for(Duration::from_millis(100)).await;
delay_for(Duration::from_millis(100)).await; HttpResponse::Created()
HttpResponse::Created()
}
})) }))
.route(web::delete().to(|| { .route(web::delete().to(|| async {
async { delay_for(Duration::from_millis(100)).await;
delay_for(Duration::from_millis(100)).await; Err::<HttpResponse, _>(error::ErrorBadRequest("err"))
Err::<HttpResponse, _>(error::ErrorBadRequest("err"))
}
})), })),
) )
.service(web::resource("/json").route(web::get().to(|| { .service(web::resource("/json").route(web::get().to(|| async {
async { delay_for(Duration::from_millis(25)).await;
delay_for(Duration::from_millis(25)).await; web::Json(MyObject {
web::Json(MyObject { name: "test".to_string(),
name: "test".to_string(), })
})
}
}))), }))),
) )
.await; .await;

View File

@ -89,7 +89,9 @@ where
>, >,
S::InitError: std::fmt::Debug, S::InitError: std::fmt::Debug,
{ {
try_init_service(app).await.expect("service initilization failed") try_init_service(app)
.await
.expect("service initilization failed")
} }
/// Fallible version of init_service that allows testing data factory errors. /// Fallible version of init_service that allows testing data factory errors.
@ -913,7 +915,8 @@ impl TestServerConfig {
/// Get first available unused address /// Get first available unused address
pub fn unused_addr() -> net::SocketAddr { pub fn unused_addr() -> net::SocketAddr {
let addr: net::SocketAddr = "127.0.0.1:0".parse().unwrap(); let addr: net::SocketAddr = "127.0.0.1:0".parse().unwrap();
let socket = Socket::new(Domain::ipv4(), Type::stream(), Some(Protocol::tcp())).unwrap(); let socket =
Socket::new(Domain::ipv4(), Type::stream(), Some(Protocol::tcp())).unwrap();
socket.bind(&addr.into()).unwrap(); socket.bind(&addr.into()).unwrap();
socket.set_reuse_address(true).unwrap(); socket.set_reuse_address(true).unwrap();
let tcp = socket.into_tcp_listener(); let tcp = socket.into_tcp_listener();

View File

@ -349,9 +349,10 @@ async fn test_body_br_streaming() {
#[actix_rt::test] #[actix_rt::test]
async fn test_head_binary() { async fn test_head_binary() {
let srv = test::start_with(test::config().h1(), || { let srv = test::start_with(test::config().h1(), || {
App::new().service(web::resource("/").route( App::new().service(
web::head().to(move || HttpResponse::Ok().body(STR)), web::resource("/")
)) .route(web::head().to(move || HttpResponse::Ok().body(STR))),
)
}); });
let mut response = srv.head("/").send().await.unwrap(); let mut response = srv.head("/").send().await.unwrap();