Prepare tests for performance optimizations

1 files changed, 9 insertions, 2 deletions

diff --git a/test/segments/kubecontext.spec b/test/segments/kubecontext.spec
index 46fe8724..746ab806 100755
--- a/test/segments/kubecontext.spec
+++ b/test/segments/kubecontext.spec
@@ -7,8 +7,6 @@ SHUNIT_PARENT=$0
 
 function setUp() {
   export TERM="xterm-256color"
-  # Load Powerlevel9k
-  source powerlevel9k.zsh-theme
 }
 
 function mockKubectl() {
@@ -70,6 +68,9 @@ function testKubeContext() {
   POWERLEVEL9K_LEFT_PROMPT_ELEMENTS=(kubecontext)
   alias kubectl=mockKubectl
 
+  # Load Powerlevel9k
+  source powerlevel9k.zsh-theme
+
   assertEquals "%K{magenta} %F{white%}⎈ %f%F{white}minikube/default %k%F{magenta}%f " "$(build_left_prompt)"
 
   unalias kubectl
@@ -79,6 +80,9 @@ function testKubeContextOtherNamespace() {
   POWERLEVEL9K_LEFT_PROMPT_ELEMENTS=(kubecontext)
   alias kubectl=mockKubectlOtherNamespace
 
+  # Load Powerlevel9k
+  source powerlevel9k.zsh-theme
+
   assertEquals "%K{magenta} %F{white%}⎈ %f%F{white}minikube/kube-system %k%F{magenta}%f " "$(build_left_prompt)"
 
   unalias kubectl
@@ -89,6 +93,9 @@ function testKubeContextPrintsNothingIfKubectlNotAvailable() {
   local POWERLEVEL9K_CUSTOM_WORLD='echo world'
   alias kubectl=noKubectl
 
+  # Load Powerlevel9k
+  source powerlevel9k.zsh-theme
+
   assertEquals "%K{white} %F{black}world %k%F{white}%f " "$(build_left_prompt)"
 
   unalias kubectl