async-graphql/tests/enum.rs

55 lines
1.0 KiB
Rust
Raw Normal View History

2020-03-07 04:40:04 +00:00
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);
let query = format!(
r#"{{
value
test_arg(input: A)
test_input(input: {{value: B}}) }}
"#
);
assert_eq!(
schema.query(&query).execute().await.unwrap(),
serde_json::json!({
"value": "A",
"test_arg": "A",
"test_input": "B",
})
);
}