OXIESEC PANEL

Name	Size	Modified	Perms
📁 ..	-	04/29/2025 04:07:53 PM	rwxr-xr-x
📄 __init__.py	0 bytes	08/04/2020 03:14:28 PM	rw-r--r--
📁 __pycache__	-	04/29/2025 04:07:53 PM	rwxr-xr-x
📁 data	-	04/29/2025 04:07:53 PM	rwxr-xr-x
📄 fakes.py	201 bytes	08/04/2020 03:14:28 PM	rw-r--r--
📄 helpers.py	1.56 KB	08/04/2020 03:14:28 PM	rw-r--r--
📄 test_dasd.py	7.06 KB	08/04/2020 03:14:28 PM	rw-r--r--
📄 test_lvm.py	12.19 KB	08/04/2020 03:14:28 PM	rw-r--r--
📄 test_multipath.py	3.56 KB	08/04/2020 03:14:28 PM	rw-r--r--
📄 test_prober.py	1.37 KB	08/04/2020 03:14:28 PM	rw-r--r--
📄 test_storage.py	1.89 KB	08/04/2020 03:14:28 PM	rw-r--r--
📄 test_utils.py	2.21 KB	08/04/2020 03:14:28 PM	rw-r--r--

Editing: test_utils.py

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)