Add unit tests

This commit is contained in:
Thaynã Moretti 2020-08-30 12:49:33 -03:00
parent 49de0e520b
commit 0c8d682d62

View file

@ -15,6 +15,7 @@ def memory_widget(module):
def meminfo_mock(
total,
available,
unit = 'kB',
free = 0,
buffers = 0,
cached = 0,
@ -31,7 +32,7 @@ def meminfo_mock(
]
for i, (key, value) in enumerate(states):
data.append('{}: {} kB'.format(key, value))
data.append('{}: {} {}'.format(key, value, unit))
return '\n'.join(data)
@ -118,3 +119,25 @@ class TestMemory(TestCase):
assert widget.full_text() == '50.0%'
assert module.state(widget) == None
@mock.patch('builtins.open', mock.mock_open(read_data=meminfo_mock(8196, 4096, 'mB')))
def test_mb_unit(self):
module = build_module()
module.update()
widget = memory_widget(module)
assert widget.full_text() == '4.00GiB/8.00GiB (50.02%)'
assert module.state(widget) == None
@mock.patch('builtins.open', mock.mock_open(read_data=meminfo_mock(2, 1, 'gB')))
def test_gb_unit(self):
module = build_module()
module.update()
widget = memory_widget(module)
assert widget.full_text() == '1.00GiB/2.00GiB (50.00%)'
assert module.state(widget) == None