61 lines
1.1 KiB
Rust
61 lines
1.1 KiB
Rust
|
use async_graphql::*;
|
||
|
|
||
|
#[tokio::test]
|
||
|
async fn test_flatten() {
|
||
|
#[derive(SimpleObject)]
|
||
|
struct A {
|
||
|
a: i32,
|
||
|
b: i32,
|
||
|
}
|
||
|
|
||
|
struct B;
|
||
|
|
||
|
#[Object]
|
||
|
impl B {
|
||
|
#[graphql(flatten)]
|
||
|
async fn a(&self) -> A {
|
||
|
A { a: 100, b: 200 }
|
||
|
}
|
||
|
|
||
|
async fn c(&self) -> i32 {
|
||
|
300
|
||
|
}
|
||
|
}
|
||
|
|
||
|
struct Query;
|
||
|
|
||
|
#[Object]
|
||
|
impl Query {
|
||
|
async fn obj(&self) -> B {
|
||
|
B
|
||
|
}
|
||
|
}
|
||
|
|
||
|
let schema = Schema::new(Query, EmptyMutation, EmptySubscription);
|
||
|
let query = "{ __type(name: \"B\") { fields { name } } }";
|
||
|
assert_eq!(
|
||
|
schema.execute(query).await.data,
|
||
|
value!({
|
||
|
"__type": {
|
||
|
"fields": [
|
||
|
{"name": "a"},
|
||
|
{"name": "b"},
|
||
|
{"name": "c"}
|
||
|
]
|
||
|
}
|
||
|
})
|
||
|
);
|
||
|
|
||
|
let query = "{ obj { a b c } }";
|
||
|
assert_eq!(
|
||
|
schema.execute(query).await.data,
|
||
|
value!({
|
||
|
"obj": {
|
||
|
"a": 100,
|
||
|
"b": 200,
|
||
|
"c": 300,
|
||
|
}
|
||
|
})
|
||
|
);
|
||
|
}
|