1 files changed, 30 insertions, 2 deletions
diff --git a/test/segments/command_execution_time.spec b/test/segments/command_execution_time.spec
index 7700aaa1..65623a00 100755
--- a/test/segments/command_execution_time.spec
+++ b/test/segments/command_execution_time.spec
@@ -7,14 +7,17 @@ SHUNIT_PARENT=$0
 
 function setUp() {
   export TERM="xterm-256color"
-  # Load Powerlevel9k
-  source powerlevel9k.zsh-theme
 }
 
 function testCommandExecutionTimeIsNotShownIfTimeIsBelowThreshold() {
   local -a POWERLEVEL9K_LEFT_PROMPT_ELEMENTS
   POWERLEVEL9K_LEFT_PROMPT_ELEMENTS=(custom_world command_execution_time)
   local POWERLEVEL9K_CUSTOM_WORLD='echo world'
+
+  # Load Powerlevel9k
+  source powerlevel9k.zsh-theme
+
+  # Override payload
   local _P9K_COMMAND_DURATION=2
 
   assertEquals "%K{white} %F{black}world %k%F{white}%f " "$(build_left_prompt)"
@@ -24,6 +27,11 @@ function testCommandExecutionTimeThresholdCouldBeChanged() {
   local -a POWERLEVEL9K_LEFT_PROMPT_ELEMENTS
   POWERLEVEL9K_LEFT_PROMPT_ELEMENTS=(command_execution_time)
   local POWERLEVEL9K_COMMAND_EXECUTION_TIME_THRESHOLD=1
+
+  # Load Powerlevel9k
+  source powerlevel9k.zsh-theme
+
+  # Override payload
   local _P9K_COMMAND_DURATION=2.03
 
   assertEquals "%K{red} %F{yellow1%}Dur %f%F{yellow1}2.03 %k%F{red}%f " "$(build_left_prompt)"
@@ -43,6 +51,11 @@ function testCommandExecutionTimePrecisionCouldBeChanged() {
   POWERLEVEL9K_LEFT_PROMPT_ELEMENTS=(command_execution_time)
   local POWERLEVEL9K_COMMAND_EXECUTION_TIME_THRESHOLD=0
   local POWERLEVEL9K_COMMAND_EXECUTION_TIME_PRECISION=4
+
+  # Load Powerlevel9k
+  source powerlevel9k.zsh-theme
+
+  # Override payload
   local _P9K_COMMAND_DURATION=0.0001
 
   assertEquals "%K{red} %F{yellow1%}Dur %f%F{yellow1}0.0001 %k%F{red}%f " "$(build_left_prompt)"
@@ -52,6 +65,11 @@ function testCommandExecutionTimePrecisionCouldBeSetToZero() {
   local -a POWERLEVEL9K_LEFT_PROMPT_ELEMENTS
   POWERLEVEL9K_LEFT_PROMPT_ELEMENTS=(command_execution_time)
   local POWERLEVEL9K_COMMAND_EXECUTION_TIME_PRECISION=0
+
+  # Load Powerlevel9k
+  source powerlevel9k.zsh-theme
+
+  # Override payload
   local _P9K_COMMAND_DURATION=23.5001
 
   assertEquals "%K{red} %F{yellow1%}Dur %f%F{yellow1}23 %k%F{red}%f " "$(build_left_prompt)"
@@ -60,6 +78,11 @@ function testCommandExecutionTimePrecisionCouldBeSetToZero() {
 function testCommandExecutionTimeIsFormattedHumandReadbleForMinuteLongCommand() {
   local -a POWERLEVEL9K_LEFT_PROMPT_ELEMENTS
   POWERLEVEL9K_LEFT_PROMPT_ELEMENTS=(command_execution_time)
+
+  # Load Powerlevel9k
+  source powerlevel9k.zsh-theme
+
+  # Override payload
   local _P9K_COMMAND_DURATION=180
 
   assertEquals "%K{red} %F{yellow1%}Dur %f%F{yellow1}03:00 %k%F{red}%f " "$(build_left_prompt)"
@@ -68,6 +91,11 @@ function testCommandExecutionTimeIsFormattedHumandReadbleForMinuteLongCommand()
 function testCommandExecutionTimeIsFormattedHumandReadbleForHourLongCommand() {
   local -a POWERLEVEL9K_LEFT_PROMPT_ELEMENTS
   POWERLEVEL9K_LEFT_PROMPT_ELEMENTS=(command_execution_time)
+
+  # Load Powerlevel9k
+  source powerlevel9k.zsh-theme
+
+  # Override payload
   local _P9K_COMMAND_DURATION=7200
 
   assertEquals "%K{red} %F{yellow1%}Dur %f%F{yellow1}02:00:00 %k%F{red}%f " "$(build_left_prompt)"