2017-08-20 19:29:23 +00:00
|
|
|
use ansi_term::Style;
|
2017-05-20 17:39:37 +00:00
|
|
|
use users::Users;
|
2017-05-20 17:14:04 +00:00
|
|
|
|
|
|
|
use fs::fields as f;
|
|
|
|
use output::cell::TextCell;
|
|
|
|
|
|
|
|
|
2017-08-20 19:29:23 +00:00
|
|
|
|
2017-05-20 19:29:22 +00:00
|
|
|
impl f::User {
|
2017-08-20 19:29:23 +00:00
|
|
|
pub fn render<C: Colours, U: Users>(&self, colours: &C, users: &U) -> TextCell {
|
2017-05-20 19:29:22 +00:00
|
|
|
let user_name = match users.get_user_by_uid(self.0) {
|
2017-05-20 17:14:04 +00:00
|
|
|
Some(user) => user.name().to_owned(),
|
2017-05-20 19:29:22 +00:00
|
|
|
None => self.0.to_string(),
|
2017-05-20 17:14:04 +00:00
|
|
|
};
|
|
|
|
|
2017-08-20 19:29:23 +00:00
|
|
|
let style = if users.get_current_uid() == self.0 { colours.you() }
|
|
|
|
else { colours.someone_else() };
|
2017-05-20 17:14:04 +00:00
|
|
|
TextCell::paint(style, user_name)
|
|
|
|
}
|
2017-05-20 17:39:37 +00:00
|
|
|
}
|
2017-05-20 17:14:04 +00:00
|
|
|
|
2017-08-20 19:29:23 +00:00
|
|
|
|
|
|
|
pub trait Colours {
|
|
|
|
fn you(&self) -> Style;
|
|
|
|
fn someone_else(&self) -> Style;
|
|
|
|
}
|
|
|
|
|
|
|
|
|
2017-05-20 17:39:37 +00:00
|
|
|
#[cfg(test)]
|
|
|
|
#[allow(unused_results)]
|
|
|
|
pub mod test {
|
2017-08-20 19:29:23 +00:00
|
|
|
use super::Colours;
|
2017-05-20 17:39:37 +00:00
|
|
|
use fs::fields as f;
|
|
|
|
use output::cell::TextCell;
|
2017-05-20 17:14:04 +00:00
|
|
|
|
2017-05-20 17:39:37 +00:00
|
|
|
use users::User;
|
|
|
|
use users::mock::MockUsers;
|
|
|
|
use ansi_term::Colour::*;
|
2017-08-20 19:29:23 +00:00
|
|
|
use ansi_term::Style;
|
|
|
|
|
|
|
|
|
|
|
|
struct TestColours;
|
|
|
|
|
|
|
|
impl Colours for TestColours {
|
|
|
|
fn you(&self) -> Style { Red.bold() }
|
|
|
|
fn someone_else(&self) -> Style { Blue.underline() }
|
|
|
|
}
|
|
|
|
|
2017-05-20 17:14:04 +00:00
|
|
|
|
2017-05-20 17:39:37 +00:00
|
|
|
#[test]
|
|
|
|
fn named() {
|
|
|
|
let mut users = MockUsers::with_current_uid(1000);
|
|
|
|
users.add_user(User::new(1000, "enoch", 100));
|
|
|
|
|
|
|
|
let user = f::User(1000);
|
|
|
|
let expected = TextCell::paint_str(Red.bold(), "enoch");
|
2017-08-20 19:29:23 +00:00
|
|
|
assert_eq!(expected, user.render(&TestColours, &users))
|
2017-05-20 17:14:04 +00:00
|
|
|
}
|
|
|
|
|
2017-05-20 17:39:37 +00:00
|
|
|
#[test]
|
|
|
|
fn unnamed() {
|
|
|
|
let users = MockUsers::with_current_uid(1000);
|
|
|
|
|
|
|
|
let user = f::User(1000);
|
2017-08-20 19:29:23 +00:00
|
|
|
let expected = TextCell::paint_str(Red.bold(), "1000");
|
|
|
|
assert_eq!(expected, user.render(&TestColours, &users));
|
2017-05-20 17:39:37 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
#[test]
|
|
|
|
fn different_named() {
|
2017-05-20 19:29:22 +00:00
|
|
|
let mut users = MockUsers::with_current_uid(0);
|
|
|
|
users.add_user(User::new(1000, "enoch", 100));
|
2017-05-20 17:39:37 +00:00
|
|
|
|
|
|
|
let user = f::User(1000);
|
2017-08-20 19:29:23 +00:00
|
|
|
let expected = TextCell::paint_str(Blue.underline(), "enoch");
|
|
|
|
assert_eq!(expected, user.render(&TestColours, &users));
|
2017-05-20 17:14:04 +00:00
|
|
|
}
|
|
|
|
|
2017-05-20 17:39:37 +00:00
|
|
|
#[test]
|
|
|
|
fn different_unnamed() {
|
|
|
|
let user = f::User(1000);
|
2017-08-20 19:29:23 +00:00
|
|
|
let expected = TextCell::paint_str(Blue.underline(), "1000");
|
|
|
|
assert_eq!(expected, user.render(&TestColours, &MockUsers::with_current_uid(0)));
|
2017-05-20 17:39:37 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
#[test]
|
|
|
|
fn overflow() {
|
|
|
|
let user = f::User(2_147_483_648);
|
|
|
|
let expected = TextCell::paint_str(Blue.underline(), "2147483648");
|
2017-08-20 19:29:23 +00:00
|
|
|
assert_eq!(expected, user.render(&TestColours, &MockUsers::with_current_uid(0)));
|
2017-05-20 17:14:04 +00:00
|
|
|
}
|
|
|
|
}
|