163 lines
4.4 KiB
Rust
163 lines
4.4 KiB
Rust
mod test_utils;
|
|
use actix_http::Request;
|
|
use actix_web::dev::{MessageBody, Service, ServiceResponse};
|
|
use actix_web::{guard, test, web, App};
|
|
use async_graphql::*;
|
|
use serde_json::json;
|
|
use test_utils::*;
|
|
|
|
#[actix_rt::test]
|
|
async fn test_playground() {
|
|
let app = test::init_service(
|
|
App::new().service(
|
|
web::resource("/")
|
|
.guard(guard::Get())
|
|
.to(test_utils::gql_playgound),
|
|
),
|
|
)
|
|
.await;
|
|
|
|
let req = test::TestRequest::with_uri("/").to_request();
|
|
|
|
let resp = app.call(req).await.unwrap();
|
|
assert!(resp.status().is_success());
|
|
let body = test::read_body(resp).await;
|
|
assert!(std::str::from_utf8(&body).unwrap().contains("graphql"));
|
|
}
|
|
|
|
#[actix_rt::test]
|
|
async fn test_add() {
|
|
let app = test::init_service(
|
|
App::new()
|
|
.data(Schema::new(AddQueryRoot, EmptyMutation, EmptySubscription))
|
|
.service(
|
|
web::resource("/")
|
|
.guard(guard::Post())
|
|
.to(gql_handle_schema::<AddQueryRoot, EmptyMutation, EmptySubscription>),
|
|
),
|
|
)
|
|
.await;
|
|
|
|
let resp = test::TestRequest::post()
|
|
.uri("/")
|
|
.set_payload(r#"{"query":"{ add(a: 10, b: 20) }"}"#)
|
|
.send_request(&app)
|
|
.await;
|
|
|
|
assert!(resp.status().is_success());
|
|
let body = test::read_body(resp).await;
|
|
assert_eq!(body, json!({"data": {"add": 30}}).to_string());
|
|
}
|
|
|
|
#[actix_rt::test]
|
|
async fn test_hello() {
|
|
let app = test::init_service(
|
|
App::new()
|
|
.data(Schema::new(
|
|
HelloQueryRoot,
|
|
EmptyMutation,
|
|
EmptySubscription,
|
|
))
|
|
.service(
|
|
web::resource("/")
|
|
.guard(guard::Post())
|
|
.to(gql_handle_schema::<HelloQueryRoot, EmptyMutation, EmptySubscription>),
|
|
),
|
|
)
|
|
.await;
|
|
|
|
let resp = test::TestRequest::post()
|
|
.uri("/")
|
|
.set_payload(r#"{"query":"{ hello }"}"#)
|
|
.send_request(&app)
|
|
.await;
|
|
|
|
assert!(resp.status().is_success());
|
|
let body = test::read_body(resp).await;
|
|
assert_eq!(
|
|
body,
|
|
json!({"data": {"hello": "Hello, world!"}}).to_string()
|
|
);
|
|
}
|
|
|
|
#[actix_rt::test]
|
|
async fn test_hello_header() {
|
|
let app = test::init_service(
|
|
App::new()
|
|
.data(Schema::new(
|
|
HelloQueryRoot,
|
|
EmptyMutation,
|
|
EmptySubscription,
|
|
))
|
|
.service(
|
|
web::resource("/")
|
|
.guard(guard::Post())
|
|
.to(gql_handle_schema_with_header::<HelloQueryRoot>),
|
|
),
|
|
)
|
|
.await;
|
|
|
|
let resp = test::TestRequest::post()
|
|
.uri("/")
|
|
.append_header(("Name", "Foo"))
|
|
.set_payload(r#"{"query":"{ hello }"}"#)
|
|
.send_request(&app)
|
|
.await;
|
|
|
|
assert!(resp.status().is_success());
|
|
let body = test::read_body(resp).await;
|
|
assert_eq!(body, json!({"data": {"hello": "Hello, Foo!"}}).to_string());
|
|
}
|
|
|
|
#[actix_rt::test]
|
|
async fn test_count() {
|
|
let app = test::init_service(
|
|
App::new()
|
|
.data(
|
|
Schema::build(CountQueryRoot, CountMutation, EmptySubscription)
|
|
.data(Count::default())
|
|
.finish(),
|
|
)
|
|
.service(
|
|
web::resource("/")
|
|
.guard(guard::Post())
|
|
.to(gql_handle_schema::<CountQueryRoot, CountMutation, EmptySubscription>),
|
|
),
|
|
)
|
|
.await;
|
|
|
|
count_action_helper(0, r#"{"query":"{ count }"}"#, &app).await;
|
|
count_action_helper(10, r#"{"query":"mutation{ addCount(count: 10) }"}"#, &app).await;
|
|
count_action_helper(
|
|
8,
|
|
r#"{"query":"mutation{ subtractCount(count: 2) }"}"#,
|
|
&app,
|
|
)
|
|
.await;
|
|
count_action_helper(
|
|
6,
|
|
r#"{"query":"mutation{ subtractCount(count: 2) }"}"#,
|
|
&app,
|
|
)
|
|
.await;
|
|
}
|
|
|
|
async fn count_action_helper<S, B, E>(expected: i32, payload: &'static str, app: &S)
|
|
where
|
|
S: Service<Request, Response = ServiceResponse<B>, Error = E>,
|
|
B: MessageBody + Unpin,
|
|
E: std::fmt::Debug,
|
|
{
|
|
let resp = test::TestRequest::post()
|
|
.uri("/")
|
|
.set_payload(payload)
|
|
.send_request(app)
|
|
.await;
|
|
|
|
assert!(resp.status().is_success());
|
|
let body = test::read_body(resp).await;
|
|
assert!(std::str::from_utf8(&body)
|
|
.unwrap()
|
|
.contains(&expected.to_string()));
|
|
}
|