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

Add unit tests

Browse source
This commit is contained in:
Thaynã Moretti 2020-08-30 12:49:33 -03:00
parent 49de0e520b
commit 0c8d682d62
1 changed files with 24 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

25
tests/modules/core/test_memory.py
View file

@ -15,6 +15,7 @@ def memory_widget(module):
def meminfo_mock( def meminfo_mock(
total, total,
available, available,
unit = 'kB',
free = 0, free = 0,
buffers = 0, buffers = 0,
cached = 0, cached = 0,
@ -31,7 +32,7 @@ def meminfo_mock(
] ]
for i, (key, value) in enumerate(states): for i, (key, value) in enumerate(states):
data.append('{}: {} kB'.format(key, value)) data.append('{}: {} {}'.format(key, value, unit))
return '\n'.join(data) return '\n'.join(data)
@ -118,3 +119,25 @@ class TestMemory(TestCase):
assert widget.full_text() == '50.0%' assert widget.full_text() == '50.0%'
assert module.state(widget) == None 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