summaryrefslogtreecommitdiffstats
path: root/tests/testsuite/username.rs
blob: 08f4e668be4467a597c18038f782937cbe76fcaf (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
use ansi_term::Color;
use std::io;

use crate::common::{self, TestCommand};

// TODO: Add tests for if root user (UID == 0)
// Requires mocking

#[test]
fn no_env_variables() -> io::Result<()> {
    let output = common::render_module("username").output()?;
    let actual = String::from_utf8(output.stdout).unwrap();
    assert_eq!("", actual);
    Ok(())
}

#[test]
fn logname_equals_user() -> io::Result<()> {
    let output = common::render_module("username")
        .env("LOGNAME", "astronaut")
        .env("USER", "astronaut")
        .output()?;
    let actual = String::from_utf8(output.stdout).unwrap();
    assert_eq!("", actual);
    Ok(())
}

#[test]
fn ssh_wo_username() -> io::Result<()> {
    // SSH connection w/o username
    let output = common::render_module("username")
        .env("SSH_CONNECTION", "192.168.223.17 36673 192.168.223.229 22")
        .output()?;
    let actual = String::from_utf8(output.stdout).unwrap();
    assert_eq!("", actual);
    Ok(())
}

#[test]
fn current_user_not_logname() -> io::Result<()> {
    let output = common::render_module("username")
        .env("LOGNAME", "astronaut")
        .env("USER", "cosmonaut")
        .output()?;
    let actual = String::from_utf8(output.stdout).unwrap();

    let expected = format!("via {} ", Color::Yellow.bold().paint("cosmonaut"));
    assert_eq!(expected, actual);
    Ok(())
}

#[test]
fn ssh_connection() -> io::Result<()> {
    let output = common::render_module("username")
        .env("USER", "astronaut")
        .env("SSH_CONNECTION", "192.168.223.17 36673 192.168.223.229 22")
        .output()?;
    let actual = String::from_utf8(output.stdout).unwrap();

    let expected = format!("via {} ", Color::Yellow.bold().paint("astronaut"));
    assert_eq!(expected, actual);
    Ok(())
}

#[test]
fn show_always() -> io::Result<()> {
    let output = common::render_module("username")
        .env("USER", "astronaut")
        .use_config(toml::toml! {
        [username]
        show_always = true})
        .output()?;
    let actual = String::from_utf8(output.stdout).unwrap();

    let expected = format!("via {} ", Color::Yellow.bold().paint("astronaut"));

    assert_eq!(expected, actual);
    Ok(())
}