def test__get_vmedia_params_by_device(self, execute_mock, mkdir_mock, exists_mock, read_params_mock, get_device_mock, mkdtemp_mock, rmtree_mock): mkdtemp_mock.return_value = "/tempdir" null_output = ["", ""] expected_params = {'a': 'b'} read_params_mock.return_value = expected_params exists_mock.side_effect = [False, False] execute_mock.side_effect = [null_output, null_output] get_device_mock.return_value = "sda" returned_params = utils._get_vmedia_params() exists_mock.assert_has_calls([ mock.call("/dev/disk/by-label/ir-vfd-dev"), mock.call("/dev/disk/by-label/IR-VFD-DEV") ]) execute_mock.assert_any_call('mount', "/dev/sda", "/tempdir") read_params_mock.assert_called_once_with("/tempdir/parameters.txt") execute_mock.assert_any_call('umount', "/tempdir") self.assertEqual(expected_params, returned_params) mkdtemp_mock.assert_called_once_with() rmtree_mock.assert_called_once_with("/tempdir")
def test__get_vmedia_params_umount_fails(self, execute_mock, mkdir_mock, exists_mock, read_params_mock, get_device_mock, mkdtemp_mock, rmtree_mock): mkdtemp_mock.return_value = "/tempdir" null_output = ["", ""] expected_params = {'a': 'b'} exists_mock.return_value = True read_params_mock.return_value = expected_params get_device_mock.return_value = "sda" execute_mock.side_effect = [ null_output, processutils.ProcessExecutionError() ] returned_params = utils._get_vmedia_params() execute_mock.assert_any_call('mount', "/dev/disk/by-label/ir-vfd-dev", "/tempdir") read_params_mock.assert_called_once_with("/tempdir/parameters.txt") execute_mock.assert_any_call('umount', "/tempdir") self.assertEqual(expected_params, returned_params) mkdtemp_mock.assert_called_once_with() rmtree_mock.assert_called_once_with("/tempdir")
def test__get_vmedia_params_by_device(self, execute_mock, mkdir_mock, exists_mock, read_params_mock, get_device_mock, mkdtemp_mock, rmtree_mock): mkdtemp_mock.return_value = "/tempdir" null_output = ["", ""] expected_params = {'a': 'b'} read_params_mock.return_value = expected_params exists_mock.side_effect = [False, False] execute_mock.side_effect = [null_output, null_output] get_device_mock.return_value = "sda" returned_params = utils._get_vmedia_params() exists_mock.assert_has_calls( [mock.call("/dev/disk/by-label/ir-vfd-dev"), mock.call("/dev/disk/by-label/IR-VFD-DEV")]) execute_mock.assert_any_call('mount', "/dev/sda", "/tempdir") read_params_mock.assert_called_once_with("/tempdir/parameters.txt") execute_mock.assert_any_call('umount', "/tempdir") self.assertEqual(expected_params, returned_params) mkdtemp_mock.assert_called_once_with() rmtree_mock.assert_called_once_with("/tempdir")
def test__get_vmedia_params( self, execute_mock, mkdir_mock, read_params_mock, mkdtemp_mock, rmtree_mock, find_mock): mkdtemp_mock.return_value = "/tempdir" find_mock.return_value = '/dev/fake' null_output = ["", ""] expected_params = {'a': 'b'} read_params_mock.return_value = expected_params execute_mock.side_effect = [null_output, null_output] returned_params = utils._get_vmedia_params() execute_mock.assert_any_call('mount', "/dev/fake", "/tempdir") read_params_mock.assert_called_once_with("/tempdir/parameters.txt") execute_mock.assert_any_call('umount', "/tempdir") self.assertEqual(expected_params, returned_params) mkdtemp_mock.assert_called_once_with() rmtree_mock.assert_called_once_with("/tempdir")
def test__get_vmedia_params_by_label(self, execute_mock, mkdir_mock, exists_mock, read_params_mock, mkdtemp_mock, rmtree_mock): mkdtemp_mock.return_value = "/tempdir" null_output = ["", ""] expected_params = {'a': 'b'} read_params_mock.return_value = expected_params exists_mock.return_value = True execute_mock.side_effect = [null_output, null_output] returned_params = utils._get_vmedia_params() execute_mock.assert_any_call('mount', "/dev/disk/by-label/ir-vfd-dev", "/tempdir") read_params_mock.assert_called_once_with("/tempdir/parameters.txt") exists_mock.assert_called_once_with("/dev/disk/by-label/ir-vfd-dev") execute_mock.assert_any_call('umount', "/tempdir") self.assertEqual(expected_params, returned_params) mkdtemp_mock.assert_called_once_with() rmtree_mock.assert_called_once_with("/tempdir")
def test__get_vmedia_params_by_label_lower_case( self, execute_mock, mkdir_mock, exists_mock, read_params_mock, mkdtemp_mock, rmtree_mock): mkdtemp_mock.return_value = "/tempdir" null_output = ["", ""] expected_params = {'a': 'b'} read_params_mock.return_value = expected_params exists_mock.side_effect = [True, False] execute_mock.side_effect = [null_output, null_output] returned_params = utils._get_vmedia_params() execute_mock.assert_any_call('mount', "/dev/disk/by-label/ir-vfd-dev", "/tempdir") read_params_mock.assert_called_once_with("/tempdir/parameters.txt") exists_mock.assert_called_once_with("/dev/disk/by-label/ir-vfd-dev") execute_mock.assert_any_call('umount', "/tempdir") self.assertEqual(expected_params, returned_params) mkdtemp_mock.assert_called_once_with() rmtree_mock.assert_called_once_with("/tempdir")
def test__get_vmedia_params_rmtree_fails(self, execute_mock, mkdir_mock, exists_mock, read_params_mock, get_device_mock, mkdtemp_mock, rmtree_mock): mkdtemp_mock.return_value = "/tempdir" rmtree_mock.side_effect = Exception null_output = ["", ""] expected_params = {'a': 'b'} exists_mock.return_value = True read_params_mock.return_value = expected_params get_device_mock.return_value = "sda" execute_mock.return_value = null_output returned_params = utils._get_vmedia_params() execute_mock.assert_any_call('mount', "/dev/disk/by-label/ir-vfd-dev", "/tempdir") read_params_mock.assert_called_once_with("/tempdir/parameters.txt") execute_mock.assert_any_call('umount', "/tempdir") self.assertEqual(expected_params, returned_params) mkdtemp_mock.assert_called_once_with() rmtree_mock.assert_called_once_with("/tempdir")