def _create_vol(self, lvm2):
"""
Creates mocked lvm volume with 2 pvs
"""
# storage2.volume('/dev/sdd')
lvm2.NotFound = Exception
lvs_returns = [
lvm2.NotFound, {
'data/vol1':
mock.Mock(lv_path='/dev/mapper/data-vol1', lv_attr='-wi-ao')
}
]
def lvs(*args, **kwds):
ret = lvs_returns.pop(0)
if isinstance(ret, Exception):
raise ret
return ret
lvm2.lvs.side_effect = lvs
lvm2.pvs.side_effect = lambda *args, **kwds: {
'/dev/sdb': mock.Mock(vg_name=None),
'/dev/sdc': mock.Mock(vg_name=None)
}
lvm2.vgs.side_effect = lvm2.NotFound
vol = lvm.LvmVolume(name='vol1',
vg='data',
pvs=['/dev/sdb', '/dev/sdc'],
size='98%FREE')
vol.ensure()
lvm2.reset_mock()
lvm2.lvs.side_effect = lvm2.pvs.side_effect = lvm2.vgs.side_effect = None
return vol
def test_extend_underlying_pvs(self, fs_fn, lvm2, mod_mount):
lvm2.pvs.side_effect = [{}, {'/dev/sdb': mock.Mock(vg_name='data')}]
lvm2.NotFound = Exception
lvm2.vgs.side_effect = lvm2.NotFound
lvs_returns = [
lvm2.NotFound, {
'data/vol1':
mock.Mock(lv_path='/dev/mapper/data-vol1', lv_attr='-wi-ao')
}
]
def lvs(*args, **kwds):
ret = lvs_returns.pop(0)
if isinstance(ret, Exception):
raise ret
return ret
lvm2.lvs.side_effect = lvs
vol = lvm.LvmVolume(name='vol1',
vg='data',
pvs=['/dev/sdb'],
size='98%FREE')
vol.ensure()
vol.pvs.append('/dev/sdc')
lvm2.pvs.side_effect = [{'/dev/sdb': mock.Mock(vg_name='data')}]*3 + \
[{'/dev/sdc': mock.Mock(vg_name=None)}]
lvs_returns.append({
'data/vol1':
mock.Mock(lv_path='/dev/mapper/data-vol1', lv_attr='-wi-ao')
})
fs = fs_fn.return_value
fs.features.get.return_value = True
with mock.patch.object(vol, 'is_fs_created') as f:
f.return_value = True
vol.ensure()
f.assert_called_once_with()
fs_fn.assert_called_once_with('ext3')
fs.features.get.assert_called_once_with('resizable')
lvm2.pvcreate.reset_mock
lvm2.pvcreate.assert_called_with('/dev/sdc')
lvm2.vgextend.assert_called_with('data', '/dev/sdc')
lvm2.lvextend.assert_called_once_with('/dev/mapper/data-vol1',
extents='98%FREE')
fs.resize.assert_called_once_with('/dev/mapper/data-vol1')
def test_ensure_new_with_existed_layout(self, lvm2, mod_mount):
lvm2.NotFound = Exception
lvs_returns = [
lvm2.NotFound, {
'data/vol1':
mock.Mock(lv_path='/dev/mapper/data-vol1', lv_attr='-wi---')
}
]
def lvs(*args, **kwds):
ret = lvs_returns.pop(0)
if isinstance(ret, Exception):
raise ret
return ret
lvm2.lvs.side_effect = lvs
lvm2.pvs.return_value = {
'/dev/sdb': mock.Mock(vg_name='data'),
'/dev/sdc': mock.Mock(vg_name='data')
}
vol = lvm.LvmVolume(name='vol1',
vg='data',
pvs=['/dev/sdb', '/dev/sdc'],
size='98%FREE')
vol.ensure()
assert len(vol.pvs) == 2
assert isinstance(vol.pvs[0], base.Volume)
assert isinstance(vol.pvs[1], base.Volume)
assert vol.device == '/dev/mapper/data-vol1'
assert lvm2.pvs.mock_calls == [
mock.call(),
mock.call(),
mock.call('/dev/sdb'),
mock.call('/dev/sdc')
]
assert lvm2.pvcreate.call_count == 0
lvm2.vgs.assert_called_once_with('data')
lvm2.lvchange.assert_called_once_with(vol.device, available='y')
def test_ensure_new(self, lvm2, mod_mount):
lvm2.NotFound = Exception
lvs_returns = [
lvm2.NotFound, {
'data/vol1':
mock.Mock(lv_path='/dev/mapper/data-vol1', lv_attr='-wi-ao')
}
]
def lvs(*args, **kwds):
ret = lvs_returns.pop(0)
if isinstance(ret, Exception):
raise ret
return ret
lvm2.lvs.side_effect = lvs
lvm2.pvs.side_effect = [{}, {}] + [{
'/dev/sdb': mock.Mock(vg_name=None)
}, {
'/dev/sdc': mock.Mock(vg_name=None)
}] * 2
lvm2.vgs.side_effect = lvm2.NotFound
vol = lvm.LvmVolume(name='vol1',
vg='data',
pvs=['/dev/sdb', '/dev/sdc'],
size='98%FREE')
vol.ensure()
assert len(vol.pvs) == 2
assert isinstance(vol.pvs[0], base.Volume)
assert isinstance(vol.pvs[1], base.Volume)
assert vol.device == '/dev/mapper/data-vol1'
calls = [mock.call('/dev/sdb'), mock.call('/dev/sdc')]
lvm2.pvcreate.assert_has_calls(calls)
lvm2.vgcreate.assert_called_once_with('data', '/dev/sdb', '/dev/sdc')
lvm2.lvcreate.assert_called_once_with('data',
name='vol1',
extents='98%FREE')