#![feature(test)]
extern crate test;
extern crate kernel;

use kernel::streams::stack::*;

#[derive(Debug,PartialEq,Clone)]
struct Entry(u16, i64);

#[test]
pub fn stack() {
    let capacity = (!0 as u16) as usize;
    let mut stack: Stack<Entry> = Stack::with_capacity(capacity);

    assert!(stack.capacity() == capacity);
    assert!(stack.len() == 0);
    assert!(stack.num_frames() == 0);

    let mut id = stack.push_frame();
    println!("LastFrame ID: {:?}", &id);
    assert!(stack.capacity() == capacity);
    assert!(stack.len() == 0);

    let _ = stack.insert(Entry(1, 1));
    assert!(stack.len() == 1);
    assert!(stack.num_frames() == 1);

    id = stack.push_frame();
    println!("LastFrame ID: {:?}", &id);
    let _ = stack.insert(Entry(2, 2));
    let _ = stack.insert(Entry(3, 3));
    let _ = stack.insert(Entry(4, 4));
    assert!(stack.capacity() == capacity);
    assert!(stack.len() == 4);
    assert!(stack.num_frames() == 2);
    assert_eq!(stack.get(|it| (*it).0 == 4, None).unwrap(), &Entry(4, 4));

    let _ = stack.pop_frame();
    assert_eq!(stack.get(|it| (*it).0 == 3, None), None);
    assert_eq!(stack.get(|it| (*it).0 == 2, None), None);
    assert!(stack.num_frames() == 1);
    assert!(stack.len() == 1);

    // assert_eq!(stack.pop_frame(), Ok(()));
    // assert_eq!(stack.pop_frame(), Err(Error::InvalidOperation));

    let _ = stack.push_frame();
    let items = [Entry(9, 9), Entry(6, 6), Entry(7, 7)];
    let _ = stack.insert_many(&items);

    assert_eq!(stack.get(|it| (*it).0 == 6, None).unwrap(), &Entry(6, 6));
    assert_eq!(stack.get(|it| (*it).0 == 1, None).unwrap(), &Entry(1, 1));
    let _ = stack.push_frame();
    let _ = stack.insert(Entry(2, 2));
    assert!(stack.num_frames() == 3);
    assert_eq!(stack.get(|it| (*it).0 == 1, None).unwrap(), &Entry(1, 1));
}