bumblebee-status/tests/modules/test_cmus.py

# 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