Added function to add swap file

Author: jasonheecs

setupLibrary.sh

@@ -39,4 +39,27 @@ function changeSSHConfig() {
 function setupUfw() {
     sudo ufw allow OpenSSH
     yes y | sudo ufw enable
+}
+
+function createSwap() {
+   local swapmem=$(($(getPhysicalMemory) * 2))
+
+   # Anything over 4GB in swap is probably unnecessary as a RAM fallback
+   if [[ ${swapmem} > 4 ]]; then
+        phymem=4
+   fi
+
+   sudo fallocate -l ${swapmem}G /swapfile
+   sudo chmod 600 /swapfile
+   sudo mkswap /swapfile
+   sudo swapon /swapfile
+}
+
+function getPhysicalMemory() {
+    local phymem=$(free -g|awk '/^Mem:/{print $2}')
+    if [[ ${phymem} == '0' ]]; then
+        echo 1
+    else
+        echo ${phymem}
+    fi
 }

tests/tests.sh

@@ -56,15 +56,21 @@ function testUfw() {
     assertContains "OpenSSH" "${ufw_status}"
 }
 
+function testSwap() {
+    createSwap
+
+    assertContains "/swapfile" "$(ls -lh /swapfile)"
+    assertContains "/swapfile" "$(sudo swapon --show)"
+}
+
 function testTeardown () {
     echo "Test Teardown"
 
     deleteTestUser
     revertSudoers
     revertSSHConfig
-
-    sudo ufw delete allow OpenSSH
-    sudo ufw disable
+    revertUfw
+    deleteSwap
 }
 
 ### Helper Functions ###
@@ -89,4 +95,14 @@ function revertSSHConfig() {
     sudo rm -rf /etc/ssh/sshd_config.old
 }
 
+function revertUfw() {
+    sudo ufw delete allow OpenSSH
+    sudo ufw disable
+}
+
+function deleteSwap() {
+    sudo swapoff /swapfile
+    sudo rm /swapfile
+}
+
 runUnitTests