async-graphql/tests/enum.rs
2020-03-17 17:26:59 +08:00

59 lines
1.1 KiB
Rust

use async_graphql::*;
#[async_std::test]
pub async fn test_enum_type() {
#[Enum]
enum MyEnum {
A,
B,
}
#[InputObject]
struct MyInput {
value: MyEnum,
}
struct Root {
value: MyEnum,
}
#[Object]
impl Root {
#[field]
async fn value(&self) -> MyEnum {
self.value
}
#[field]
async fn test_arg(&self, input: MyEnum) -> MyEnum {
input
}
#[field]
async fn test_input<'a>(&self, input: MyInput) -> MyEnum {
input.value
}
}
let schema = Schema::new(
Root { value: MyEnum::A },
GQLEmptyMutation,
GQLEmptySubscription,
);
let query = format!(
r#"{{
value
testArg(input: A)
testInput(input: {{value: B}}) }}
"#
);
assert_eq!(
schema.query(&query).execute().await.unwrap(),
serde_json::json!({
"value": "A",
"testArg": "A",
"testInput": "B",
})
);
}