tests/common.rs


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

use clap::{App, Arg};
use starship::context::Context;
use starship::modules;
use std::path::PathBuf;

pub fn render_segment<T>(module: &str, path: T) -> String
where
    T: Into<PathBuf>,
{
    render_segment_with_status(module, &path.into(), "0")
}

pub fn render_segment_with_status<T>(module: &str, path: T, status: &str) -> String
where
    T: Into<PathBuf>,
{
    // Create an `Arg` with status_code of "0"
    let args = App::new("starship")
        .arg(Arg::with_name("status_code"))
        .get_matches_from(vec!["starship", status]);
    let context = Context::new_with_dir(args, path.into());

    let segment = modules::handle(module, &context);

    segment.unwrap().output()
}