120 lines
3.6 KiB
Python
120 lines
3.6 KiB
Python
# Copyright 2013, 2014 IBM Corp.
|
|
from oslo import i18n
|
|
i18n.install('cinder')
|
|
import unittest
|
|
import mox
|
|
from powervc.volume.manager.manager import PowerVCCinderManager
|
|
from powervc.volume.driver.service import PowerVCService
|
|
|
|
fake_volume_type = {'id': '',
|
|
'name': 'fake_volume_type'
|
|
}
|
|
|
|
|
|
fake_volume = {'display_name': 'fake_volume',
|
|
'display_description': 'This is a fake volume',
|
|
'volume_type_id': '',
|
|
'status': '',
|
|
'host': 'powervc',
|
|
'size': 1,
|
|
'availability_zone': 'nova',
|
|
'bootable': 0,
|
|
'snapshot_id': '',
|
|
'source_volid': '',
|
|
'metadata': {},
|
|
'project_id': 'admin',
|
|
'user_id': 'admin',
|
|
'attached_host': 'fake_attached_host',
|
|
'mountpoint': '',
|
|
'instance_uuid': '',
|
|
'attach_status': ''}
|
|
|
|
fake_payload = {'volume_id': '', 'display_name': ''}
|
|
|
|
fake_context = {}
|
|
|
|
|
|
class FakeDBVolume():
|
|
def __init__(self):
|
|
pass
|
|
|
|
|
|
class FakeVolume():
|
|
def __init__(self):
|
|
pass
|
|
|
|
def __dict__(self):
|
|
return None
|
|
|
|
|
|
class FakePowerVCService(PowerVCService):
|
|
def __init__(self):
|
|
pass
|
|
|
|
fake_db_volume = FakeDBVolume()
|
|
|
|
|
|
class Test(unittest.TestCase):
|
|
def setUp(self):
|
|
self.moxer = mox.Mox()
|
|
|
|
def __init__(self):
|
|
pass
|
|
|
|
PowerVCCinderManager.__init__ = __init__
|
|
self.manager = PowerVCCinderManager()
|
|
|
|
def tearDown(self):
|
|
pass
|
|
|
|
def test_handle_powervc_volume_create_not_create(self):
|
|
self.manager._service = self.moxer.CreateMock(PowerVCService)
|
|
self.moxer.StubOutWithMock(self.manager,
|
|
'_get_local_volume_by_pvc_id')
|
|
self.moxer.StubOutWithMock(self.manager._service,
|
|
'get_volume_by_id')
|
|
self.moxer.StubOutWithMock(self.manager, '_insert_pvc_volume')
|
|
|
|
pvc_id = ''
|
|
self.manager._get_local_volume_by_pvc_id(fake_context, pvc_id)\
|
|
.AndReturn(fake_db_volume)
|
|
|
|
self.moxer.ReplayAll()
|
|
|
|
self.manager._handle_powervc_volume_create(context=fake_context,
|
|
payload=fake_payload)
|
|
|
|
self.moxer.UnsetStubs()
|
|
self.moxer.VerifyAll()
|
|
|
|
def test_handle_powervc_volume_create_create(self):
|
|
self.manager._service = self.moxer.CreateMock(PowerVCService)
|
|
self.moxer.StubOutWithMock(self.manager,
|
|
'_get_local_volume_by_pvc_id')
|
|
self.moxer.StubOutWithMock(self.manager._service,
|
|
'get_volume_by_id')
|
|
self.moxer.StubOutWithMock(self.manager, '_insert_pvc_volume')
|
|
|
|
pvc_id = ''
|
|
volume_id = ''
|
|
fake_volume_instance = FakeVolume()
|
|
self.manager._get_local_volume_by_pvc_id(fake_context, pvc_id)\
|
|
.AndReturn(None)
|
|
self.manager._insert_pvc_volume(fake_context, {}).AndReturn(None)
|
|
self.manager._service.get_volume_by_id(volume_id)\
|
|
.AndReturn(fake_volume_instance)
|
|
self.manager._service.get_volumes()\
|
|
.AndReturn([fake_volume_instance])
|
|
|
|
self.moxer.ReplayAll()
|
|
|
|
self.manager._handle_powervc_volume_create(context=fake_context,
|
|
payload=fake_payload)
|
|
|
|
self.moxer.UnsetStubs()
|
|
self.moxer.VerifyAll()
|
|
|
|
if __name__ == "__main__":
|
|
# import sys;sys.argv = ['', 'Test.testName']
|
|
unittest.main()
|