anna/async-graphql
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Add tests to limit complexity and depth.

Browse Source
This commit is contained in:
sunli 2020-05-05 13:46:56 +08:00
parent 381198cfdb
commit d59f3acc50
2 changed files with 138 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
src/types/connection/edge.rs
View File

@ -42,13 +42,12 @@ where
registry.create_type::<Self, _>(|registry| {
E::create_type_info(registry);
let extra_fields = if let Some(registry::Type::Object { fields, .. }) =
registry.types.get_mut(E::type_name().as_ref())
registry.types.remove(E::type_name().as_ref())
{
fields.clone()
fields
} else {
unreachable!()
};
registry.types.remove(E::type_name().as_ref());
registry::Type::Object {
name: Self::type_name().to_string(),

136
tests/complexity_and_depth.rs Normal file
View File

@ -0,0 +1,136 @@
use async_graphql::*;
#[async_std::test]
pub async fn test_complexity_and_depth() {
struct Query;
struct MyObj;
#[Object]
impl MyObj {
async fn a(&self) -> i32 {
1
}
async fn b(&self) -> i32 {
2
}
async fn c(&self) -> MyObj {
MyObj
}
}
#[Object]
impl Query {
async fn value(&self) -> i32 {
1
}
async fn obj(&self) -> MyObj {
MyObj
}
}
let query = "{ a:value b:value c:value }";
let schema = Schema::build(Query, EmptyMutation, EmptySubscription)
.limit_complexity(2)
.finish();
assert_eq!(
schema.execute(&query).await.unwrap_err(),
Error::Query {
pos: Pos { line: 0, column: 0 },
path: None,
err: QueryError::TooComplex,
}
);
let query = "{ a:value b:value }";
let schema = Schema::build(Query, EmptyMutation, EmptySubscription)
.limit_complexity(2)
.finish();
assert_eq!(
schema.execute(&query).await.unwrap().data,
serde_json::json!({
"a": 1,
"b": 1,
})
);
let query = "{ obj { a b } }";
let schema = Schema::build(Query, EmptyMutation, EmptySubscription)
.limit_complexity(2)
.finish();
assert_eq!(
schema.execute(&query).await.unwrap_err(),
Error::Query {
pos: Pos { line: 0, column: 0 },
path: None,
err: QueryError::TooComplex,
}
);
let query = "{ obj { a } }";
let schema = Schema::build(Query, EmptyMutation, EmptySubscription)
.limit_complexity(2)
.finish();
assert_eq!(
schema.execute(&query).await.unwrap().data,
serde_json::json!({
"obj": { "a": 1 }
})
);
let query = r#"
{
obj { # 0
a b c { # 1
a b c { # 2
a b # 3
}
}
}
}"#;
let schema = Schema::build(Query, EmptyMutation, EmptySubscription)
.limit_depth(2)
.finish();
assert_eq!(
schema.execute(&query).await.unwrap_err(),
Error::Query {
pos: Pos { line: 0, column: 0 },
path: None,
err: QueryError::TooDeep,
}
);
let query = r#"
{
obj { # 0
a b c { # 1
a b c { # 2
a b # 3
}
}
}
}"#;
let schema = Schema::build(Query, EmptyMutation, EmptySubscription)
.limit_depth(3)
.finish();
assert_eq!(
schema.execute(&query).await.unwrap().data,
serde_json::json!({
"obj": {
"a": 1,
"b": 2,
"c": {
"a": 1,
"b": 2,
"c": {
"a": 1,
"b": 2,
}
}
}
})
);
}