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 test_server_shutdown(self):
"""
- shutdown current instance
Expected: server is down
"""
del self.vmmaster
with self.assertRaises(RuntimeError):
server_is_up(self.address, wait=1)
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 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)