// Original author of this test is <https://github.com/davidpdrsn>.
use juniper::*;

pub struct Context;

impl juniper::Context for Context {}

pub struct Query;

graphql_object!(Query: Context |&self| {
    field users(&executor) -> Vec<User> {
        let lh = executor.look_ahead();
        assert_eq!(lh.field_name(), "users");
        vec![User]
    }

    field countries(&executor) -> Vec<Country> {
        let lh = executor.look_ahead();
        assert_eq!(lh.field_name(), "countries");
        vec![Country]
    }
});

#[derive(Clone)]
pub struct User;

graphql_object!(User: Context |&self| {
    field id() -> i32 {
        1
    }
});

#[derive(Clone)]
pub struct Country;

graphql_object!(Country: Context |&self| {
    field id() -> i32 {
        2
    }
});

type Schema = juniper::RootNode<'static, Query, EmptyMutation<Context>>;

#[test]
fn users() {
    let ctx = Context;

    let query = r#"{ users { id } }"#;

    let (_, errors) = juniper::execute(
        query,
        None,
        &Schema::new(Query, EmptyMutation::<Context>::new()),
        &juniper::Variables::new(),
        &ctx,
    )
    .unwrap();

    assert_eq!(errors.len(), 0);
}

#[test]
fn countries() {
    let ctx = Context;

    let query = r#"{ countries { id } }"#;

    let (_, errors) = juniper::execute(
        query,
        None,
        &Schema::new(Query, EmptyMutation::new()),
        &juniper::Variables::new(),
        &ctx,
    )
    .unwrap();

    assert_eq!(errors.len(), 0);
}

#[test]
fn both() {
    let ctx = Context;

    let query = r#"
    {
        countries { id }
        users { id }
    }
    "#;

    let (_, errors) = juniper::execute(
        query,
        None,
        &Schema::new(Query, EmptyMutation::<Context>::new()),
        &juniper::Variables::new(),
        &ctx,
    )
    .unwrap();

    assert_eq!(errors.len(), 0);
}

#[test]
fn both_in_different_order() {
    let ctx = Context;

    let query = r#"
    {
        users { id }
        countries { id }
    }
    "#;

    let (_, errors) = juniper::execute(
        query,
        None,
        &Schema::new(Query, EmptyMutation::<Context>::new()),
        &juniper::Variables::new(),
        &ctx,
    )
    .unwrap();

    assert_eq!(errors.len(), 0);
}