169 lines
6.7 KiB
Python
169 lines
6.7 KiB
Python
# Copyright 2012 NTT Data. All Rights Reserved.
|
|
# Copyright 2012 Yahoo! Inc. All Rights Reserved.
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License"); you may
|
|
# not use this file except in compliance with the License. You may obtain
|
|
# a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
|
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
|
# License for the specific language governing permissions and limitations
|
|
# under the License.
|
|
|
|
import contextlib
|
|
|
|
import mock
|
|
from oslo.config import cfg
|
|
|
|
from nova import exception
|
|
from nova.openstack.common import processutils
|
|
from nova import test
|
|
from nova import utils
|
|
from nova.virt.libvirt import lvm
|
|
from nova.virt.libvirt import utils as libvirt_utils
|
|
|
|
CONF = cfg.CONF
|
|
|
|
|
|
class LvmTestCase(test.NoDBTestCase):
|
|
def test_get_volume_size(self):
|
|
executes = []
|
|
|
|
def fake_execute(*cmd, **kwargs):
|
|
executes.append(cmd)
|
|
return 123456789, None
|
|
|
|
expected_commands = [('blockdev', '--getsize64', '/dev/foo')]
|
|
self.stubs.Set(utils, 'execute', fake_execute)
|
|
size = lvm.get_volume_size('/dev/foo')
|
|
self.assertEqual(expected_commands, executes)
|
|
self.assertEqual(size, 123456789)
|
|
|
|
def test_lvm_clear(self):
|
|
def fake_lvm_size(path):
|
|
return lvm_size
|
|
|
|
def fake_execute(*cmd, **kwargs):
|
|
executes.append(cmd)
|
|
|
|
self.stubs.Set(lvm, 'get_volume_size', fake_lvm_size)
|
|
self.stubs.Set(utils, 'execute', fake_execute)
|
|
|
|
# Test the correct dd commands are run for various sizes
|
|
lvm_size = 1
|
|
executes = []
|
|
expected_commands = [('dd', 'bs=1', 'if=/dev/zero', 'of=/dev/v1',
|
|
'seek=0', 'count=1', 'conv=fdatasync')]
|
|
lvm.clear_volume('/dev/v1')
|
|
self.assertEqual(expected_commands, executes)
|
|
|
|
lvm_size = 1024
|
|
executes = []
|
|
expected_commands = [('dd', 'bs=1024', 'if=/dev/zero', 'of=/dev/v2',
|
|
'seek=0', 'count=1', 'conv=fdatasync')]
|
|
lvm.clear_volume('/dev/v2')
|
|
self.assertEqual(expected_commands, executes)
|
|
|
|
lvm_size = 1025
|
|
executes = []
|
|
expected_commands = [('dd', 'bs=1024', 'if=/dev/zero', 'of=/dev/v3',
|
|
'seek=0', 'count=1', 'conv=fdatasync')]
|
|
expected_commands += [('dd', 'bs=1', 'if=/dev/zero', 'of=/dev/v3',
|
|
'seek=1024', 'count=1', 'conv=fdatasync')]
|
|
lvm.clear_volume('/dev/v3')
|
|
self.assertEqual(expected_commands, executes)
|
|
|
|
lvm_size = 1048576
|
|
executes = []
|
|
expected_commands = [('dd', 'bs=1048576', 'if=/dev/zero', 'of=/dev/v4',
|
|
'seek=0', 'count=1', 'oflag=direct')]
|
|
lvm.clear_volume('/dev/v4')
|
|
self.assertEqual(expected_commands, executes)
|
|
|
|
lvm_size = 1048577
|
|
executes = []
|
|
expected_commands = [('dd', 'bs=1048576', 'if=/dev/zero', 'of=/dev/v5',
|
|
'seek=0', 'count=1', 'oflag=direct')]
|
|
expected_commands += [('dd', 'bs=1', 'if=/dev/zero', 'of=/dev/v5',
|
|
'seek=1048576', 'count=1', 'conv=fdatasync')]
|
|
lvm.clear_volume('/dev/v5')
|
|
self.assertEqual(expected_commands, executes)
|
|
|
|
lvm_size = 1234567
|
|
executes = []
|
|
expected_commands = [('dd', 'bs=1048576', 'if=/dev/zero', 'of=/dev/v6',
|
|
'seek=0', 'count=1', 'oflag=direct')]
|
|
expected_commands += [('dd', 'bs=1024', 'if=/dev/zero', 'of=/dev/v6',
|
|
'seek=1024', 'count=181', 'conv=fdatasync')]
|
|
expected_commands += [('dd', 'bs=1', 'if=/dev/zero', 'of=/dev/v6',
|
|
'seek=1233920', 'count=647', 'conv=fdatasync')]
|
|
lvm.clear_volume('/dev/v6')
|
|
self.assertEqual(expected_commands, executes)
|
|
|
|
# Test volume_clear_size limits the size
|
|
lvm_size = 10485761
|
|
CONF.set_override('volume_clear_size', '1', 'libvirt')
|
|
executes = []
|
|
expected_commands = [('dd', 'bs=1048576', 'if=/dev/zero', 'of=/dev/v7',
|
|
'seek=0', 'count=1', 'oflag=direct')]
|
|
lvm.clear_volume('/dev/v7')
|
|
self.assertEqual(expected_commands, executes)
|
|
|
|
CONF.set_override('volume_clear_size', '2', 'libvirt')
|
|
lvm_size = 1048576
|
|
executes = []
|
|
expected_commands = [('dd', 'bs=1048576', 'if=/dev/zero', 'of=/dev/v9',
|
|
'seek=0', 'count=1', 'oflag=direct')]
|
|
lvm.clear_volume('/dev/v9')
|
|
self.assertEqual(expected_commands, executes)
|
|
|
|
# Test volume_clear=shred
|
|
CONF.set_override('volume_clear', 'shred', 'libvirt')
|
|
CONF.set_override('volume_clear_size', '0', 'libvirt')
|
|
lvm_size = 1048576
|
|
executes = []
|
|
expected_commands = [('shred', '-n3', '-s1048576', '/dev/va')]
|
|
lvm.clear_volume('/dev/va')
|
|
self.assertEqual(expected_commands, executes)
|
|
|
|
CONF.set_override('volume_clear', 'shred', 'libvirt')
|
|
CONF.set_override('volume_clear_size', '1', 'libvirt')
|
|
lvm_size = 10485761
|
|
executes = []
|
|
expected_commands = [('shred', '-n3', '-s1048576', '/dev/vb')]
|
|
lvm.clear_volume('/dev/vb')
|
|
self.assertEqual(expected_commands, executes)
|
|
|
|
# Test volume_clear=none does nothing
|
|
CONF.set_override('volume_clear', 'none', 'libvirt')
|
|
executes = []
|
|
expected_commands = []
|
|
lvm.clear_volume('/dev/vc')
|
|
self.assertEqual(expected_commands, executes)
|
|
|
|
# Test volume_clear=invalid falls back to the default 'zero'
|
|
CONF.set_override('volume_clear', 'invalid', 'libvirt')
|
|
lvm_size = 1
|
|
executes = []
|
|
expected_commands = [('dd', 'bs=1', 'if=/dev/zero', 'of=/dev/vd',
|
|
'seek=0', 'count=1', 'conv=fdatasync')]
|
|
lvm.clear_volume('/dev/vd')
|
|
self.assertEqual(expected_commands, executes)
|
|
|
|
def test_fail_remove_all_logical_volumes(self):
|
|
def fake_execute(*args, **kwargs):
|
|
if 'vol2' in args:
|
|
raise processutils.ProcessExecutionError('Error')
|
|
|
|
with contextlib.nested(
|
|
mock.patch.object(lvm, 'clear_volume'),
|
|
mock.patch.object(libvirt_utils, 'execute',
|
|
side_effect=fake_execute)) as (mock_clear, mock_execute):
|
|
self.assertRaises(exception.VolumesNotRemoved,
|
|
lvm.remove_volumes,
|
|
['vol1', 'vol2', 'vol3'])
|
|
self.assertEqual(3, mock_execute.call_count)
|