extern crate ohmers;
extern crate redis;
extern crate rustc_serialize;

use std::collections::HashSet;
use std::iter::FromIterator;

use ohmers::{Ohmer, OhmerError, with};
use redis::Commands;
use rustc_serialize::Encodable;

#[derive(RustcEncodable, RustcDecodable, PartialEq, Debug)]
struct Thing {
    id: usize,
    name: String,
}

impl Default for Thing {
    fn default() -> Self {
        Thing {
            id: 0,
            name: "".to_string(),
        }
    }
}
impl Ohmer for Thing {
    fn id(&self) -> usize { self.id }
    fn set_id(&mut self, id: usize) { self.id = id; }
    fn unique_fields<'a>(&self) -> HashSet<&'a str> { HashSet::from_iter(vec!["name"]) }
}

#[test]
fn test_unique() {
    let client = redis::Client::open("redis://127.0.0.1/").unwrap();
    let _:() = client.del("Thing:uniques:name").unwrap();
    let mut thing = Thing { id: 0, name: "Door".to_string() };
    thing.save(&client).unwrap();
    let len:usize = client.hlen("Thing:uniques:name").unwrap();
    assert_eq!(1, len);

    let mut thing2 = Thing { id: 0, name: "Door".to_string() };
    assert_eq!(thing2.save(&client).unwrap_err(), OhmerError::UniqueIndexViolation("name".to_string()));

    let thing3:Thing = with("name", "Door", &client).unwrap().unwrap();
    assert_eq!(thing, thing3);

    assert!(with::<Thing, _>("name", "Window", &client).unwrap().is_none());
}