mirror of
https://github.com/Llewellynvdm/starship.git
synced 2024-11-28 15:56:28 +00:00
1c66869117
Add a configuration option (show_sync_count) to the git_status module that will show/hide the counts ahead/behind of the tracked branch. Currently have this default to false, and would opt-in to show this information.
417 lines
11 KiB
Rust
417 lines
11 KiB
Rust
use ansi_term::Color;
|
|
use git2::Repository;
|
|
use std::env;
|
|
use std::fs::{self, File};
|
|
use std::io;
|
|
use std::process::Command;
|
|
|
|
use crate::common::{self, TestCommand};
|
|
|
|
#[test]
|
|
#[ignore]
|
|
fn shows_behind() -> io::Result<()> {
|
|
let fixture_repo_dir = common::create_fixture_repo()?;
|
|
let repo_dir = common::new_tempdir()?.path().join("rocket");
|
|
|
|
Repository::clone(fixture_repo_dir.to_str().unwrap(), &repo_dir.as_path()).unwrap();
|
|
|
|
Command::new("git")
|
|
.args(&["reset", "--hard", "HEAD^"])
|
|
.current_dir(repo_dir.as_path())
|
|
.output()?;
|
|
|
|
let output = common::render_module("git_status")
|
|
.arg("--path")
|
|
.arg(repo_dir)
|
|
.output()?;
|
|
let actual = String::from_utf8(output.stdout).unwrap();
|
|
let expected = Color::Red.bold().paint(format!("[{}] ", "⇣")).to_string();
|
|
|
|
assert_eq!(expected, actual);
|
|
|
|
Ok(())
|
|
}
|
|
|
|
#[test]
|
|
#[ignore]
|
|
fn shows_behind_with_count() -> io::Result<()> {
|
|
let fixture_repo_dir = common::create_fixture_repo()?;
|
|
let repo_dir = common::new_tempdir()?.path().join("rocket");
|
|
|
|
Repository::clone(fixture_repo_dir.to_str().unwrap(), &repo_dir.as_path()).unwrap();
|
|
|
|
Command::new("git")
|
|
.args(&["reset", "--hard", "HEAD^"])
|
|
.current_dir(repo_dir.as_path())
|
|
.output()?;
|
|
|
|
let output = common::render_module("git_status")
|
|
.use_config(toml::toml! {
|
|
[git_status]
|
|
show_sync_count = true
|
|
})
|
|
.arg("--path")
|
|
.arg(repo_dir)
|
|
.output()?;
|
|
let actual = String::from_utf8(output.stdout).unwrap();
|
|
let expected = Color::Red
|
|
.bold()
|
|
.paint(format!("[{}] ", "⇣1"))
|
|
.to_string();
|
|
|
|
assert_eq!(expected, actual);
|
|
|
|
Ok(())
|
|
}
|
|
|
|
#[test]
|
|
#[ignore]
|
|
fn shows_ahead() -> io::Result<()> {
|
|
let fixture_repo_dir = common::create_fixture_repo()?;
|
|
let repo_dir = common::new_tempdir()?.path().join("rocket");
|
|
|
|
Repository::clone(fixture_repo_dir.to_str().unwrap(), &repo_dir.as_path()).unwrap();
|
|
|
|
File::create(repo_dir.join("readme.md"))?;
|
|
|
|
Command::new("git")
|
|
.args(&["commit", "-am", "Update readme"])
|
|
.current_dir(&repo_dir)
|
|
.output()?;
|
|
|
|
let output = common::render_module("git_status")
|
|
.arg("--path")
|
|
.arg(repo_dir)
|
|
.output()?;
|
|
let actual = String::from_utf8(output.stdout).unwrap();
|
|
let expected = Color::Red.bold().paint(format!("[{}] ", "⇡")).to_string();
|
|
|
|
assert_eq!(expected, actual);
|
|
|
|
Ok(())
|
|
}
|
|
|
|
#[test]
|
|
#[ignore]
|
|
fn shows_ahead_with_count() -> io::Result<()> {
|
|
let fixture_repo_dir = common::create_fixture_repo()?;
|
|
let repo_dir = common::new_tempdir()?.path().join("rocket");
|
|
|
|
Repository::clone(fixture_repo_dir.to_str().unwrap(), &repo_dir.as_path()).unwrap();
|
|
|
|
File::create(repo_dir.join("readme.md"))?;
|
|
|
|
Command::new("git")
|
|
.args(&["commit", "-am", "Update readme"])
|
|
.current_dir(&repo_dir)
|
|
.output()?;
|
|
|
|
let output = common::render_module("git_status")
|
|
.use_config(toml::toml! {
|
|
[git_status]
|
|
show_sync_count = true
|
|
})
|
|
.arg("--path")
|
|
.arg(repo_dir)
|
|
.output()?;
|
|
let actual = String::from_utf8(output.stdout).unwrap();
|
|
let expected = Color::Red
|
|
.bold()
|
|
.paint(format!("[{}] ", "⇡1"))
|
|
.to_string();
|
|
|
|
assert_eq!(expected, actual);
|
|
|
|
Ok(())
|
|
}
|
|
|
|
#[test]
|
|
#[ignore]
|
|
fn shows_diverged() -> io::Result<()> {
|
|
let fixture_repo_dir = common::create_fixture_repo()?;
|
|
let repo_dir = common::new_tempdir()?.path().join("rocket");
|
|
|
|
Repository::clone(fixture_repo_dir.to_str().unwrap(), &repo_dir.as_path()).unwrap();
|
|
|
|
Command::new("git")
|
|
.args(&["reset", "--hard", "HEAD^"])
|
|
.current_dir(repo_dir.as_path())
|
|
.output()?;
|
|
|
|
fs::write(repo_dir.join("Cargo.toml"), " ")?;
|
|
|
|
Command::new("git")
|
|
.args(&["commit", "-am", "Update readme"])
|
|
.current_dir(repo_dir.as_path())
|
|
.output()?;
|
|
|
|
let output = common::render_module("git_status")
|
|
.arg("--path")
|
|
.arg(repo_dir)
|
|
.output()?;
|
|
let actual = String::from_utf8(output.stdout).unwrap();
|
|
let expected = Color::Red.bold().paint(format!("[{}] ", "⇕")).to_string();
|
|
|
|
assert_eq!(expected, actual);
|
|
|
|
Ok(())
|
|
}
|
|
|
|
#[test]
|
|
#[ignore]
|
|
fn shows_diverged_with_count() -> io::Result<()> {
|
|
let fixture_repo_dir = common::create_fixture_repo()?;
|
|
let repo_dir = common::new_tempdir()?.path().join("rocket");
|
|
|
|
Repository::clone(fixture_repo_dir.to_str().unwrap(), &repo_dir.as_path()).unwrap();
|
|
|
|
Command::new("git")
|
|
.args(&["reset", "--hard", "HEAD^"])
|
|
.current_dir(repo_dir.as_path())
|
|
.output()?;
|
|
|
|
fs::write(repo_dir.join("Cargo.toml"), " ")?;
|
|
|
|
Command::new("git")
|
|
.args(&["commit", "-am", "Update readme"])
|
|
.current_dir(repo_dir.as_path())
|
|
.output()?;
|
|
|
|
let output = common::render_module("git_status")
|
|
.use_config(toml::toml! {
|
|
[git_status]
|
|
show_sync_count = true
|
|
})
|
|
.arg("--path")
|
|
.arg(repo_dir)
|
|
.output()?;
|
|
let actual = String::from_utf8(output.stdout).unwrap();
|
|
let expected = Color::Red
|
|
.bold()
|
|
.paint(format!("[{}] ", "⇕⇡1⇣1"))
|
|
.to_string();
|
|
|
|
assert_eq!(expected, actual);
|
|
|
|
Ok(())
|
|
}
|
|
|
|
#[test]
|
|
#[ignore]
|
|
fn shows_conflicted() -> io::Result<()> {
|
|
let fixture_repo_dir = common::create_fixture_repo()?;
|
|
let repo_dir = common::new_tempdir()?.path().join("rocket");
|
|
|
|
Repository::clone(fixture_repo_dir.to_str().unwrap(), &repo_dir.as_path()).unwrap();
|
|
|
|
Command::new("git")
|
|
.args(&["reset", "--hard", "HEAD^"])
|
|
.current_dir(repo_dir.as_path())
|
|
.output()?;
|
|
|
|
fs::write(repo_dir.join("readme.md"), "# goodbye")?;
|
|
|
|
Command::new("git")
|
|
.args(&["add", "."])
|
|
.current_dir(repo_dir.as_path())
|
|
.output()?;
|
|
|
|
Command::new("git")
|
|
.args(&["commit", "-m", "Change readme"])
|
|
.current_dir(repo_dir.as_path())
|
|
.output()?;
|
|
|
|
Command::new("git")
|
|
.args(&["pull", "--rebase"])
|
|
.current_dir(repo_dir.as_path())
|
|
.output()?;
|
|
|
|
let output = common::render_module("git_status")
|
|
.arg("--path")
|
|
.arg(repo_dir)
|
|
.output()?;
|
|
let actual = String::from_utf8(output.stdout).unwrap();
|
|
let expected = Color::Red.bold().paint(format!("[{}] ", "=")).to_string();
|
|
|
|
assert_eq!(expected, actual);
|
|
|
|
Ok(())
|
|
}
|
|
|
|
#[test]
|
|
#[ignore]
|
|
fn shows_untracked_file() -> io::Result<()> {
|
|
let fixture_repo_dir = common::create_fixture_repo()?;
|
|
let repo_dir = common::new_tempdir()?.path().join("rocket");
|
|
|
|
Repository::clone(fixture_repo_dir.to_str().unwrap(), &repo_dir.as_path()).unwrap();
|
|
|
|
File::create(repo_dir.join("license"))?;
|
|
|
|
let output = common::render_module("git_status")
|
|
.arg("--path")
|
|
.arg(repo_dir)
|
|
.output()?;
|
|
let actual = String::from_utf8(output.stdout).unwrap();
|
|
let expected = Color::Red.bold().paint(format!("[{}] ", "?")).to_string();
|
|
|
|
assert_eq!(expected, actual);
|
|
|
|
Ok(())
|
|
}
|
|
|
|
#[test]
|
|
#[ignore]
|
|
fn doesnt_show_untracked_file_if_disabled() -> io::Result<()> {
|
|
let fixture_repo_dir = common::create_fixture_repo()?;
|
|
let repo_dir = common::new_tempdir()?.path().join("rocket");
|
|
|
|
Repository::clone(fixture_repo_dir.to_str().unwrap(), &repo_dir.as_path()).unwrap();
|
|
|
|
File::create(repo_dir.join("license"))?;
|
|
|
|
Command::new("git")
|
|
.args(&["config", "status.showUntrackedFiles", "no"])
|
|
.current_dir(repo_dir.as_path())
|
|
.output()?;
|
|
|
|
let output = common::render_module("git_status")
|
|
.arg("--path")
|
|
.arg(repo_dir)
|
|
.output()?;
|
|
let actual = String::from_utf8(output.stdout).unwrap();
|
|
let expected = "";
|
|
|
|
assert_eq!(expected, actual);
|
|
|
|
Ok(())
|
|
}
|
|
|
|
#[test]
|
|
#[ignore]
|
|
fn shows_stashed() -> io::Result<()> {
|
|
let fixture_repo_dir = common::create_fixture_repo()?;
|
|
let repo_dir = common::new_tempdir()?.path().join("rocket");
|
|
|
|
Repository::clone(fixture_repo_dir.to_str().unwrap(), &repo_dir.as_path()).unwrap();
|
|
|
|
File::create(repo_dir.join("readme.md"))?;
|
|
|
|
Command::new("git")
|
|
.arg("stash")
|
|
.current_dir(repo_dir.as_path())
|
|
.output()?;
|
|
|
|
let output = common::render_module("git_status")
|
|
.arg("--path")
|
|
.arg(repo_dir)
|
|
.output()?;
|
|
let actual = String::from_utf8(output.stdout).unwrap();
|
|
let expected = Color::Red.bold().paint(format!("[{}] ", "$")).to_string();
|
|
|
|
assert_eq!(expected, actual);
|
|
|
|
Ok(())
|
|
}
|
|
|
|
#[test]
|
|
#[ignore]
|
|
fn shows_modified() -> io::Result<()> {
|
|
let fixture_repo_dir = common::create_fixture_repo()?;
|
|
let repo_dir = common::new_tempdir()?.path().join("rocket");
|
|
|
|
Repository::clone(fixture_repo_dir.to_str().unwrap(), &repo_dir.as_path()).unwrap();
|
|
|
|
File::create(repo_dir.join("readme.md"))?;
|
|
|
|
let output = common::render_module("git_status")
|
|
.arg("--path")
|
|
.arg(repo_dir)
|
|
.output()?;
|
|
let actual = String::from_utf8(output.stdout).unwrap();
|
|
let expected = Color::Red.bold().paint(format!("[{}] ", "!")).to_string();
|
|
|
|
assert_eq!(expected, actual);
|
|
|
|
Ok(())
|
|
}
|
|
|
|
#[test]
|
|
#[ignore]
|
|
fn shows_staged_file() -> io::Result<()> {
|
|
let fixture_repo_dir = common::create_fixture_repo()?;
|
|
let repo_dir = common::new_tempdir()?.path().join("rocket");
|
|
|
|
Repository::clone(fixture_repo_dir.to_str().unwrap(), &repo_dir.as_path()).unwrap();
|
|
|
|
File::create(repo_dir.join("license"))?;
|
|
|
|
Command::new("git")
|
|
.args(&["add", "."])
|
|
.current_dir(repo_dir.as_path())
|
|
.output()?;
|
|
|
|
let output = common::render_module("git_status")
|
|
.arg("--path")
|
|
.arg(repo_dir)
|
|
.output()?;
|
|
let actual = String::from_utf8(output.stdout).unwrap();
|
|
let expected = Color::Red.bold().paint(format!("[{}] ", "+")).to_string();
|
|
|
|
assert_eq!(expected, actual);
|
|
|
|
Ok(())
|
|
}
|
|
|
|
#[test]
|
|
#[ignore]
|
|
fn shows_renamed_file() -> io::Result<()> {
|
|
let fixture_repo_dir = common::create_fixture_repo()?;
|
|
let repo_dir = common::new_tempdir()?.path().join("rocket");
|
|
|
|
Repository::clone(fixture_repo_dir.to_str().unwrap(), &repo_dir.as_path()).unwrap();
|
|
|
|
Command::new("git")
|
|
.args(&["mv", "readme.md", "readme.md.bak"])
|
|
.current_dir(repo_dir.as_path())
|
|
.output()?;
|
|
|
|
Command::new("git")
|
|
.args(&["add", "-A"])
|
|
.current_dir(repo_dir.as_path())
|
|
.output()?;
|
|
|
|
let output = common::render_module("git_status")
|
|
.arg("--path")
|
|
.arg(repo_dir)
|
|
.output()?;
|
|
let actual = String::from_utf8(output.stdout).unwrap();
|
|
let expected = Color::Red.bold().paint(format!("[{}] ", "»")).to_string();
|
|
|
|
assert_eq!(expected, actual);
|
|
|
|
Ok(())
|
|
}
|
|
|
|
#[test]
|
|
#[ignore]
|
|
fn shows_deleted_file() -> io::Result<()> {
|
|
let fixture_repo_dir = common::create_fixture_repo()?;
|
|
let repo_dir = common::new_tempdir()?.path().join("rocket");
|
|
|
|
Repository::clone(fixture_repo_dir.to_str().unwrap(), &repo_dir.as_path()).unwrap();
|
|
|
|
fs::remove_file(repo_dir.join("readme.md"))?;
|
|
|
|
let output = common::render_module("git_status")
|
|
.arg("--path")
|
|
.arg(repo_dir)
|
|
.output()?;
|
|
let actual = String::from_utf8(output.stdout).unwrap();
|
|
let expected = Color::Red.bold().paint(format!("[{}] ", "✘")).to_string();
|
|
|
|
assert_eq!(expected, actual);
|
|
|
|
Ok(())
|
|
}
|