// Test data structures.

extern crate sharp_pencil;

use sharp_pencil::datastructures::MultiDict;


#[test]
fn test_multi_dict_basic_interface() {
    let mut multi_dict = MultiDict::new();
    multi_dict.add("Content-Type".to_string(), "text/plain".to_string());
    multi_dict.add("X-Foo".to_string(), "bar".to_string());
    assert!(multi_dict.get::<str>("X-Foo") != None);
    assert!(multi_dict.get::<str>("Content-Type") != None);

    multi_dict.set("Content-Type", "foo/bar".to_string());
    assert!(multi_dict.get::<str>("Content-Type").unwrap() == "foo/bar");

    multi_dict.add("X-Foo".to_string(), "bar2".to_string());

    let mut all_keys = vec![];
    for key in multi_dict.keys() {
        all_keys.push(key);
    }
    assert!(all_keys.len() == 2);
    let mut all_values = vec![];
    for value in multi_dict.values() {
        all_values.push(value);
    }
    assert!(all_values.len() == 2);
}