Merge pull request #668 from sergeysova/upgrade-actix-web
upgrade actix to a v4.0 beta.10
This commit is contained in:
commit
3c6c93395d
@ -13,12 +13,12 @@ categories = ["network-programming", "asynchronous"]
|
|||||||
|
|
||||||
[dependencies]
|
[dependencies]
|
||||||
async-graphql = { path = "../..", version = "=2.11.0" }
|
async-graphql = { path = "../..", version = "=2.11.0" }
|
||||||
|
actix = "0.12.0"
|
||||||
actix = "0.12"
|
actix-http = "3.0.0-beta.11"
|
||||||
actix-http = "3.0.0-beta.10"
|
actix-web = { version = "4.0.0-beta.10", default-features = false }
|
||||||
actix-web = { version = "4.0.0-beta.9", default-features = false }
|
|
||||||
actix-web-actors = "4.0.0-beta.7"
|
actix-web-actors = "4.0.0-beta.7"
|
||||||
futures-util = { version = "0.3.13", default-features = false }
|
async-channel = "1.6.1"
|
||||||
|
futures-util = { version = "0.3.17", default-features = false }
|
||||||
serde_json = "1.0.64"
|
serde_json = "1.0.64"
|
||||||
serde_urlencoded = "0.7.0"
|
serde_urlencoded = "0.7.0"
|
||||||
futures-channel = "0.3.13"
|
futures-channel = "0.3.13"
|
||||||
|
@ -3,23 +3,22 @@
|
|||||||
#![allow(clippy::upper_case_acronyms)]
|
#![allow(clippy::upper_case_acronyms)]
|
||||||
#![warn(missing_docs)]
|
#![warn(missing_docs)]
|
||||||
|
|
||||||
mod subscription;
|
|
||||||
|
|
||||||
pub use subscription::WSSubscription;
|
|
||||||
|
|
||||||
use std::future::Future;
|
use std::future::Future;
|
||||||
use std::io::{self, ErrorKind};
|
use std::io::{self, ErrorKind};
|
||||||
use std::pin::Pin;
|
use std::pin::Pin;
|
||||||
|
|
||||||
|
use actix_http::error::PayloadError;
|
||||||
use actix_web::dev::{Payload, PayloadStream};
|
use actix_web::dev::{Payload, PayloadStream};
|
||||||
use actix_web::error::PayloadError;
|
|
||||||
use actix_web::http::{Method, StatusCode};
|
use actix_web::http::{Method, StatusCode};
|
||||||
use actix_web::{http, Error, FromRequest, HttpRequest, HttpResponse, Responder, Result};
|
use actix_web::{http, Error, FromRequest, HttpRequest, HttpResponse, Responder, Result};
|
||||||
|
use futures_util::future::{self, FutureExt};
|
||||||
|
use futures_util::{StreamExt, TryStreamExt};
|
||||||
|
|
||||||
use async_graphql::http::MultipartOptions;
|
use async_graphql::http::MultipartOptions;
|
||||||
use async_graphql::ParseRequestError;
|
use async_graphql::ParseRequestError;
|
||||||
use futures_channel::mpsc;
|
pub use subscription::WSSubscription;
|
||||||
use futures_util::future::{self, FutureExt};
|
|
||||||
use futures_util::{SinkExt, StreamExt, TryStreamExt};
|
mod subscription;
|
||||||
|
|
||||||
/// Extractor for GraphQL request.
|
/// Extractor for GraphQL request.
|
||||||
///
|
///
|
||||||
@ -40,7 +39,6 @@ type BatchToRequestMapper =
|
|||||||
impl FromRequest for Request {
|
impl FromRequest for Request {
|
||||||
type Error = Error;
|
type Error = Error;
|
||||||
type Future = future::Map<<BatchRequest as FromRequest>::Future, BatchToRequestMapper>;
|
type Future = future::Map<<BatchRequest as FromRequest>::Future, BatchToRequestMapper>;
|
||||||
type Config = MultipartOptions;
|
|
||||||
|
|
||||||
fn from_request(req: &HttpRequest, payload: &mut Payload<PayloadStream>) -> Self::Future {
|
fn from_request(req: &HttpRequest, payload: &mut Payload<PayloadStream>) -> Self::Future {
|
||||||
BatchRequest::from_request(req, payload).map(|res| {
|
BatchRequest::from_request(req, payload).map(|res| {
|
||||||
@ -69,10 +67,12 @@ impl BatchRequest {
|
|||||||
impl FromRequest for BatchRequest {
|
impl FromRequest for BatchRequest {
|
||||||
type Error = Error;
|
type Error = Error;
|
||||||
type Future = Pin<Box<dyn Future<Output = Result<BatchRequest>>>>;
|
type Future = Pin<Box<dyn Future<Output = Result<BatchRequest>>>>;
|
||||||
type Config = MultipartOptions;
|
|
||||||
|
|
||||||
fn from_request(req: &HttpRequest, payload: &mut Payload<PayloadStream>) -> Self::Future {
|
fn from_request(req: &HttpRequest, payload: &mut Payload<PayloadStream>) -> Self::Future {
|
||||||
let config = req.app_data::<Self::Config>().cloned().unwrap_or_default();
|
let config = req
|
||||||
|
.app_data::<MultipartOptions>()
|
||||||
|
.cloned()
|
||||||
|
.unwrap_or_default();
|
||||||
|
|
||||||
if req.method() == Method::GET {
|
if req.method() == Method::GET {
|
||||||
let res = serde_urlencoded::from_str(req.query_string());
|
let res = serde_urlencoded::from_str(req.query_string());
|
||||||
@ -84,7 +84,7 @@ impl FromRequest for BatchRequest {
|
|||||||
.and_then(|value| value.to_str().ok())
|
.and_then(|value| value.to_str().ok())
|
||||||
.map(|value| value.to_string());
|
.map(|value| value.to_string());
|
||||||
|
|
||||||
let (mut tx, rx) = mpsc::channel(16);
|
let (tx, rx) = async_channel::bounded(16);
|
||||||
|
|
||||||
// Payload is !Send so we create indirection with a channel
|
// Payload is !Send so we create indirection with a channel
|
||||||
let mut payload = payload.take();
|
let mut payload = payload.take();
|
||||||
@ -118,7 +118,7 @@ impl FromRequest for BatchRequest {
|
|||||||
}
|
}
|
||||||
PayloadError::Http2Payload(e) if e.is_io() => e.into_io().unwrap(),
|
PayloadError::Http2Payload(e) if e.is_io() => e.into_io().unwrap(),
|
||||||
PayloadError::Http2Payload(e) => io::Error::new(ErrorKind::Other, e),
|
PayloadError::Http2Payload(e) => io::Error::new(ErrorKind::Other, e),
|
||||||
_ => io::Error::new(ErrorKind::Other, "unknown error"),
|
_ => io::Error::new(ErrorKind::Other, e),
|
||||||
})
|
})
|
||||||
.into_async_read(),
|
.into_async_read(),
|
||||||
config,
|
config,
|
||||||
|
@ -3,20 +3,19 @@ use std::str::FromStr;
|
|||||||
use std::time::{Duration, Instant};
|
use std::time::{Duration, Instant};
|
||||||
|
|
||||||
use actix::{
|
use actix::{
|
||||||
Actor, ActorContext, ActorFutureExt, ActorStreamExt, AsyncContext, ContextFutureSpawner,
|
Actor, ActorContext, AsyncContext, ContextFutureSpawner, StreamHandler, WrapFuture, WrapStream,
|
||||||
StreamHandler, WrapFuture, WrapStream,
|
|
||||||
};
|
};
|
||||||
use actix_http::ws::Item;
|
use actix::{ActorFutureExt, ActorStreamExt};
|
||||||
use actix_web::error::{Error, PayloadError};
|
use actix_http::error::PayloadError;
|
||||||
use actix_web::web::{BufMut, Bytes, BytesMut};
|
use actix_http::ws;
|
||||||
|
use actix_web::web::Bytes;
|
||||||
use actix_web::{HttpRequest, HttpResponse};
|
use actix_web::{HttpRequest, HttpResponse};
|
||||||
use actix_web_actors::ws::{CloseReason, Message, ProtocolError, WebsocketContext};
|
use actix_web_actors::ws::{CloseReason, Message, ProtocolError, WebsocketContext};
|
||||||
use async_graphql::http::{WebSocket, WebSocketProtocols, WsMessage, ALL_WEBSOCKET_PROTOCOLS};
|
|
||||||
use async_graphql::{Data, ObjectType, Result, Schema, SubscriptionType};
|
|
||||||
use futures_channel::mpsc;
|
|
||||||
use futures_util::future::Ready;
|
use futures_util::future::Ready;
|
||||||
use futures_util::stream::Stream;
|
use futures_util::stream::Stream;
|
||||||
use futures_util::SinkExt;
|
|
||||||
|
use async_graphql::http::{WebSocket, WebSocketProtocols, WsMessage, ALL_WEBSOCKET_PROTOCOLS};
|
||||||
|
use async_graphql::{Data, ObjectType, Result, Schema, SubscriptionType};
|
||||||
|
|
||||||
const HEARTBEAT_INTERVAL: Duration = Duration::from_secs(5);
|
const HEARTBEAT_INTERVAL: Duration = Duration::from_secs(5);
|
||||||
const CLIENT_TIMEOUT: Duration = Duration::from_secs(10);
|
const CLIENT_TIMEOUT: Duration = Duration::from_secs(10);
|
||||||
@ -26,9 +25,9 @@ pub struct WSSubscription<Query, Mutation, Subscription, F> {
|
|||||||
schema: Schema<Query, Mutation, Subscription>,
|
schema: Schema<Query, Mutation, Subscription>,
|
||||||
protocol: WebSocketProtocols,
|
protocol: WebSocketProtocols,
|
||||||
last_heartbeat: Instant,
|
last_heartbeat: Instant,
|
||||||
messages: Option<mpsc::UnboundedSender<Bytes>>,
|
messages: Option<async_channel::Sender<Vec<u8>>>,
|
||||||
initializer: Option<F>,
|
initializer: Option<F>,
|
||||||
continuation: BytesMut,
|
continuation: Vec<u8>,
|
||||||
}
|
}
|
||||||
|
|
||||||
impl<Query, Mutation, Subscription>
|
impl<Query, Mutation, Subscription>
|
||||||
@ -43,7 +42,7 @@ where
|
|||||||
schema: Schema<Query, Mutation, Subscription>,
|
schema: Schema<Query, Mutation, Subscription>,
|
||||||
request: &HttpRequest,
|
request: &HttpRequest,
|
||||||
stream: T,
|
stream: T,
|
||||||
) -> Result<HttpResponse, Error>
|
) -> Result<HttpResponse, actix_web::error::Error>
|
||||||
where
|
where
|
||||||
T: Stream<Item = Result<Bytes, PayloadError>> + 'static,
|
T: Stream<Item = Result<Bytes, PayloadError>> + 'static,
|
||||||
{
|
{
|
||||||
@ -67,7 +66,7 @@ where
|
|||||||
request: &HttpRequest,
|
request: &HttpRequest,
|
||||||
stream: T,
|
stream: T,
|
||||||
initializer: F,
|
initializer: F,
|
||||||
) -> Result<HttpResponse, Error>
|
) -> Result<HttpResponse, actix_web::error::Error>
|
||||||
where
|
where
|
||||||
T: Stream<Item = Result<Bytes, PayloadError>> + 'static,
|
T: Stream<Item = Result<Bytes, PayloadError>> + 'static,
|
||||||
F: FnOnce(serde_json::Value) -> R + Unpin + Send + 'static,
|
F: FnOnce(serde_json::Value) -> R + Unpin + Send + 'static,
|
||||||
@ -96,7 +95,7 @@ where
|
|||||||
last_heartbeat: Instant::now(),
|
last_heartbeat: Instant::now(),
|
||||||
messages: None,
|
messages: None,
|
||||||
initializer: Some(initializer),
|
initializer: Some(initializer),
|
||||||
continuation: BytesMut::new(),
|
continuation: Vec::new(),
|
||||||
},
|
},
|
||||||
&ALL_WEBSOCKET_PROTOCOLS,
|
&ALL_WEBSOCKET_PROTOCOLS,
|
||||||
request,
|
request,
|
||||||
@ -127,7 +126,7 @@ where
|
|||||||
fn started(&mut self, ctx: &mut Self::Context) {
|
fn started(&mut self, ctx: &mut Self::Context) {
|
||||||
self.send_heartbeats(ctx);
|
self.send_heartbeats(ctx);
|
||||||
|
|
||||||
let (tx, rx) = mpsc::unbounded();
|
let (tx, rx) = async_channel::unbounded();
|
||||||
|
|
||||||
WebSocket::with_data(
|
WebSocket::with_data(
|
||||||
self.schema.clone(),
|
self.schema.clone(),
|
||||||
@ -179,22 +178,21 @@ where
|
|||||||
None
|
None
|
||||||
}
|
}
|
||||||
Message::Continuation(item) => match item {
|
Message::Continuation(item) => match item {
|
||||||
Item::FirstText(bytes) | Item::FirstBinary(bytes) => {
|
ws::Item::FirstText(bytes) | ws::Item::FirstBinary(bytes) => {
|
||||||
self.continuation.clear();
|
self.continuation = bytes.to_vec();
|
||||||
self.continuation.put(bytes);
|
|
||||||
None
|
None
|
||||||
}
|
}
|
||||||
Item::Continue(bytes) => {
|
ws::Item::Continue(bytes) => {
|
||||||
self.continuation.put(bytes);
|
self.continuation.extend_from_slice(&bytes);
|
||||||
None
|
None
|
||||||
}
|
}
|
||||||
Item::Last(bytes) => {
|
ws::Item::Last(bytes) => {
|
||||||
self.continuation.put(bytes);
|
self.continuation.extend_from_slice(&bytes);
|
||||||
Some(std::mem::take(&mut self.continuation).freeze())
|
Some(std::mem::take(&mut self.continuation))
|
||||||
}
|
}
|
||||||
},
|
},
|
||||||
Message::Text(s) => Some(s.into_bytes()),
|
Message::Text(s) => Some(s.into_bytes().to_vec()),
|
||||||
Message::Binary(bytes) => Some(bytes),
|
Message::Binary(bytes) => Some(bytes.to_vec()),
|
||||||
Message::Close(_) => {
|
Message::Close(_) => {
|
||||||
ctx.stop();
|
ctx.stop();
|
||||||
None
|
None
|
||||||
@ -203,7 +201,7 @@ where
|
|||||||
};
|
};
|
||||||
|
|
||||||
if let Some(message) = message {
|
if let Some(message) = message {
|
||||||
let mut sender = self.messages.as_ref().unwrap().clone();
|
let sender = self.messages.as_ref().unwrap().clone();
|
||||||
|
|
||||||
async move { sender.send(message).await }
|
async move { sender.send(message).await }
|
||||||
.into_actor(self)
|
.into_actor(self)
|
||||||
|
@ -1,14 +1,16 @@
|
|||||||
mod test_utils;
|
use actix_http::Method;
|
||||||
use actix_http::Request;
|
use actix_web::dev::{AnyBody, Service};
|
||||||
use actix_web::dev::{MessageBody, Service, ServiceResponse};
|
use actix_web::{guard, test, web, web::Data, App};
|
||||||
use actix_web::{guard, test, web, App};
|
|
||||||
use async_graphql::*;
|
|
||||||
use serde_json::json;
|
use serde_json::json;
|
||||||
|
|
||||||
|
use async_graphql::*;
|
||||||
use test_utils::*;
|
use test_utils::*;
|
||||||
|
|
||||||
|
mod test_utils;
|
||||||
|
|
||||||
#[actix_rt::test]
|
#[actix_rt::test]
|
||||||
async fn test_playground() {
|
async fn test_playground() {
|
||||||
let app = test::init_service(
|
let srv = test::init_service(
|
||||||
App::new().service(
|
App::new().service(
|
||||||
web::resource("/")
|
web::resource("/")
|
||||||
.guard(guard::Get())
|
.guard(guard::Get())
|
||||||
@ -16,20 +18,26 @@ async fn test_playground() {
|
|||||||
),
|
),
|
||||||
)
|
)
|
||||||
.await;
|
.await;
|
||||||
|
|
||||||
let req = test::TestRequest::with_uri("/").to_request();
|
let req = test::TestRequest::with_uri("/").to_request();
|
||||||
|
let response = srv.call(req).await.unwrap();
|
||||||
let resp = app.call(req).await.unwrap();
|
assert!(response.status().is_success());
|
||||||
assert!(resp.status().is_success());
|
let body = response.response().body();
|
||||||
let body = test::read_body(resp).await;
|
if let AnyBody::Bytes(bytes) = body {
|
||||||
assert!(std::str::from_utf8(&body).unwrap().contains("graphql"));
|
assert!(std::str::from_utf8(&bytes).unwrap().contains("graphql"));
|
||||||
|
} else {
|
||||||
|
panic!("response body must be Bytes {:?}", body);
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
#[actix_rt::test]
|
#[actix_rt::test]
|
||||||
async fn test_add() {
|
async fn test_add() {
|
||||||
let app = test::init_service(
|
let srv = test::init_service(
|
||||||
App::new()
|
App::new()
|
||||||
.data(Schema::new(AddQueryRoot, EmptyMutation, EmptySubscription))
|
.app_data(Data::new(Schema::new(
|
||||||
|
AddQueryRoot,
|
||||||
|
EmptyMutation,
|
||||||
|
EmptySubscription,
|
||||||
|
)))
|
||||||
.service(
|
.service(
|
||||||
web::resource("/")
|
web::resource("/")
|
||||||
.guard(guard::Post())
|
.guard(guard::Post())
|
||||||
@ -37,27 +45,32 @@ async fn test_add() {
|
|||||||
),
|
),
|
||||||
)
|
)
|
||||||
.await;
|
.await;
|
||||||
|
let response = srv
|
||||||
let resp = test::TestRequest::post()
|
.call(
|
||||||
.uri("/")
|
test::TestRequest::with_uri("/")
|
||||||
|
.method(Method::POST)
|
||||||
.set_payload(r#"{"query":"{ add(a: 10, b: 20) }"}"#)
|
.set_payload(r#"{"query":"{ add(a: 10, b: 20) }"}"#)
|
||||||
.send_request(&app)
|
.to_request(),
|
||||||
.await;
|
)
|
||||||
|
.await
|
||||||
assert!(resp.status().is_success());
|
.unwrap();
|
||||||
let body = test::read_body(resp).await;
|
assert!(response.status().is_success());
|
||||||
assert_eq!(body, json!({"data": {"add": 30}}).to_string());
|
let body = response.response().body();
|
||||||
|
assert_eq!(
|
||||||
|
body,
|
||||||
|
&AnyBody::Bytes(json!({"data": {"add": 30}}).to_string().into_bytes().into())
|
||||||
|
);
|
||||||
}
|
}
|
||||||
|
|
||||||
#[actix_rt::test]
|
#[actix_rt::test]
|
||||||
async fn test_hello() {
|
async fn test_hello() {
|
||||||
let app = test::init_service(
|
let srv = test::init_service(
|
||||||
App::new()
|
App::new()
|
||||||
.data(Schema::new(
|
.app_data(Data::new(Schema::new(
|
||||||
HelloQueryRoot,
|
HelloQueryRoot,
|
||||||
EmptyMutation,
|
EmptyMutation,
|
||||||
EmptySubscription,
|
EmptySubscription,
|
||||||
))
|
)))
|
||||||
.service(
|
.service(
|
||||||
web::resource("/")
|
web::resource("/")
|
||||||
.guard(guard::Post())
|
.guard(guard::Post())
|
||||||
@ -66,29 +79,37 @@ async fn test_hello() {
|
|||||||
)
|
)
|
||||||
.await;
|
.await;
|
||||||
|
|
||||||
let resp = test::TestRequest::post()
|
let response = srv
|
||||||
.uri("/")
|
.call(
|
||||||
|
test::TestRequest::with_uri("/")
|
||||||
|
.method(Method::POST)
|
||||||
.set_payload(r#"{"query":"{ hello }"}"#)
|
.set_payload(r#"{"query":"{ hello }"}"#)
|
||||||
.send_request(&app)
|
.to_request(),
|
||||||
.await;
|
)
|
||||||
|
.await
|
||||||
assert!(resp.status().is_success());
|
.unwrap();
|
||||||
let body = test::read_body(resp).await;
|
assert!(response.status().is_success());
|
||||||
|
let body = response.response().body();
|
||||||
assert_eq!(
|
assert_eq!(
|
||||||
body,
|
body,
|
||||||
json!({"data": {"hello": "Hello, world!"}}).to_string()
|
&AnyBody::Bytes(
|
||||||
|
json!({"data": {"hello": "Hello, world!"}})
|
||||||
|
.to_string()
|
||||||
|
.into_bytes()
|
||||||
|
.into()
|
||||||
|
)
|
||||||
);
|
);
|
||||||
}
|
}
|
||||||
|
|
||||||
#[actix_rt::test]
|
#[actix_rt::test]
|
||||||
async fn test_hello_header() {
|
async fn test_hello_header() {
|
||||||
let app = test::init_service(
|
let srv = test::init_service(
|
||||||
App::new()
|
App::new()
|
||||||
.data(Schema::new(
|
.app_data(Data::new(Schema::new(
|
||||||
HelloQueryRoot,
|
HelloQueryRoot,
|
||||||
EmptyMutation,
|
EmptyMutation,
|
||||||
EmptySubscription,
|
EmptySubscription,
|
||||||
))
|
)))
|
||||||
.service(
|
.service(
|
||||||
web::resource("/")
|
web::resource("/")
|
||||||
.guard(guard::Post())
|
.guard(guard::Post())
|
||||||
@ -97,27 +118,38 @@ async fn test_hello_header() {
|
|||||||
)
|
)
|
||||||
.await;
|
.await;
|
||||||
|
|
||||||
let resp = test::TestRequest::post()
|
let response = srv
|
||||||
.uri("/")
|
.call(
|
||||||
.append_header(("Name", "Foo"))
|
test::TestRequest::with_uri("/")
|
||||||
|
.method(Method::POST)
|
||||||
|
.insert_header(("Name", "Foo"))
|
||||||
.set_payload(r#"{"query":"{ hello }"}"#)
|
.set_payload(r#"{"query":"{ hello }"}"#)
|
||||||
.send_request(&app)
|
.to_request(),
|
||||||
.await;
|
)
|
||||||
|
.await
|
||||||
assert!(resp.status().is_success());
|
.unwrap();
|
||||||
let body = test::read_body(resp).await;
|
assert!(response.status().is_success());
|
||||||
assert_eq!(body, json!({"data": {"hello": "Hello, Foo!"}}).to_string());
|
let body = response.response().body();
|
||||||
|
assert_eq!(
|
||||||
|
body,
|
||||||
|
&AnyBody::Bytes(
|
||||||
|
json!({"data": {"hello": "Hello, Foo!"}})
|
||||||
|
.to_string()
|
||||||
|
.into_bytes()
|
||||||
|
.into()
|
||||||
|
)
|
||||||
|
);
|
||||||
}
|
}
|
||||||
|
|
||||||
#[actix_rt::test]
|
#[actix_rt::test]
|
||||||
async fn test_count() {
|
async fn test_count() {
|
||||||
let app = test::init_service(
|
let srv = test::init_service(
|
||||||
App::new()
|
App::new()
|
||||||
.data(
|
.app_data(Data::new(
|
||||||
Schema::build(CountQueryRoot, CountMutation, EmptySubscription)
|
Schema::build(CountQueryRoot, CountMutation, EmptySubscription)
|
||||||
.data(Count::default())
|
.data(Count::default())
|
||||||
.finish(),
|
.finish(),
|
||||||
)
|
))
|
||||||
.service(
|
.service(
|
||||||
web::resource("/")
|
web::resource("/")
|
||||||
.guard(guard::Post())
|
.guard(guard::Post())
|
||||||
@ -126,37 +158,87 @@ async fn test_count() {
|
|||||||
)
|
)
|
||||||
.await;
|
.await;
|
||||||
|
|
||||||
count_action_helper(0, r#"{"query":"{ count }"}"#, &app).await;
|
let response = srv
|
||||||
count_action_helper(10, r#"{"query":"mutation{ addCount(count: 10) }"}"#, &app).await;
|
.call(
|
||||||
count_action_helper(
|
test::TestRequest::with_uri("/")
|
||||||
8,
|
.method(Method::POST)
|
||||||
r#"{"query":"mutation{ subtractCount(count: 2) }"}"#,
|
.set_payload(r#"{"query":"{ count }"}"#)
|
||||||
&app,
|
.to_request(),
|
||||||
)
|
)
|
||||||
.await;
|
.await
|
||||||
count_action_helper(
|
.unwrap();
|
||||||
6,
|
assert!(response.status().is_success());
|
||||||
r#"{"query":"mutation{ subtractCount(count: 2) }"}"#,
|
let body = response.response().body();
|
||||||
&app,
|
assert_eq!(
|
||||||
|
body,
|
||||||
|
&AnyBody::Bytes(
|
||||||
|
json!({"data": {"count": 0}})
|
||||||
|
.to_string()
|
||||||
|
.into_bytes()
|
||||||
|
.into()
|
||||||
)
|
)
|
||||||
.await;
|
);
|
||||||
}
|
|
||||||
|
|
||||||
async fn count_action_helper<S, B, E>(expected: i32, payload: &'static str, app: &S)
|
let response = srv
|
||||||
where
|
.call(
|
||||||
S: Service<Request, Response = ServiceResponse<B>, Error = E>,
|
test::TestRequest::with_uri("/")
|
||||||
B: MessageBody + Unpin,
|
.method(Method::POST)
|
||||||
E: std::fmt::Debug,
|
.set_payload(r#"{"query":"mutation{ addCount(count: 10) }"}"#)
|
||||||
{
|
.to_request(),
|
||||||
let resp = test::TestRequest::post()
|
)
|
||||||
.uri("/")
|
.await
|
||||||
.set_payload(payload)
|
.unwrap();
|
||||||
.send_request(app)
|
assert!(response.status().is_success());
|
||||||
.await;
|
let body = response.response().body();
|
||||||
|
assert_eq!(
|
||||||
|
body,
|
||||||
|
&AnyBody::Bytes(
|
||||||
|
json!({"data": {"addCount": 10}})
|
||||||
|
.to_string()
|
||||||
|
.into_bytes()
|
||||||
|
.into()
|
||||||
|
)
|
||||||
|
);
|
||||||
|
|
||||||
assert!(resp.status().is_success());
|
let response = srv
|
||||||
let body = test::read_body(resp).await;
|
.call(
|
||||||
assert!(std::str::from_utf8(&body)
|
test::TestRequest::with_uri("/")
|
||||||
.unwrap()
|
.method(Method::POST)
|
||||||
.contains(&expected.to_string()));
|
.set_payload(r#"{"query":"mutation{ subtractCount(count: 2) }"}"#)
|
||||||
|
.to_request(),
|
||||||
|
)
|
||||||
|
.await
|
||||||
|
.unwrap();
|
||||||
|
assert!(response.status().is_success());
|
||||||
|
let body = response.response().body();
|
||||||
|
assert_eq!(
|
||||||
|
body,
|
||||||
|
&AnyBody::Bytes(
|
||||||
|
json!({"data": {"subtractCount": 8}})
|
||||||
|
.to_string()
|
||||||
|
.into_bytes()
|
||||||
|
.into()
|
||||||
|
)
|
||||||
|
);
|
||||||
|
|
||||||
|
let response = srv
|
||||||
|
.call(
|
||||||
|
test::TestRequest::with_uri("/")
|
||||||
|
.method(Method::POST)
|
||||||
|
.set_payload(r#"{"query":"mutation{ subtractCount(count: 2) }"}"#)
|
||||||
|
.to_request(),
|
||||||
|
)
|
||||||
|
.await
|
||||||
|
.unwrap();
|
||||||
|
assert!(response.status().is_success());
|
||||||
|
let body = response.response().body();
|
||||||
|
assert_eq!(
|
||||||
|
body,
|
||||||
|
&AnyBody::Bytes(
|
||||||
|
json!({"data": {"subtractCount": 6}})
|
||||||
|
.to_string()
|
||||||
|
.into_bytes()
|
||||||
|
.into()
|
||||||
|
)
|
||||||
|
);
|
||||||
}
|
}
|
Loading…
Reference in New Issue
Block a user