From 2efbd7e1880f62801db2d3739e3d33b2dfc1b154 Mon Sep 17 00:00:00 2001
From: Dominik Ritter <dritter03@googlemail.com>
Date: Wed, 18 Jul 2018 22:20:50 +0200
Subject: Add tests for load segment

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

(limited to 'test')

diff --git a/test/segments/load.spec b/test/segments/load.spec
new file mode 100755
index 00000000..45e28db3
--- /dev/null
+++ b/test/segments/load.spec
@@ -0,0 +1,137 @@
+#!/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
+
+  P9K_HOME=$(pwd)
+  ### Test specific
+  # Create default folder and git init it.
+  FOLDER=/tmp/powerlevel9k-test/load-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
+}
+
+function testLoadSegmentWorksOnOsx() {
+    sysctl() {
+        if [[ "$*" == 'vm.loadavg' ]]; then
+            echo "vm.loadavg: { 1,38 1,45 2,16 }";
+        fi
+
+        if [[ "$*" == '-n hw.logicalcpu' ]]; then
+            echo "4";
+        fi
+    }
+
+    local OS="OSX" # Fake OSX
+    local POWERLEVEL9K_LOAD_WHICH=1
+
+    assertEquals "%K{green} %F{black%}L %f%F{black}1.38 " "$(prompt_load left 1 false ${FOLDER})"
+
+    unfunction sysctl
+}
+
+function testLoadSegmentWorksOnBsd() {
+    sysctl() {
+        if [[ "$*" == 'vm.loadavg' ]]; then
+            echo "vm.loadavg: { 1,38 1,45 2,16 }";
+        fi
+
+        if [[ "$*" == '-n hw.ncpu' ]]; then
+            echo "4";
+        fi
+    }
+
+    local OS="BSD" # Fake BSD
+    local POWERLEVEL9K_LOAD_WHICH=1
+
+    assertEquals "%K{green} %F{black%}L %f%F{black}1.38 " "$(prompt_load left 1 false ${FOLDER})"
+
+    unfunction sysctl
+}
+
+function testLoadSegmentWorksOnLinux() {
+    # Prepare loadavg
+    mkdir proc
+    echo "1.38 0.01 0.05 1/87 8641" > proc/loadavg
+
+    alias nproc="echo 4"
+
+    local OS="Linux" # Fake Linux
+    local POWERLEVEL9K_LOAD_WHICH=1
+
+    assertEquals "%K{green} %F{black%}L %f%F{black}1.38 " "$(prompt_load left 1 false ${FOLDER})"
+
+    unalias nproc
+}
+
+# Test normal state. This test is not OS specific.
+# We test it as the Linux version, but that
+# does not matter here.
+function testLoadSegmentNormalState() {
+    # Prepare loadavg
+    mkdir proc
+    echo "1.00 0.01 0.05 1/87 8641" > proc/loadavg
+
+    alias nproc="echo 4"
+
+    local OS="Linux" # Fake Linux
+    local POWERLEVEL9K_LOAD_WHICH=1
+
+    assertEquals "%K{green} %F{black%}L %f%F{black}1.00 " "$(prompt_load left 1 false ${FOLDER})"
+
+    unalias nproc
+}
+
+# Test warning state. This test is not OS specific.
+# We test it as the Linux version, but that
+# does not matter here.
+function testLoadSegmentWarningState() {
+    # Prepare loadavg
+    mkdir proc
+    echo "2.01 0.01 0.05 1/87 8641" > proc/loadavg
+
+    alias nproc="echo 4"
+
+    local OS="Linux" # Fake Linux
+    local POWERLEVEL9K_LOAD_WHICH=1
+
+    assertEquals "%K{yellow} %F{black%}L %f%F{black}2.01 " "$(prompt_load left 1 false ${FOLDER})"
+
+    unalias nproc
+}
+
+# Test critical state. This test is not OS specific.
+# We test it as the Linux version, but that
+# does not matter here.
+function testLoadSegmentCriticalState() {
+    # Prepare loadavg
+    mkdir proc
+    echo "2.81 0.01 0.05 1/87 8641" > proc/loadavg
+
+    alias nproc="echo 4"
+
+    local OS="Linux" # Fake Linux
+    local POWERLEVEL9K_LOAD_WHICH=1
+
+    assertEquals "%K{red} %F{black%}L %f%F{black}2.81 " "$(prompt_load left 1 false ${FOLDER})"
+
+    unalias nproc
+}
+
+source shunit2/source/2.1/src/shunit2
\ No newline at end of file
-- 
cgit v1.2.3