mod common;

pub use crate::common::*;
use drop_tracer::DropTracer;
use easy_node::{NrCell, NwCell};
use std::cmp::Ordering;
use std::collections::hash_map::DefaultHasher;
use std::collections::HashSet;
use std::hash::{Hash, Hasher};

#[test]
fn new() {
    DropTracer::test_drop(|tracer| {
        let result = NrCell::new((42, tracer.new_item()));

        assert_eq!(result.borrow().0, 42);
        assert_eq!(NrCell::strong_count(&result), 1);
        assert_eq!(NrCell::weak_count(&result), 0);
    });
}

#[test]
fn new_cyclic() {
    with_self_cycle();
    with_parent_and_child_cycle();

    fn with_self_cycle() {
        DropTracer::test_drop(|tracer| {
            let result = NrCell::new_cyclic(|w| {
                let value = (42, tracer.new_item());
                CyclicCell::new(value, w)
            });

            assert_eq!(result.borrow().val().0, 42);
            assert_eq!(result.borrow().me(), &NrCell::downgrade(&result));
            assert_eq!(NrCell::strong_count(&result), 1);
            assert_eq!(NrCell::weak_count(&result), 1);
            assert_eq!(NwCell::strong_count(result.borrow().me()), 1);
            assert_eq!(NwCell::weak_count(result.borrow().me()), 1);
        });
    }

    fn with_parent_and_child_cycle() {
        DropTracer::test_drop(|tracer| {
            let result = NrCell::new_cyclic(|w| {
                let c_value = tracer.new_item();
                let p_value = tracer.new_item();
                let child = NrCell::new(ChildCell::new(c_value, w.clone()));
                let parent = ParentCell::new(p_value, child);
                parent
            });

            let rb = result.borrow();
            assert_eq!(NrCell::strong_count(&result), 1);
            assert_eq!(NrCell::weak_count(&result), 1);
            assert_eq!(NrCell::strong_count(rb.child()), 1);
            assert_eq!(NrCell::weak_count(rb.child()), 0);
            assert_eq!(NwCell::strong_count(rb.child().borrow().parent()), 1);
            assert_eq!(NwCell::weak_count(rb.child().borrow().parent()), 1);
            assert_eq!(*rb.child().borrow().parent(), NrCell::downgrade(&result));
        });
    }
}

#[test]
fn downgrade() {
    with_normal();
    with_double();
    with_self_cycle();
    with_parent_and_child_cycle();

    fn with_normal() {
        DropTracer::test_drop(|tracer| {
            let target = NrCell::new(tracer.new_item());

            let result = NrCell::downgrade(&target);

            assert_eq!(NrCell::strong_count(&target), 1);
            assert_eq!(NwCell::strong_count(&result), 1);
            assert_eq!(NrCell::weak_count(&target), 1);
            assert_eq!(NwCell::weak_count(&result), 1);
        });
    }

    fn with_double() {
        DropTracer::test_drop(|tracer| {
            let target = NrCell::new(tracer.new_item());

            let result1 = NrCell::downgrade(&target);
            let result2 = NrCell::downgrade(&target);

            assert_eq!(NrCell::strong_count(&target), 1);
            assert_eq!(NwCell::strong_count(&result1), 1);
            assert_eq!(NwCell::strong_count(&result2), 1);
            assert_eq!(NrCell::weak_count(&target), 2);
            assert_eq!(NwCell::weak_count(&result1), 2);
            assert_eq!(NwCell::weak_count(&result2), 2);
        });
    }

    fn with_self_cycle() {
        DropTracer::test_drop(|tracer| {
            let target = NrCell::new_cyclic(|w| {
                let value = tracer.new_item();
                CyclicCell::new(value, w)
            });

            let result = NrCell::downgrade(&target);

            assert_eq!(NrCell::strong_count(&target), 1);
            assert_eq!(NwCell::strong_count(&result), 1);
            assert_eq!(NrCell::weak_count(&target), 2);
            assert_eq!(NwCell::weak_count(&result), 2);
        });
    }

    fn with_parent_and_child_cycle() {
        DropTracer::test_drop(|tracer| {
            let target = NrCell::new_cyclic(|w| {
                let c_value = tracer.new_item();
                let p_value = tracer.new_item();
                let child = NrCell::new(ChildCell::new(c_value, w.clone()));
                let parent = ParentCell::new(p_value, child);
                parent
            });

            let result = NrCell::downgrade(&target);

            assert_eq!(NrCell::strong_count(&target), 1);
            assert_eq!(NwCell::strong_count(&result), 1);
            assert_eq!(NrCell::weak_count(&target), 2);
            assert_eq!(NwCell::weak_count(&result), 2);
        });
    }
}

