def test_determine_block_device_storage(self): _test_devices = {'a': '/dev/bcache0'} self.storage_list.side_effect = _test_devices.keys() self.storage_get.side_effect = lambda _, key: _test_devices.get(key) self.assertEqual(utils.determine_block_device(), '/dev/bcache0') self.config.assert_called_with('ephemeral-device') self.storage_get.assert_called_with('location', 'a') self.storage_list.assert_called_with('ephemeral-device')
def test_determine_block_device_storage(self): _test_devices = { 'a': '/dev/bcache0' } self.storage_list.side_effect = _test_devices.keys() self.storage_get.side_effect = lambda _, key: _test_devices.get(key) self.assertEqual(utils.determine_block_device(), '/dev/bcache0') self.config.assert_called_with('ephemeral-device') self.storage_get.assert_called_with('location', 'a') self.storage_list.assert_called_with('ephemeral-device')
def test_determine_block_device_none(self): self.storage_list.return_value = [] self.assertEqual(utils.determine_block_device(), None) self.config.assert_called_with('ephemeral-device') self.storage_list.assert_called_with('ephemeral-device')
def test_determine_block_device(self, mock_os): self.test_config.set('ephemeral-device', '/dev/sdd') mock_os.path.exists.return_value = True self.assertEqual(utils.determine_block_device(), '/dev/sdd') self.config.assert_called_with('ephemeral-device')