Fix Test for context segment

This fixes an error on travis, because there is no sudo allowed. We could allow it, but I think it is better to mock sudo anyway.
Powerlevel9k · Jul 30, 2018 · 5181483 · 5181483
1 parent 3b0397d
commit 5181483
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/test/segments/context.spec b/test/segments/context.spec
@@ -31,6 +31,9 @@ function testContextSegmentDoesNotGetRenderedWithDefaultUser() {
 }
 
 function testContextSegmentDoesGetRenderedWhenSshConnectionIsOpen() {
+    function sudo() {
+        return 0
+    }
     local SSH_CLIENT="putty"
     local -a POWERLEVEL9K_LEFT_PROMPT_ELEMENTS
     POWERLEVEL9K_LEFT_PROMPT_ELEMENTS=(context)
@@ -39,6 +42,8 @@ function testContextSegmentDoesGetRenderedWhenSshConnectionIsOpen() {
     source powerlevel9k.zsh-theme
 
     assertEquals "%K{black} %F{yellow}%n@%m %k%F{black}%f " "$(build_left_prompt)"
+
+    unfunction sudo
 }
 
 function testContextSegmentWithForeignUser() {