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
27

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

#[allow(dead_code)]
pub fn render_module<T>(module: &str, path: T) -> String
where
    T: Into<PathBuf>,
{
    render_module_with_status(module, path.into(), "0")
}

pub fn render_module_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 module = modules::handle(module, &context);

    module.unwrap().to_string()
}