Prepare tests for performance optimizations

1 files changed, 12 insertions, 2 deletions

diff --git a/test/segments/go_version.spec b/test/segments/go_version.spec
index 1be2e8c8..6d8f13b9 100755
--- a/test/segments/go_version.spec
+++ b/test/segments/go_version.spec
@@ -7,8 +7,6 @@ SHUNIT_PARENT=$0
 
 function setUp() {
   export TERM="xterm-256color"
-  # Load Powerlevel9k
-  source powerlevel9k.zsh-theme
 }
 
 function mockGo() {
@@ -39,6 +37,9 @@ function testGo() {
   local -a POWERLEVEL9K_LEFT_PROMPT_ELEMENTS
   POWERLEVEL9K_LEFT_PROMPT_ELEMENTS=(go_version)
 
+  # Load Powerlevel9k
+  source powerlevel9k.zsh-theme
+
   local PWD="$HOME/go/src/github.com/bhilburn/powerlevel9k"
 
   assertEquals "%K{green} %F{grey93%} %f%F{255}go1.5.3 %k%F{green}%f " "$(build_left_prompt)"
@@ -52,6 +53,9 @@ function testGoSegmentPrintsNothingIfEmptyGopath() {
   local POWERLEVEL9K_CUSTOM_WORLD='echo world'
   alias go=mockGoEmptyGopath
 
+  # Load Powerlevel9k
+  source powerlevel9k.zsh-theme
+
   assertEquals "%K{white} %F{black}world %k%F{white}%f " "$(build_left_prompt)"
 }
 
@@ -61,6 +65,9 @@ function testGoSegmentPrintsNothingIfNotInGopath() {
   local POWERLEVEL9K_CUSTOM_WORLD='echo world'
   alias go=mockGo
 
+  # Load Powerlevel9k
+  source powerlevel9k.zsh-theme
+
   assertEquals "%K{white} %F{black}world %k%F{white}%f " "$(build_left_prompt)"
 }
 
@@ -70,6 +77,9 @@ function testGoSegmentPrintsNothingIfGoIsNotAvailable() {
   local POWERLEVEL9K_CUSTOM_WORLD='echo world'
   alias go=noGo
 
+  # Load Powerlevel9k
+  source powerlevel9k.zsh-theme
+
   assertEquals "%K{white} %F{black}world %k%F{white}%f " "$(build_left_prompt)"
 
   unalias go