#[test]
fn clone() {
    let target = NrCell::new(42);

    let result = NrCell::clone(&target);

    assert_eq!(NrCell::strong_count(&target), 2);
    assert_eq!(NrCell::strong_count(&result), 2);
    assert_eq!(NrCell::weak_count(&target), 0);
    assert_eq!(NrCell::weak_count(&result), 0);
    assert_eq!(&result, &target);
    assert_eq!(*result, *target);
}

#[test]
fn drop() {
    with_weak();
    with_strong();
    with_self_cycle();
    with_parent_and_child_cycle();

    fn with_strong() {
        DropTracer::test_drop(|tracer| {
            let target = NrCell::new(tracer.new_item());
            let strong = NrCell::clone(&target);

            std::mem::drop(target);

            assert_eq!(NrCell::strong_count(&strong), 1);
            assert_eq!(NrCell::weak_count(&strong), 0);
        });
    }

    fn with_weak() {
        DropTracer::test_drop(|tracer| {
            let target = NrCell::new(tracer.new_item());
            let weak = NrCell::downgrade(&target);

            std::mem::drop(target);

            assert_eq!(NwCell::strong_count(&weak), 0);
            assert_eq!(NwCell::weak_count(&weak), 0);
        });
    }

    fn with_self_cycle() {
        DropTracer::test_drop(|tracer| {
            let target = NrCell::new_cyclic(|w| {
                let value = tracer.new_item();
                CyclicCell::new(value, w)
            });

            let viewer = NrCell::downgrade(&target);

            std::mem::drop(target);

            assert_eq!(NwCell::weak_count(&viewer), 0);
            assert_eq!(NwCell::strong_count(&viewer), 0);
        });
    }

    fn with_parent_and_child_cycle() {
        DropTracer::test_drop(|tracer| {
            let target = NrCell::new_cyclic(|w| {
                let c_value = tracer.new_item();
                let p_value = tracer.new_item();
                let child = NrCell::new(ChildCell::new(c_value, w.clone()));
                let parent = ParentCell::new(p_value, child);
                parent
            });

            let viewer = NrCell::downgrade(&target);

            std::mem::drop(target);

            assert_eq!(NwCell::weak_count(&viewer), 0);
            assert_eq!(NwCell::strong_count(&viewer), 0);
        });
    }
}

#[test]
fn default() {
    let result = <NrCell<i32> as Default>::default();
    assert_eq!(*result.borrow(), i32::default());
}

#[test]
fn hash() {
    with_same();
    with_diff();

    fn with_same() {
        let target1 = NrCell::new(());
        let target2 = NrCell::clone(&target1);
        let mut hasher1 = DefaultHasher::new();
        let mut hasher2 = DefaultHasher::new();

        target1.hash(&mut hasher1);
        target2.hash(&mut hasher2);

        assert_eq!(hasher1.finish(), hasher2.finish());
    }

    fn with_diff() {
        let target1 = NrCell::new(());
        let target2 = NrCell::new(());
        let mut hasher1 = DefaultHasher::new();
        let mut hasher2 = DefaultHasher::new();

        target1.hash(&mut hasher1);
        target2.hash(&mut hasher2);

        assert_ne!(hasher1.finish(), hasher2.finish());
    }
}

#[test]
fn cmp() {
    with_same();
    with_diff();

    fn with_same() {
        let target1 = NrCell::new(());
        let target2 = NrCell::clone(&target1);

        let result = target1.cmp(&target2);

        assert_eq!(result, Ordering::Equal);
    }

    fn with_diff() {
        let target1 = NrCell::new(());
        let target2 = NrCell::new(());

        let result1 = target1.cmp(&target2);
        let result2 = target2.cmp(&target1);
        let results = [result1, result2];

        let expecteds = [Ordering::Less, Ordering::Greater];
        assert_eq!(HashSet::from(results), HashSet::from(expecteds));
    }
}

#[test]
fn eq() {
    with_same();
    with_diff();

    fn with_same() {
        let target1 = NrCell::new(());
        let target2 = NrCell::clone(&target1);

        let result = target1.eq(&target2);

        assert!(result);
    }

    fn with_diff() {
        let target1 = NrCell::new(());
        let target2 = NrCell::new(());

        let result = target1.eq(&target2);

        assert!(!result);
    }
}

#[test]
fn partial_cmp() {
    with_same();
    with_diff();

    fn with_same() {
        let target1 = NrCell::new(());
        let target2 = NrCell::clone(&target1);

        let result = target1.partial_cmp(&target2);

        assert_eq!(result, Some(Ordering::Equal));
    }

    fn with_diff() {
        let target1 = NrCell::new(());
        let target2 = NrCell::new(());

        let result1 = target1.partial_cmp(&target2);
        let result2 = target2.partial_cmp(&target1);
        let results = [result1, result2];

        let expecteds = [Some(Ordering::Less), Some(Ordering::Greater)];
        assert_eq!(HashSet::from(results), HashSet::from(expecteds));
    }
}