[tests/cmus] Add unit tests for cmus module

tests/modules/test_cmus.py

@@ -0,0 +1,118 @@
+# pylint: disable=C0103,C0111
+
+import mock
+import unittest
+
+import tests.mocks as mocks
+
+from bumblebee.config import Config
+from bumblebee.input import I3BarInput, LEFT_MOUSE
+from bumblebee.modules.cmus import Module
+
+class TestCmusModule(unittest.TestCase):
+    def setUp(self):
+        self._stdin, self._select, self.stdin, self.select = mocks.epoll_mock("bumblebee.input")
+
+        self.popen = mocks.MockPopen()
+
+        self.config = Config()
+        self.input = I3BarInput()
+        self.engine = mock.Mock()
+        self.engine.input = self.input
+        self.input.need_event = True
+
+        self.module = Module(engine=self.engine, config={ "config": self.config })
+        for widget in self.module.widgets():
+            widget.link_module(self.module)
+            self.anyWidget = widget
+
+        self.songTemplate = """
+status {status}
+file /path/to/file
+duration {duration}
+position {position}
+tag title {title}
+tag artist {artist}
+tag album {album}
+tag tracknumber 1
+tag date 1984
+tag comment comment
+        """
+
+    def tearDown(self):
+        self._stdin.stop()
+        self._select.stop()
+        self.popen.cleanup()
+
+    def test_read_song(self):
+        self.popen.mock.communicate.return_value = ("song", None)
+        self.module.update_all()
+        self.popen.assert_call("cmus-remote -Q")
+
+    def test_handle_runtimeerror(self):
+        self.popen.mock.communicate.side_effect = RuntimeError("error loading song")
+        self.module.update_all()
+        self.assertEquals(self.module.description(self.anyWidget), " -  /")
+
+    def test_format(self):
+        self.popen.mock.communicate.return_value = (self.songTemplate.format(
+            artist="an artist", title="a title", duration="100", position="20",
+            album="an album", status="irrelevant"
+        ), None)
+        self.module.update_all()
+        self.assertEquals(self.module.description(self.anyWidget),
+            "an artist - a title 00:20/01:40"
+        )
+
+    def test_repeat(self):
+        self.popen.mock.communicate.return_value = ("set repeat false", None)
+        self.module.update_all()
+        self.assertTrue("repeat-off" in self.module.state(self.module.widget("cmus.repeat")))
+        self.popen.mock.communicate.return_value = ("set repeat true", None)
+        self.module.update_all()
+        self.assertTrue("repeat-on" in self.module.state(self.module.widget("cmus.repeat")))
+
+    def test_shuffle(self):
+        self.popen.mock.communicate.return_value = ("set shuffle false", None)
+        self.module.update_all()
+        self.assertTrue("shuffle-off" in self.module.state(self.module.widget("cmus.shuffle")))
+        self.popen.mock.communicate.return_value = ("set shuffle true", None)
+        self.module.update_all()
+        self.assertTrue("shuffle-on" in self.module.state(self.module.widget("cmus.shuffle")))
+
+    def test_prevnext(self):
+        self.assertTrue("prev" in self.module.state(self.module.widget("cmus.prev")))
+        self.assertTrue("next" in self.module.state(self.module.widget("cmus.next")))
+
+    def test_main(self):
+        self.popen.mock.communicate.return_value = ("status paused", None)
+        self.module.update_all()
+        self.assertTrue("paused" in self.module.state(self.module.widget("cmus.main")))
+
+        self.popen.mock.communicate.return_value = ("status playing", None)
+        self.module.update_all()
+        self.assertTrue("playing" in self.module.state(self.module.widget("cmus.main")))
+
+        self.popen.mock.communicate.return_value = ("status stopped", None)
+        self.module.update_all()
+        self.assertTrue("stopped" in self.module.state(self.module.widget("cmus.main")))
+
+    def test_widget(self):
+        self.assertEquals(len(self.module.widgets()), 5)
+
+        for idx, val in enumerate(["prev", "main", "next", "shuffle", "repeat"]):
+            self.assertEquals(self.module.widgets()[idx].name, "cmus.{}".format(val))
+
+    def test_interaction(self):
+        events = [
+            {"widget": "cmus.shuffle", "action": "cmus-remote -S"},
+            {"widget": "cmus.repeat", "action": "cmus-remote -R"},
+            {"widget": "cmus.next", "action": "cmus-remote -n"},
+            {"widget": "cmus.prev", "action": "cmus-remote -r"},
+            {"widget": "cmus.main", "action": "cmus-remote -u"},
+        ]
+        for event in events:
+            mocks.mouseEvent(stdin=self.stdin, inp=self.input, module=self.module, instance=self.module.widget(event["widget"]).id, button=LEFT_MOUSE)
+            self.popen.assert_call(event["action"])
+
+# vim: tabstop=8 expandtab shiftwidth=4 softtabstop=4