From e59a870a1e039e12da2097401f925146ecc1a5fb Mon Sep 17 00:00:00 2001
From: Tobi-wan Kenobi <github@tobi-wan-kenobi.at>
Date: Sun, 5 Mar 2017 13:58:15 +0100
Subject: [PATCH] [tests] Add unit tests for memory module

---
 tests/modules/test_memory.py | 52 ++++++++++++++++++++++++++++++++++++
 1 file changed, 52 insertions(+)
 create mode 100644 tests/modules/test_memory.py

diff --git a/tests/modules/test_memory.py b/tests/modules/test_memory.py
new file mode 100644
index 0000000..94c61f9
--- /dev/null
+++ b/tests/modules/test_memory.py
@@ -0,0 +1,52 @@
+# pylint: disable=C0103,C0111
+
+import mock
+import unittest
+
+import tests.mocks as mocks
+
+from bumblebee.input import LEFT_MOUSE
+from bumblebee.modules.memory import Module
+
+class VirtualMemory(object):
+    def __init__(self, percent):
+        self.percent = percent
+
+class TestMemoryModule(unittest.TestCase):
+    def setUp(self):
+        mocks.setup_test(self, Module)
+        self._psutil = mock.patch("bumblebee.modules.memory.psutil")
+        self.psutil = self._psutil.start()
+
+    def tearDown(self):
+        self._psutil.stop()
+        mocks.teardown_test(self)
+
+    def test_leftclick(self):
+        mocks.mouseEvent(stdin=self.stdin, button=LEFT_MOUSE, inp=self.input, module=self.module)
+        self.popen.assert_call("gnome-system-monitor")
+
+    def test_warning(self):
+        self.config.set("memory.critical", "80")
+        self.config.set("memory.warning", "70")
+        self.psutil.virtual_memory.return_value = VirtualMemory(75)
+        self.module.update_all()
+        self.assertTrue("warning" in self.module.state(self.anyWidget))
+
+    def test_critical(self):
+        self.config.set("memory.critical", "80")
+        self.config.set("memory.warning", "70")
+        self.psutil.virtual_memory.return_value = VirtualMemory(81)
+        self.module.update_all()
+        self.assertTrue("critical" in self.module.state(self.anyWidget))
+
+    def test_usage(self):
+        rv = VirtualMemory(50)
+        rv.total = 1000
+        rv.available = 500
+        self.psutil.virtual_memory.return_value = rv
+        self.module.update_all()
+        self.assertEquals("500.00B/1000.00B (50.00%)", self.module.memory_usage(self.anyWidget))
+        self.assertEquals(None, self.module.state(self.anyWidget))
+
+# vim: tabstop=8 expandtab shiftwidth=4 softtabstop=4