From a7576074760eb773b2f7bbdfdad56f7dadae296c Mon Sep 17 00:00:00 2001
From: Dominik Ritter <dritter03@googlemail.com>
Date: Wed, 18 Jul 2018 22:16:40 +0200
Subject: [PATCH] Add tests for disk_usage segment

---
 test/segments/disk_usage.spec | 112 ++++++++++++++++++++++++++++++++++
 1 file changed, 112 insertions(+)
 create mode 100755 test/segments/disk_usage.spec

diff --git a/test/segments/disk_usage.spec b/test/segments/disk_usage.spec
new file mode 100755
index 00000000..574445d9
--- /dev/null
+++ b/test/segments/disk_usage.spec
@@ -0,0 +1,112 @@
+#!/usr/bin/env zsh
+#vim:ft=zsh ts=2 sw=2 sts=2 et fenc=utf-8
+
+# Required for shunit2 to run correctly
+setopt shwordsplit
+SHUNIT_PARENT=$0
+
+function setUp() {
+  export TERM="xterm-256color"
+  # Load Powerlevel9k
+  source powerlevel9k.zsh-theme
+
+  # Test specific
+  P9K_HOME=$(pwd)
+  FOLDER=/tmp/powerlevel9k-test
+  mkdir -p $FOLDER
+  cd $FOLDER
+}
+
+function tearDown() {
+  # Go back to powerlevel9k folder
+  cd "${P9K_HOME}"
+  # Remove eventually created test-specific folder
+  rm -fr "${FOLDER}"
+  # At least remove test folder completely
+  rm -fr /tmp/powerlevel9k-test
+  unset FOLDER
+  unset P9K_HOME
+
+  # Remove IP cache file
+  rm -f ${POWERLEVEL9K_PUBLIC_IP_FILE}
+}
+
+function testDiskUsageSegmentWhenDiskIsAlmostFull() {
+  df() {
+      echo "Filesystem     1K-blocks      Used Available Use% Mounted on
+/dev/disk1     487219288 471466944  15496344  97% /"
+  }
+  local POWERLEVEL9K_LEFT_PROMPT_ELEMENTS=(disk_usage)
+
+  assertEquals "%K{red} %F{white%}hdd  %f%F{white}97%% %k%F{red}%f " "$(build_left_prompt)"
+
+  unfunction df
+}
+
+function testDiskUsageSegmentWhenDiskIsVeryFull() {
+  df() {
+      echo "Filesystem     1K-blocks      Used Available Use% Mounted on
+/dev/disk1     487219288 471466944  15496344  94% /"
+  }
+  local POWERLEVEL9K_LEFT_PROMPT_ELEMENTS=(disk_usage)
+
+  assertEquals "%K{yellow} %F{black%}hdd  %f%F{black}94%% %k%F{yellow}%f " "$(build_left_prompt)"
+
+  unfunction df
+}
+
+function testDiskUsageSegmentWhenDiskIsQuiteEmpty() {
+  df() {
+      echo "Filesystem     1K-blocks      Used Available Use% Mounted on
+/dev/disk1     487219288 471466944  15496344  4% /"
+  }
+  local POWERLEVEL9K_LEFT_PROMPT_ELEMENTS=(disk_usage)
+
+  assertEquals "%K{black} %F{yellow%}hdd  %f%F{yellow}4%% %k%F{black}%f " "$(build_left_prompt)"
+
+  unfunction df
+}
+
+function testDiskUsageSegmentPrintsNothingIfDiskIsQuiteEmptyAndOnlyWarningsShouldBeDisplayed() {
+  df() {
+      echo "Filesystem     1K-blocks      Used Available Use% Mounted on
+/dev/disk1     487219288 471466944  15496344  4% /"
+  }
+
+  local POWERLEVEL9K_LEFT_PROMPT_ELEMENTS=(disk_usage custom_world)
+  local POWERLEVEL9K_DISK_USAGE_ONLY_WARNING=true
+  local POWERLEVEL9K_CUSTOM_WORLD='echo world'
+
+  assertEquals "%K{white} %F{black}world %k%F{white}%f " "$(build_left_prompt)"
+
+  unfunction df
+}
+
+function testDiskUsageSegmentWarningLevelCouldBeAdjusted() {
+  df() {
+    echo "Filesystem     1K-blocks      Used Available Use% Mounted on
+/dev/disk1     487219288 471466944  15496344  11% /"
+  }
+  local POWERLEVEL9K_LEFT_PROMPT_ELEMENTS=(disk_usage)
+  local POWERLEVEL9K_DISK_USAGE_WARNING_LEVEL=10
+
+  assertEquals "%K{yellow} %F{black%}hdd  %f%F{black}11%% %k%F{yellow}%f " "$(build_left_prompt)"
+
+  unfunction df
+}
+
+function testDiskUsageSegmentCriticalLevelCouldBeAdjusted() {
+  df() {
+    echo "Filesystem     1K-blocks      Used Available Use% Mounted on
+/dev/disk1     487219288 471466944  15496344  11% /"
+  }
+  local POWERLEVEL9K_LEFT_PROMPT_ELEMENTS=(disk_usage)
+  local POWERLEVEL9K_DISK_USAGE_WARNING_LEVEL=5
+  local POWERLEVEL9K_DISK_USAGE_CRITICAL_LEVEL=10
+
+  assertEquals "%K{red} %F{white%}hdd  %f%F{white}11%% %k%F{red}%f " "$(build_left_prompt)"
+
+  unfunction df
+}
+
+source shunit2/source/2.1/src/shunit2
\ No newline at end of file