use tlid::{Pool, Wrapping};

#[test]
fn check_overflow() {
    let mut pool = Pool::<Wrapping<i16>>::new(1..4);
    assert_eq!(pool.next(), 1);
    assert_eq!(pool.next(), 2);
    assert_eq!(pool.next(), 3);
    assert_eq!(pool.next(), 1);
    assert_eq!(pool.next(), 2);
    assert_eq!(pool.next(), 3);
}

#[test]
fn check_overflow_u8() {
    let mut pool = Pool::<Wrapping<u8>>::new(253..255);
    assert_eq!(pool.next(), 253);
    assert_eq!(pool.next(), 254);
    assert_eq!(pool.next(), 253);
    assert_eq!(pool.next(), 254);
}

#[test]
#[should_panic]
fn check_overflow_lower() {
    let mut pool = Pool::<Wrapping<u8>>::new(100..99);
    assert_eq!(pool.next(), 100);
    assert_eq!(pool.next(), 100);
    assert_eq!(pool.next(), 100);
}

#[test]
#[should_panic]
fn check_overflow_same() {
    let mut pool = Pool::<Wrapping<u8>>::new(100..100);
    assert_eq!(pool.next(), 100);
    assert_eq!(pool.next(), 100);
}

#[test]
fn check_overflow_one() {
    let mut pool = Pool::<Wrapping<u8>>::new(100..101);
    assert_eq!(pool.next(), 100);
    assert_eq!(pool.next(), 100);
    assert_eq!(pool.next(), 100);
    assert_eq!(pool.next(), 100);
}

#[test]
fn check_overflow_two() {
    let mut pool = Pool::<Wrapping<u8>>::new(100..102);
    assert_eq!(pool.next(), 100);
    assert_eq!(pool.next(), 101);
    assert_eq!(pool.next(), 100);
    assert_eq!(pool.next(), 101);
}