def setUp(self):
setup_config('data/config.py')
self.address = ("localhost", 9001)
with patch(
'core.connection.Virsh',
Mock(),
), patch('core.network.Network',
Mock()), patch('core.db.Database', DatabaseMock()):
from vmmaster.server import VMMasterServer
self.vmmaster = VMMasterServer(reactor, self.address[1])
self.pool = self.vmmaster.app.pool
server_is_up(self.address)
self.desired_caps = {
'desiredCapabilities': {
'platform': self.pool.platforms.platforms.keys()[0]
}
}
from core.config import config
config.SESSION_TIMEOUT = 1
self.ctx = self.vmmaster.app.app_context()
self.ctx.push()
def setUp(self):
setup_config('data/config_with_preloaded.py')
self.address = ("localhost", 9001)
self.desired_caps = {'desiredCapabilities': {'platform': 'origin_1'}}
mocked_image = Mock(id=1,
status='active',
get=Mock(return_value='snapshot'),
min_disk=20,
min_ram=2,
instance_type_flavorid=1)
type(mocked_image).name = PropertyMock(return_value='test_origin_1')
with patch(
'core.connection.Virsh',
Mock(),
), patch('core.network.Network', Mock()), patch(
'core.db.Database',
Mock(add=Mock(side_effect=set_primary_key))), patch.multiple(
'core.utils.openstack_utils',
neutron_client=Mock(return_value=Mock()),
glance_client=Mock(return_value=Mock()),
nova_client=Mock(return_value=Mock())), patch.multiple(
'vmpool.clone.OpenstackClone',
_wait_for_activated_service=custom_wait,
get_network_id=Mock(return_value=1),
get_network_name=Mock(
return_value='Local-Net')), patch.multiple(
'vmpool.platforms.OpenstackPlatforms',
images=Mock(return_value=[mocked_image]),
flavor_params=Mock(return_value={
'vcpus': 1,
'ram': 2
}),
limits=Mock(
return_value={
'maxTotalCores': 10,
'maxTotalInstances': 10,
'maxTotalRAMSize': 100,
'totalCoresUsed': 0,
'totalInstancesUsed': 0,
'totalRAMUsed': 0
})):
from vmmaster.server import VMMasterServer
self.vmmaster = VMMasterServer(reactor, self.address[1])
self.ctx = self.vmmaster.app.app_context()
self.ctx.push()
self.pool = self.vmmaster.app.pool
server_is_up(self.address)
def vmmaster_server_mock(port):
with patch('core.network.Network',
Mock(return_value=Mock(get_ip=Mock(return_value='0')))), patch(
'core.connection.Virsh',
Mock()), patch('core.db.Database', DatabaseMock()), patch(
'core.utils.init.home_dir',
Mock(return_value=fake_home_dir())), patch(
'core.logger.setup_logging',
Mock(return_value=Mock())), patch(
'core.sessions.SessionWorker', Mock()):
from vmmaster.server import VMMasterServer
return VMMasterServer(reactor, port)
def setUp(self):
with patch(
'core.connection.Virsh',
Mock(),
), patch('core.network.Network',
Mock()), patch('core.db.Database', DatabaseMock()), patch(
'core.sessions.SessionWorker', Mock()):
from vmmaster.server import VMMasterServer
self.vmmaster = VMMasterServer(reactor, self.address[1])
self.pool = self.vmmaster.app.pool
server_is_up(self.address)
def vmmaster_server_mock(port):
mocked_image = Mock(
id=1, status='active',
get=Mock(return_value='snapshot'),
min_disk=20,
min_ram=2,
instance_type_flavorid=1,
)
type(mocked_image).name = PropertyMock(
return_value='test_origin_1')
with patch(
'core.db.Database', DatabaseMock()
), patch(
'core.utils.init.home_dir', Mock(return_value=fake_home_dir())
), patch(
'core.logger.setup_logging', Mock(return_value=Mock())
), patch(
'core.sessions.SessionWorker', Mock()
), patch.multiple(
'vmpool.platforms.OpenstackPlatforms',
images=Mock(return_value=[mocked_image]),
flavor_params=Mock(return_value={'vcpus': 1, 'ram': 2}),
limits=Mock(return_value={
'maxTotalCores': 10, 'maxTotalInstances': 10,
'maxTotalRAMSize': 100, 'totalCoresUsed': 0,
'totalInstancesUsed': 0, 'totalRAMUsed': 0}),
), patch.multiple(
'core.utils.openstack_utils',
nova_client=Mock(return_value=Mock())
), patch.multiple(
'core.db.models.OpenstackClone',
_wait_for_activated_service=custom_wait,
ping_vm=Mock(return_value=True)
):
from vmmaster.server import VMMasterServer
return VMMasterServer(reactor, port)
def runserver():
"""
Run server
"""
from vmmaster.server import VMMasterServer
VMMasterServer(reactor, config.PORT).run()