Linux ip-172-26-7-228 5.4.0-1103-aws #111~18.04.1-Ubuntu SMP Tue May 23 20:04:10 UTC 2023 x86_64
Your IP : 3.140.198.3
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)
|