mirror of
https://github.com/starship/starship.git
synced 2024-11-25 17:54:08 +01:00
3daf3ddf26
Implement a timer module that takes a commandline argument, the number of seconds the last job took to complete, and displays it if appropriate. Alters shell initialization files to compute this number using date +%s where needed. Adds a config section to configure minimum amount of time before timer is shown (default is 2s)
81 lines
2.0 KiB
Rust
81 lines
2.0 KiB
Rust
use ansi_term::Color;
|
|
use std::fs;
|
|
use std::io;
|
|
use std::path::Path;
|
|
use tempfile::TempDir;
|
|
|
|
use crate::common::{self, TestCommand};
|
|
|
|
#[test]
|
|
fn config_blank_duration_1s() -> io::Result<()> {
|
|
let output = common::render_module("cmd_duration")
|
|
.arg("--cmd-duration=1")
|
|
.output()?;
|
|
let actual = String::from_utf8(output.stdout).unwrap();
|
|
|
|
let expected = "";
|
|
assert_eq!(expected, actual);
|
|
Ok(())
|
|
}
|
|
|
|
#[test]
|
|
fn config_blank_duration_5s() -> io::Result<()> {
|
|
let output = common::render_module("cmd_duration")
|
|
.arg("--cmd-duration=5")
|
|
.output()?;
|
|
let actual = String::from_utf8(output.stdout).unwrap();
|
|
|
|
let expected = format!("{} ", Color::Yellow.bold().paint("took 5s"));
|
|
assert_eq!(expected, actual);
|
|
Ok(())
|
|
}
|
|
|
|
#[test]
|
|
fn config_5s_duration_3s() -> io::Result<()> {
|
|
let output = common::render_module("cmd_duration")
|
|
.use_config(toml::toml! {
|
|
[cmd_duration]
|
|
min_time = 5
|
|
})
|
|
.arg("--cmd-duration=3")
|
|
.output()?;
|
|
let actual = String::from_utf8(output.stdout).unwrap();
|
|
|
|
let expected = "";
|
|
assert_eq!(expected, actual);
|
|
Ok(())
|
|
}
|
|
|
|
#[test]
|
|
fn config_5s_duration_10s() -> io::Result<()> {
|
|
let output = common::render_module("cmd_duration")
|
|
.use_config(toml::toml! {
|
|
[cmd_duration]
|
|
min_time = 5
|
|
})
|
|
.arg("--cmd-duration=10")
|
|
.output()?;
|
|
let actual = String::from_utf8(output.stdout).unwrap();
|
|
|
|
let expected = format!("{} ", Color::Yellow.bold().paint("took 10s"));
|
|
assert_eq!(expected, actual);
|
|
Ok(())
|
|
}
|
|
|
|
#[test]
|
|
fn config_disabled() -> io::Result<()> {
|
|
let output = common::render_module("cmd_duration")
|
|
.use_config(toml::toml! {
|
|
[cmd_duration]
|
|
disabled = true
|
|
min_time = 5
|
|
})
|
|
.arg("--cmd-duration=10")
|
|
.output()?;
|
|
let actual = String::from_utf8(output.stdout).unwrap();
|
|
|
|
let expected = "";
|
|
assert_eq!(expected, actual);
|
|
Ok(())
|
|
}
|