l.behm/bumblebee-status
1
1
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

[tests/core/module] Add tests for widget retrieval by name

Browse source
This commit is contained in:
Tobias Witek 2020-03-05 21:17:06 +01:00
parent 28b891ef37
commit 47950240d0
1 changed files with 12 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
tests/core/test_module.py
View file

@ -16,8 +16,9 @@ class module(unittest.TestCase):
def setUp(self): def setUp(self):
self.invalidModuleName = 'invalid-module-name' self.invalidModuleName = 'invalid-module-name'
self.validModuleName = 'test' self.validModuleName = 'test'
self.someWidget = core.widget.Widget('randomeWidgetContent') self.someWidget = core.widget.Widget('randomeWidgetContent', name='A')
self.anotherWidget = core.widget.Widget('more Widget content') self.anotherWidget = core.widget.Widget('more Widget content', name='B')
self.unusedWidgetName = 'C'
def test_loadinvalid_module(self): def test_loadinvalid_module(self):
config = unittest.mock.MagicMock() config = unittest.mock.MagicMock()
@ -80,4 +81,13 @@ class module(unittest.TestCase):
self.assertEqual(1, len(module.widgets())) self.assertEqual(1, len(module.widgets()))
self.assertEqual('error: !!', module.widget().full_text()) self.assertEqual('error: !!', module.widget().full_text())
def test_get_widget_by_name(self):
cfg = core.config.Config([])
module = TestModule(config=cfg, widgets=[self.someWidget, self.anotherWidget])
self.assertEqual(self.someWidget, module.widget(self.someWidget.name()))
self.assertEqual(self.anotherWidget, module.widget(self.anotherWidget.name()))
self.assertEqual(None, module.widget(self.unusedWidgetName))
self.assertEqual(self.someWidget, module.widget())
# vim: tabstop=8 expandtab shiftwidth=4 softtabstop=4 # vim: tabstop=8 expandtab shiftwidth=4 softtabstop=4