From cc31a05c526ab7160ab675fa590f206a76b0380c Mon Sep 17 00:00:00 2001
From: Dominik Ritter <dritter03@googlemail.com>
Date: Wed, 18 Jul 2018 22:25:06 +0200
Subject: [PATCH] Add test for ram segment

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

diff --git a/test/segments/ram.spec b/test/segments/ram.spec
new file mode 100755
index 00000000..104b8c82
--- /dev/null
+++ b/test/segments/ram.spec
@@ -0,0 +1,62 @@
+#!/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/ram-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 testRamSegmentWorksOnOsx() {
+    local POWERLEVEL9K_LEFT_PROMPT_ELEMENTS=(ram)
+    local OS="OSX" # Fake OSX
+    alias vm_stat="echo 'Mach Virtual Memory Statistics: (page size of 4096 bytes)
+Pages free:                              299687.
+Pages active:                           1623792.
+Pages inactive:                         1313411.
+'"
+
+    assertEquals "%K{yellow} %F{black%}RAM %f%F{black}6.15G " "$(prompt_ram left 1 false ${FOLDER})"
+
+    unalias vm_stat
+}
+
+function testRamSegmentWorksOnBsd() {
+    local POWERLEVEL9K_LEFT_PROMPT_ELEMENTS=(ram)
+    local OS="BSD" # Fake BSD
+    mkdir -p var/run
+    echo "avail memory 5678B 299687 4444G 299" > var/run/dmesg.boot
+
+    assertEquals "%K{yellow} %F{black%}RAM %f%F{black}0.29M " "$(prompt_ram left 1 false ${FOLDER})"
+}
+
+function testRamSegmentWorksOnLinux() {
+    local POWERLEVEL9K_LEFT_PROMPT_ELEMENTS=(ram)
+    local OS="Linux" # Fake Linux
+    mkdir proc
+    echo "MemAvailable: 299687" > proc/meminfo
+
+    assertEquals "%K{yellow} %F{black%}RAM %f%F{black}0.29G " "$(prompt_ram left 1 false ${FOLDER})"
+}
+
+source shunit2/source/2.1/src/shunit2
\ No newline at end of file