[tests] Refactor by removing getter/setter methods

Instead of having get/set methods for simple attributes, use the
attributes directly.

see #23
This commit is contained in:
Tobi-wan Kenobi 2016-12-09 11:42:02 +01:00
parent c52cb99518
commit 068968bbf5
3 changed files with 19 additions and 34 deletions

View file

@ -31,49 +31,34 @@ class MockOutput(object):
class MockWidget(object):
def __init__(self, text):
self._text = text
self._module = None
def set_module(self, name):
self._module = name
self.attr_module = None
def update(self, widgets):
pass
def module(self):
return self._module
return self.attr_module
def full_text(self):
return self._text
class MockTheme(object):
def __init__(self):
self._prefix = None
self._suffix = None
self._fg = None
self._bg = None
def set_prefix(self, value):
self._prefix = value
def set_suffix(self, value):
self._suffix = value
def set_fg(self, value):
self._fg = value
def set_bg(self, value):
self._bg = value
self.attr_prefix = None
self.attr_suffix = None
self.attr_fg = None
self.attr_bg = None
def prefix(self, widget):
return self._prefix
return self.attr_prefix
def suffix(self, widget):
return self._suffix
return self.attr_suffix
def fg(self, widget):
return self._fg
return self.attr_fg
def bg(self, widget):
return self._bg
return self.attr_bg
# vim: tabstop=8 expandtab shiftwidth=4 softtabstop=4