DGNum/rust-irc
6
0
Fork 0
Code Issues 1 Pull requests 3 Projects Releases Packages Wiki Activity Actions

Added tests for the user tracking system.

Browse source
This commit is contained in:
Aaron Weiss 2014-10-09 14:32:06 -04:00
parent d6921eae74
commit 6261449e97
1 changed files with 49 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

49
src/bot.rs
View file

@ -273,4 +273,53 @@ mod test {
b.output().unwrap();
assert_eq!(b.conn.writer().deref_mut().get_ref(), "JOIN :#test\r\nJOIN :#test2\r\n".as_bytes());
}
#[test]
fn generate_user_list() {
let r = BufReader::new(":flare.to.ca.fyrechat.net 353 test @ #test :test test2 test3\r\n".as_bytes());
let w = MemWriter::new();
let c = Connection::new(w, r).unwrap();
let mut b = IrcBot::from_connection(c, |_, _, _, _| { Ok(()) }).unwrap();
b.output().unwrap();
let vec_res = match b.chanlists.borrow_mut().find_mut(&String::from_str("#test")) {
Some(v) => Ok(v.clone()),
None => Err("Could not find vec for channel."),
};
assert!(vec_res.is_ok());
let vec = vec_res.unwrap();
assert_eq!(vec, vec![String::from_str("test"), String::from_str("test2"), String::from_str("test3")]);
}
#[test]
fn add_to_user_list() {
let r = BufReader::new(":flare.to.ca.fyrechat.net 353 test @ #test :test test2\r\n:test3!test@test JOIN :#test\r\n".as_bytes());
let w = MemWriter::new();
let c = Connection::new(w, r).unwrap();
let mut b = IrcBot::from_connection(c, |_, _, _, _| { Ok(()) }).unwrap();
b.output().unwrap();
let vec_res = match b.chanlists.borrow_mut().find_mut(&String::from_str("#test")) {
Some(v) => Ok(v.clone()),
None => Err("Could not find vec for channel."),
};
assert!(vec_res.is_ok());
let vec = vec_res.unwrap();
assert_eq!(vec, vec![String::from_str("test"), String::from_str("test2"), String::from_str("test3")]);
}
#[test]
fn remove_from_user_list() {
let r = BufReader::new(":flare.to.ca.fyrechat.net 353 test @ #test :test test2 test3\r\n:test3!test@test PART #test :\r\n".as_bytes());
let w = MemWriter::new();
let c = Connection::new(w, r).unwrap();
let mut b = IrcBot::from_connection(c, |_, _, _, _| { Ok(()) }).unwrap();
b.output().unwrap();
let vec_res = match b.chanlists.borrow_mut().find_mut(&String::from_str("#test")) {
Some(v) => Ok(v.clone()),
None => Err("Could not find vec for channel."),
};
assert!(vec_res.is_ok());
let vec = vec_res.unwrap();
// n.b. ordering is not guaranteed, this only ought to hold because we're removing the last user
assert_eq!(vec, vec![String::from_str("test"), String::from_str("test2")]);
}
}