use derivative::Derivative;
use entity::*;
use entity_inmemory::InmemoryDatabase;

#[test]
fn build_method_on_ent_will_populate_with_global_database() {
    #[derive(Clone, Ent, EntBuilder)]
    struct TestEnt {
        #[ent(id)]
        id: Id,

        #[ent(database)]
        database: WeakDatabaseRc,

        #[ent(created)]
        created: u64,

        #[ent(last_updated)]
        last_updated: u64,
    }

    // With no global database set
    let builder = TestEnt::build();
    assert!(
        WeakDatabaseRc::ptr_eq(&builder.database, &WeakDatabaseRc::new()),
        "Builder configured with unexpected database"
    );

    entity::global::with_db(InmemoryDatabase::default(), || {
        // With global database set
        let builder = TestEnt::build();
        assert!(
            !WeakDatabaseRc::ptr_eq(&builder.database, &WeakDatabaseRc::new()),
            "Builder not configured with global database"
        );
    });
}

#[test]
fn default_fills_in_required_fields() {
    #[derive(Clone, Ent, EntBuilder)]
    struct TestEnt {
        #[ent(id)]
        id: Id,

        #[ent(database)]
        database: WeakDatabaseRc,

        #[ent(created)]
        created: u64,

        #[ent(last_updated)]
        last_updated: u64,
    }

    let ent = TestEntBuilder::default()
        .finish()
        .expect("Failed to create test ent");

    assert_eq!(ent.id, EPHEMERAL_ID);
    assert!(WeakDatabaseRc::ptr_eq(
        &ent.database,
        &WeakDatabaseRc::new()
    ));
    assert!(ent.created > 0);
    assert!(ent.last_updated > 0);
}

#[test]
fn produces_an_error_enum_for_each_struct_field() {
    #[derive(Clone, Ent, EntBuilder)]
    struct TestEnt {
        #[ent(id)]
        id: Id,

        #[ent(database)]
        database: WeakDatabaseRc,

        #[ent(created)]
        created: u64,

        #[ent(last_updated)]
        last_updated: u64,

        #[ent(field)]
        field1: u32,

        #[ent(field)]
        field2: String,

        #[ent(edge(type = "TestEnt"))]
        edge1: Option<Id>,

        #[ent(edge(type = "TestEnt"))]
        edge2: Id,

        #[ent(edge(type = "TestEnt"))]
        edge3: Vec<Id>,
    }

    assert_eq!(
        TestEntBuilderError::MissingField1.to_string(),
        "Missing field1"
    );
    assert_eq!(
        TestEntBuilderError::MissingField2.to_string(),
        "Missing field2"
    );
    assert_eq!(
        TestEntBuilderError::MissingEdge1.to_string(),
        "Missing edge1"
    );
    assert_eq!(
        TestEntBuilderError::MissingEdge2.to_string(),
        "Missing edge2"
    );
    assert_eq!(
        TestEntBuilderError::MissingEdge3.to_string(),
        "Missing edge3"
    );
}

#[test]
fn default_returns_a_builder_with_all_normal_fields_set_to_none() {
    #[derive(Clone, Ent, EntBuilder)]
    struct TestEnt {
        #[ent(id)]
        id: Id,

        #[ent(database)]
        database: WeakDatabaseRc,

        #[ent(created)]
        created: u64,

        #[ent(last_updated)]
        last_updated: u64,

        #[ent(field)]
        field1: u32,

        #[ent(field)]
        field2: String,

        #[ent(edge(type = "TestEnt"))]
        edge1: Option<Id>,

        #[ent(edge(type = "TestEnt"))]
        edge2: Id,

        #[ent(edge(type = "TestEnt"))]
        edge3: Vec<Id>,
    }

    let builder = TestEntBuilder::default();
    assert_eq!(builder.id, EPHEMERAL_ID);
    assert!(WeakDatabaseRc::ptr_eq(
        &builder.database,
        &WeakDatabaseRc::new()
    ));
    assert!(builder.created > 0);
    assert!(builder.last_updated > 0);
    assert_eq!(builder.field1, None);
    assert_eq!(builder.field2, None);
    assert_eq!(builder.edge1, None);
    assert_eq!(builder.edge2, None);
    assert_eq!(builder.edge3, None);
}

