From 7136059dcd4d3b3fe6857dcf0542bd850a50d813 Mon Sep 17 00:00:00 2001
From: Matan Kushner <hello@matchai.me>
Date: Thu, 4 Apr 2019 14:32:22 -0400
Subject: [PATCH] Add tests to dir home truncation

---
 azure-pipelines.yml |  3 ++-
 src/modules/dir.rs  | 27 ++++++++++++++++++++-------
 2 files changed, 22 insertions(+), 8 deletions(-)

diff --git a/azure-pipelines.yml b/azure-pipelines.yml
index 602efe35..6fe1934f 100644
--- a/azure-pipelines.yml
+++ b/azure-pipelines.yml
@@ -54,5 +54,6 @@ jobs:
         condition: eq( variables['Agent.OS'], 'Windows_NT' )
       - script: cargo build --all
         displayName: Cargo build
-      - script: cargo test --all
+        # Until env stubbing is solved, make tests run on a single thread
+      - script: cargo test --all -- --test-threads=1
         displayName: Cargo test
diff --git a/src/modules/dir.rs b/src/modules/dir.rs
index 5f17d709..098d0f14 100644
--- a/src/modules/dir.rs
+++ b/src/modules/dir.rs
@@ -5,7 +5,7 @@ use ansi_term::{Color, Style};
 use clap::ArgMatches;
 
 /// Creates a segment with the current directory
-pub fn segment(args: &ArgMatches) -> Segment {
+pub fn segment(_: &ArgMatches) -> Segment {
     const COLOR_DIR: Color = Color::Cyan;
     const HOME_SYMBOL: char = '~';
 
@@ -28,14 +28,12 @@ pub fn segment(args: &ArgMatches) -> Segment {
     }
 }
 
-// fn truncate_dir(directory: PathBuf, truncate_to: u8) {
-
-// }
-
 #[cfg(test)]
 mod tests {
+    // TODO: Look into stubbing `env` so that tests can be run in parallel
     use super::*;
     use clap::{App, Arg};
+    use std::path::Path;
 
     #[test]
     fn truncate_home_dir() {
@@ -43,8 +41,23 @@ mod tests {
             .arg(Arg::with_name("status_code"))
             .get_matches_from(vec!["starship", "0"]);
 
-        env::set_current_dir("~/dev/");
+        let home_dir = dirs::home_dir().unwrap();
+        env::set_current_dir(&home_dir).unwrap();
+
         let segment = segment(&args);
-        assert_eq!(segment.style, Style::from(Color::Green));
+        assert_eq!(segment.value, "~");
+    }
+
+    #[test]
+    fn dont_truncate_non_home_dir() {
+        let args = App::new("starship")
+            .arg(Arg::with_name("status_code"))
+            .get_matches_from(vec!["starship", "0"]);
+
+        let root_dir = Path::new("/");
+        env::set_current_dir(&root_dir).unwrap();
+
+        let segment = segment(&args);
+        assert_eq!(segment.value, "/");
     }
 }