From f12ff18385bf5d3217dea76a0771620aa3fce89f Mon Sep 17 00:00:00 2001
From: Jan De Dobbeleer <jan.de.dobbeleer@gmail.com>
Date: Fri, 5 Mar 2021 17:11:06 +0100
Subject: [PATCH] refactor(test): mapped locations on paths

relates to #484
---
 src/segment_path_test.go | 18 +++++++++++++-----
 1 file changed, 13 insertions(+), 5 deletions(-)

diff --git a/src/segment_path_test.go b/src/segment_path_test.go
index c22ba8fb3983..4294ebcef8ca 100644
--- a/src/segment_path_test.go
+++ b/src/segment_path_test.go
@@ -309,33 +309,38 @@ func TestAgnosterPathStyles(t *testing.T) {
 
 func TestGetFullPath(t *testing.T) {
 	cases := []struct {
-		Style               string
-		FolderSeparatorIcon string
-		Pwd                 string
-		Pswd                string
-		Expected            string
+		Style                  string
+		FolderSeparatorIcon    string
+		Pwd                    string
+		Pswd                   string
+		Expected               string
+		DisableMappedLocations bool
 	}{
 		{Style: Full, Pwd: "", Expected: ""},
 		{Style: Full, Pwd: "/", Expected: "/"},
 		{Style: Full, Pwd: "/usr/home", Expected: "~"},
 		{Style: Full, Pwd: "/usr/home/abc", Expected: "~/abc"},
+		{Style: Full, Pwd: "/usr/home/abc", Expected: "/usr/home/abc", DisableMappedLocations: true},
 		{Style: Full, Pwd: "/a/b/c/d", Expected: "/a/b/c/d"},
 
 		{Style: Full, FolderSeparatorIcon: "|", Pwd: "", Expected: ""},
 		{Style: Full, FolderSeparatorIcon: "|", Pwd: "/", Expected: "|"},
 		{Style: Full, FolderSeparatorIcon: "|", Pwd: "/usr/home", Expected: "~"},
+		{Style: Full, FolderSeparatorIcon: "|", Pwd: "/usr/home", Expected: "|usr|home", DisableMappedLocations: true},
 		{Style: Full, FolderSeparatorIcon: "|", Pwd: "/usr/home/abc", Expected: "~|abc"},
 		{Style: Full, FolderSeparatorIcon: "|", Pwd: "/a/b/c/d", Expected: "|a|b|c|d"},
 
 		{Style: Folder, Pwd: "", Expected: "."},
 		{Style: Folder, Pwd: "/", Expected: "/"},
 		{Style: Folder, Pwd: "/usr/home", Expected: "~"},
+		{Style: Folder, Pwd: "/usr/home", Expected: "home", DisableMappedLocations: true},
 		{Style: Folder, Pwd: "/usr/home/abc", Expected: "abc"},
 		{Style: Folder, Pwd: "/a/b/c/d", Expected: "d"},
 
 		{Style: Folder, FolderSeparatorIcon: "|", Pwd: "", Expected: "."},
 		{Style: Folder, FolderSeparatorIcon: "|", Pwd: "/", Expected: "|"},
 		{Style: Folder, FolderSeparatorIcon: "|", Pwd: "/usr/home", Expected: "~"},
+		{Style: Folder, FolderSeparatorIcon: "|", Pwd: "/usr/home", Expected: "home", DisableMappedLocations: true},
 		{Style: Folder, FolderSeparatorIcon: "|", Pwd: "/usr/home/abc", Expected: "abc"},
 		{Style: Folder, FolderSeparatorIcon: "|", Pwd: "/a/b/c/d", Expected: "d"},
 	}
@@ -357,6 +362,9 @@ func TestGetFullPath(t *testing.T) {
 		if tc.FolderSeparatorIcon != "" {
 			props.values[FolderSeparatorIcon] = tc.FolderSeparatorIcon
 		}
+		if tc.DisableMappedLocations {
+			props.values[MappedLocationsEnabled] = false
+		}
 		path := &path{
 			env:   env,
 			props: props,