From 1b7e41964c84b124e91fafec66230c7c678a4145 Mon Sep 17 00:00:00 2001
From: Dominik Ritter <dritter03@googlemail.com>
Date: Thu, 15 Nov 2018 01:55:42 +0100
Subject: [PATCH] Fix configuration errors in the tests

---
 test/segments/vcs-git.spec | 4 +++-
 test/segments/vcs-hg.spec  | 4 +++-
 2 files changed, 6 insertions(+), 2 deletions(-)

diff --git a/test/segments/vcs-git.spec b/test/segments/vcs-git.spec
index bb51766c..bb2aef5b 100755
--- a/test/segments/vcs-git.spec
+++ b/test/segments/vcs-git.spec
@@ -491,6 +491,8 @@ function testDetectingUntrackedFilesInCleanSubdirectoryWorks() {
 }
 
 function testBranchNameScriptingVulnerability() {
+  local -a POWERLEVEL9K_LEFT_PROMPT_ELEMENTS
+  POWERLEVEL9K_LEFT_PROMPT_ELEMENTS=(vcs)
   echo "#!/bin/sh\n\necho 'hacked'\n" > evil_script.sh
   chmod +x evil_script.sh
 
@@ -498,7 +500,7 @@ function testBranchNameScriptingVulnerability() {
   git add . 2>/dev/null
   git commit -m "Initial commit" >/dev/null
 
-  assertEquals "%K{002} %F{000} %f%F{000} \$(./evil_script.sh) %k%F{002}%f " "$(__p9k_build_left_prompt)"
+  assertEquals "%K{002} %F{000} %f%F{000} \$(./evil_script.sh) %k%F{002}%f " "$(build_left_prompt)"
 }
 
 source shunit2/shunit2
diff --git a/test/segments/vcs-hg.spec b/test/segments/vcs-hg.spec
index 53afbaac..de7c5d38 100755
--- a/test/segments/vcs-hg.spec
+++ b/test/segments/vcs-hg.spec
@@ -205,6 +205,8 @@ function testBookmarkIconWorks() {
 }
 
 function testBranchNameScriptingVulnerability() {
+  local -a POWERLEVEL9K_LEFT_PROMPT_ELEMENTS
+  POWERLEVEL9K_LEFT_PROMPT_ELEMENTS=(vcs)
   echo "#!/bin/sh\n\necho 'hacked'\n" > evil_script.sh
   chmod +x evil_script.sh
 
@@ -212,7 +214,7 @@ function testBranchNameScriptingVulnerability() {
   hg add . >/dev/null
   hg commit -m "Initial commit" >/dev/null
 
-  assertEquals "%K{002} %F{000} %f%F{000} \$(./evil_script.sh) %k%F{002}%f " "$(build_left_prompt)"
+  assertEquals "%K{002} %F{000} \$(./evil_script.sh) %k%F{002}%f " "$(build_left_prompt)"
 }
 
 source shunit2/shunit2