81bb72cf69
Added required code changes to let mypy pass when running from the toplevel kiwi namespace. This now finally Fixes #1644
72 lines
2.2 KiB
Python
72 lines
2.2 KiB
Python
import logging
|
|
from mock import patch
|
|
from pytest import (
|
|
raises, fixture
|
|
)
|
|
|
|
from kiwi.storage.loop_device import LoopDevice
|
|
|
|
from kiwi.exceptions import KiwiLoopSetupError
|
|
|
|
|
|
class TestLoopDevice:
|
|
@fixture(autouse=True)
|
|
def inject_fixtures(self, caplog):
|
|
self._caplog = caplog
|
|
|
|
@patch('os.path.exists')
|
|
def setup(self, mock_exists):
|
|
mock_exists.return_value = False
|
|
self.loop = LoopDevice('loop-file', 20, 4096)
|
|
|
|
def test_loop_setup_invalid(self):
|
|
with raises(KiwiLoopSetupError):
|
|
LoopDevice('loop-file-does-not-exist-and-no-size-given')
|
|
|
|
def test_get_device(self):
|
|
assert self.loop.get_device() == ''
|
|
|
|
def test_is_loop(self):
|
|
assert self.loop.is_loop() is True
|
|
|
|
@patch('os.path.exists')
|
|
@patch('kiwi.storage.loop_device.Command.run')
|
|
@patch('kiwi.storage.loop_device.CommandCapabilities.has_option_in_help')
|
|
def test_create(
|
|
self, mock_has_option_in_help, mock_command, mock_exists
|
|
):
|
|
mock_has_option_in_help.return_value = True
|
|
mock_exists.return_value = False
|
|
self.loop.create()
|
|
call = mock_command.call_args_list[0]
|
|
assert mock_command.call_args_list[0] == \
|
|
call([
|
|
'qemu-img', 'create', 'loop-file', '20M'
|
|
])
|
|
call = mock_command.call_args_list[1]
|
|
assert mock_command.call_args_list[1] == \
|
|
call([
|
|
'losetup', '--sector-size', '4096',
|
|
'-f', '--show', 'loop-file'
|
|
])
|
|
mock_has_option_in_help.return_value = False
|
|
mock_command.reset_mock()
|
|
self.loop.create()
|
|
assert mock_command.call_args_list[1] == \
|
|
call([
|
|
'losetup', '--logical-blocksize', '4096',
|
|
'-f', '--show', 'loop-file'
|
|
])
|
|
self.loop.node_name = None
|
|
|
|
@patch('kiwi.storage.loop_device.Command.run')
|
|
def test_destructor(self, mock_command):
|
|
self.loop.node_name = '/dev/loop0'
|
|
mock_command.side_effect = Exception
|
|
self.loop.__del__()
|
|
with self._caplog.at_level(logging.WARNING):
|
|
mock_command.assert_called_once_with(
|
|
['losetup', '-d', '/dev/loop0']
|
|
)
|
|
self.loop.node_name = None
|