[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:
parent
c52cb99518
commit
068968bbf5
3 changed files with 19 additions and 34 deletions
|
@ -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
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue