Viewing file:  test_utils.py (2.21 KB) -rw-r--r--
Select action/file-type:
 (+) |  (+) |  (+) | Code (+) | Session (+) |  (+) | SDB (+) |  (+) |  (+) |  (+) |  (+) |  (+) |
import testtools
from mock import call
from probert import utils
from probert.tests.helpers import random_string, simple_mocked_open
class ProbertTestUtils(testtools.TestCase):
def setUp(self):
super(ProbertTestUtils, self).setUp()
def test_utils_dict_merge(self):
r1 = {'relations': ['m1', 'x1']}
r2 = {'relations': ['m2', 'x2']}
combined = {'relations': ['m1', 'm2', 'x1', 'x2']}
test_result = utils.dict_merge(r1, r2)
self.assertEqual(sorted(combined['relations']),
sorted(test_result['relations']))
def test_utils_dict_merge_lists(self):
r1 = ['m1', 'x1']
r2 = ['m2', 'x2']
combined = ['m1', 'm2', 'x1', 'x2']
test_result = utils.dict_merge(r1, r2)
self.assertEqual(sorted(combined), sorted(test_result))
def test_utils_dict_merge_dicts(self):
r1 = {'storage': {'/dev/sda': {'DEVTYPE': 'disk'}}}
r2 = {'storage': {'/dev/sda': {'ID_MODEL': 'AWESOME'}}}
combined = {
'storage': {
'/dev/sda': {
'DEVTYPE': 'disk',
'ID_MODEL': 'AWESOME',
}
}
}
test_result = utils.dict_merge(r1, r2)
self.assertEqual(sorted(combined), sorted(test_result))
def test_utils_read_sys_block_size_bytes(self):
devname = random_string()
expected_fname = '/sys/class/block/%s/size' % devname
expected_bytes = 10737418240
content = '20971520'
with simple_mocked_open(content=content) as m_open:
result = utils.read_sys_block_size_bytes(devname)
self.assertEqual(expected_bytes, result)
self.assertEqual([call(expected_fname)], m_open.call_args_list)
def test_utils_read_sys_block_size_bytes_strips_value(self):
devname = random_string()
expected_fname = '/sys/class/block/%s/size' % devname
expected_bytes = 10737418240
content = ' 20971520 \n '
with simple_mocked_open(content=content) as m_open:
result = utils.read_sys_block_size_bytes(devname)
self.assertEqual(expected_bytes, result)
self.assertEqual([call(expected_fname)], m_open.call_args_list)
|