class SpokeLVMTest(unittest.TestCase):
def __init__(self, methodName):
"""Setup config data."""
unittest.TestCase.__init__(self, methodName)
common_config = '../../contrib/spoke.conf'
custom_config = '/tmp/spoke.conf'
config_files = (common_config, custom_config)
self.config = config.setup(config_files)
self.lv_units = self.config.get('LVM', 'lv_units')
self.lv_name = 'testlv01'
self.lv_size = 1
self.TWOPLACES = decimal.Decimal(10) ** -2 # same as Decimal('0.01')
self.lv_size_dec = str(decimal.Decimal(self.lv_size).quantize(self.TWOPLACES))
self.lv_size_fmt = self.lv_size_dec + self.lv_units.upper()
self.vg_name = test_vg_name
# If we don't get an empty volume group here, abort
try:
self.lvm = SpokeLVM(self.vg_name)
lv_list = self.lvm.get()['data']
except Exception, e:
print 'Exception'
print e
sys.exit(1)
if lv_list != []:
print 'Volume group %s is not empty' % self.vg_name
sys.exit(1)
class SpokeLVMTest(unittest.TestCase):
def __init__(self, methodName):
"""Setup config data."""
unittest.TestCase.__init__(self, methodName)
common_config = '../../contrib/spoke.conf'
custom_config = '/tmp/spoke.conf'
config_files = (common_config, custom_config)
self.config = config.setup(config_files)
self.lv_units = self.config.get('LVM', 'lv_units')
self.lv_name = 'testlv01'
self.lv_size = 1
self.TWOPLACES = decimal.Decimal(10)**-2 # same as Decimal('0.01')
self.lv_size_dec = str(
decimal.Decimal(self.lv_size).quantize(self.TWOPLACES))
self.lv_size_fmt = self.lv_size_dec + self.lv_units.upper()
self.vg_name = test_vg_name
# If we don't get an empty volume group here, abort
try:
self.lvm = SpokeLVM(self.vg_name)
lv_list = self.lvm.get()['data']
except Exception, e:
print 'Exception'
print e
sys.exit(1)
if lv_list != []:
print 'Volume group %s is not empty' % self.vg_name
sys.exit(1)
def search(vg_name, lv_name=None):
from spoke.lib.lvm import SpokeLVM
lv = SpokeLVM(vg_name)
try:
result = lv.get(lv_name)
except error.SpokeError as e:
result = common.handle_error(e)
return result
def search(vg_name, lv_name=None):
from spoke.lib.lvm import SpokeLVM
lv = SpokeLVM(vg_name)
try:
result = lv.get(lv_name)
except error.SpokeError as e:
result = common.handle_error(e)
return result
def search(lv_name=None, vg_name=None):
try:
conf = _spoke_config(_salt_config('config'))
if not vg_name:
vg_name = conf.get('LVM', 'lv_def_vg_name')
lv = SpokeLVM(vg_name)
result = lv.get(lv_name)
except error.SpokeError as e:
result = common.handle_error(e)
return result
def test_get_multiple_lvs(self):
"""Search for all LVs; return list."""
lv_name = 'testlv02'
lv_size = 2
lv_size_dec = str(decimal.Decimal(lv_size).quantize(self.TWOPLACES))
lv_size_fmt = lv_size_dec + self.lv_units.upper()
lvm = SpokeLVM(self.vg_name)
lvm.create(lv_name, lv_size)
result = lvm.get()['data']
expected_result = [{'lv_size':self.lv_size_fmt,
'lv_name':self.lv_name},
{'lv_size': lv_size_fmt,
'lv_name': lv_name}]
self.assertEqual(result, expected_result)
lvm.delete(lv_name)
def test_get_multiple_lvs(self):
"""Search for all LVs; return list."""
lv_name = 'testlv02'
lv_size = 2
lv_size_dec = str(decimal.Decimal(lv_size).quantize(self.TWOPLACES))
lv_size_fmt = lv_size_dec + self.lv_units.upper()
lvm = SpokeLVM(self.vg_name)
lvm.create(lv_name, lv_size)
result = lvm.get()['data']
expected_result = [{
'lv_size': self.lv_size_fmt,
'lv_name': self.lv_name
}, {
'lv_size': lv_size_fmt,
'lv_name': lv_name
}]
self.assertEqual(result, expected_result)
lvm.delete(lv_name)
