Changeset 11815 for main/waeup.kofa/trunk/src/waeup/kofa/utils/tests

Timestamp:

24 Sep 2014, 15:06:49 (10 years ago)

Author:

uli

Message:

Add helper to check swap space.

File:

• main/waeup.kofa/trunk/src/waeup/kofa/utils/tests/test_utils.py (modified) (3 diffs)

main/waeup.kofa/trunk/src/waeup/kofa/utils/tests/test_utils.py

@@ -18 +18 @@
 ## Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
 ##
+import psutil
 import unittest
 from waeup.kofa.interfaces import IKofaUtils
@@ -26 +27 @@
 class KofaUtilsTestCase(unittest.TestCase):
 
+    def get_cleared_util(self):
+        # Helper: get a `KofUtil` instance with all values of
+        # SYSTEM_MAX_LOAD dict set to ``None``
+        util = KofaUtils()
+        for key, val in util.SYSTEM_MAX_LOAD.items():
+            util.SYSTEM_MAX_LOAD[key] = None
+        return util
+
     def test_iface(self):
         # KofaUtils fullfill IKofaUtils expectations
@@ -31 +40 @@
         verify.verifyClass(IKofaUtils, KofaUtils)
         verify.verifyObject(IKofaUtils, utils)
+
+    def test_expensive_actions_allowed_swap_none(self):
+        # unset swap maximum values make KofUtils ignore swap values
+        utils = self.get_cleared_util()
+        utils.SYSTEM_MAX_LOAD['swap-mem'] = None
+        assert utils.expensive_actions_allowed() == True
+
+    @unittest.skipIf(
+        psutil.swap_memory().percent >= 99.0,
+        reason="System swap use over 99%. Cannot set higher allowed value.")
+    def test_expensive_actions_allowed_swap_ok(self):
+        # We can react to high swap values
+        utils = self.get_cleared_util()
+        utils.SYSTEM_MAX_LOAD['swap-mem'] = 99.0    # positive number
+        assert utils.expensive_actions_allowed() == True
+        utils.SYSTEM_MAX_LOAD['swap-mem'] = -1.0    # negative number
+        assert utils.expensive_actions_allowed() == True
+
+    @unittest.skipIf(
+        not psutil.swap_memory().percent,
+        reason="Can test swapping behavior only if actually swapping")
+    def test_expensive_actions_allowed_swap_too_much(self):
+        # We can react if too much swap is used
+        utils = self.get_cleared_util()
+        utils.SYSTEM_MAX_LOAD['swap-mem'] = 0.0     # positive number
+        assert utils.expensive_actions_allowed() == False
+        utils.SYSTEM_MAX_LOAD['swap-mem'] = -100.0  # negative number
+        assert utils.expensive_actions_allowed() == False