#[test]
fn build_fails_when_struct_field_is_not_set() {
    #[derive(Clone, Derivative, Ent, EntBuilder)]
    #[derivative(Debug)]
    struct TestEnt {
        #[ent(id)]
        id: Id,

        #[derivative(Debug = "ignore")]
        #[ent(database)]
        database: WeakDatabaseRc,

        #[ent(created)]
        created: u64,

        #[ent(last_updated)]
        last_updated: u64,

        #[ent(field)]
        field1: u32,

        #[ent(field)]
        field2: String,

        #[ent(edge(type = "TestEnt"))]
        edge1: Option<Id>,

        #[ent(edge(type = "TestEnt"))]
        edge2: Id,

        #[ent(edge(type = "TestEnt"))]
        edge3: Vec<Id>,
    }

    assert_eq!(
        TestEntBuilder::default()
            .field2(String::from("test"))
            .edge1(None)
            .edge2(0)
            .edge3(vec![])
            .finish()
            .unwrap_err(),
        TestEntBuilderError::MissingField1,
    );

    assert_eq!(
        TestEntBuilder::default()
            .field1(0)
            .edge1(None)
            .edge2(0)
            .edge3(vec![])
            .finish()
            .unwrap_err(),
        TestEntBuilderError::MissingField2,
    );

    assert_eq!(
        TestEntBuilder::default()
            .field1(0)
            .field2(String::from("test"))
            .edge2(0)
            .edge3(vec![])
            .finish()
            .unwrap_err(),
        TestEntBuilderError::MissingEdge1,
    );

    assert_eq!(
        TestEntBuilder::default()
            .field1(0)
            .field2(String::from("test"))
            .edge1(None)
            .edge3(vec![])
            .finish()
            .unwrap_err(),
        TestEntBuilderError::MissingEdge2,
    );

    assert_eq!(
        TestEntBuilder::default()
            .field1(0)
            .field2(String::from("test"))
            .edge1(None)
            .edge2(0)
            .finish()
            .unwrap_err(),
        TestEntBuilderError::MissingEdge3,
    );
}

#[test]
fn build_succeeds_when_all_struct_fields_are_set() {
    #[derive(Clone, Derivative, Ent, EntBuilder)]
    #[derivative(Debug)]
    struct TestEnt {
        #[ent(id)]
        id: Id,

        #[derivative(Debug = "ignore")]
        #[ent(database)]
        database: WeakDatabaseRc,

        #[ent(created)]
        created: u64,

        #[ent(last_updated)]
        last_updated: u64,

        #[ent(field)]
        field1: u32,

        #[ent(field)]
        field2: String,

        #[ent(edge(type = "TestEnt"))]
        edge1: Option<Id>,

        #[ent(edge(type = "TestEnt"))]
        edge2: Id,

        #[ent(edge(type = "TestEnt"))]
        edge3: Vec<Id>,
    }

    let ent = TestEntBuilder::default()
        .id(1)
        .database(WeakDatabaseRc::new())
        .created(2)
        .last_updated(3)
        .field1(4)
        .field2(String::from("test"))
        .edge1(Some(5))
        .edge2(6)
        .edge3(vec![7, 8])
        .finish()
        .expect("Failed to build ent!");
    assert_eq!(ent.id, 1);
    assert_eq!(ent.created, 2);
    assert_eq!(ent.last_updated, 3);
    assert_eq!(ent.field1, 4);
    assert_eq!(ent.field2, String::from("test"));
    assert_eq!(ent.edge1, Some(5));
    assert_eq!(ent.edge2, 6);
    assert_eq!(ent.edge3, vec![7, 8]);
}

#[test]
fn builder_sets_computed_field_caches_to_none() {
    #[derive(Clone, Derivative, Ent, EntBuilder)]
    #[derivative(Debug)]
    struct TestEnt {
        #[ent(id)]
        id: Id,

        #[derivative(Debug = "ignore")]
        #[ent(database)]
        database: WeakDatabaseRc,

        #[ent(created)]
        created: u64,

        #[ent(last_updated)]
        last_updated: u64,

        #[ent(field)]
        real_field: u32,

        #[ent(field(computed = "123"))]
        computed_field: Option<u32>,
    }

    let ent = TestEntBuilder::default()
        .id(1)
        .database(WeakDatabaseRc::new())
        .created(2)
        .last_updated(3)
        .real_field(4)
        .finish()
        .expect("Failed to build ent!");
    assert_eq!(ent.id, 1);
    assert_eq!(ent.created, 2);
    assert_eq!(ent.last_updated, 3);
    assert_eq!(ent.real_field, 4);
    assert_eq!(ent.computed_field, None);
}

#[test]
fn supports_generic_fields() {
    #[derive(Clone, Ent, EntBuilder)]
    struct TestEnt<T>
    where
        T: ValueLike + Clone + Send + Sync + 'static,
    {
        #[ent(id)]
        id: Id,

        #[ent(database)]
        database: WeakDatabaseRc,

        #[ent(created)]
        created: u64,

        #[ent(last_updated)]
        last_updated: u64,

        #[ent(field)]
        generic_field: T,
    }

    let builder = TestEntBuilder::default();
    assert_eq!(builder.generic_field, None);

    let ent = builder
        .generic_field(3)
        .finish()
        .expect("Failed to create with generic field");
    assert_eq!(ent.generic_field, 3);
}