118 lines
4.5 KiB
Python
118 lines
4.5 KiB
Python
# 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
|