def test_create_vip_1(self, mock_f1, mock_f2):
"""Exception"""
cmd.create_vip(self.ctx, self.vip, self.server_farm)
rollback_fn = self.ctx.add_rollback.call_args[0][0]
rollback_fn(False)
self.assertTrue(mock_f2.called)
mock_f2.assert_called_once_with(self.ctx, self.vip)
def lb_add_vip(conf, tenant_id, lb_id, vip_dict):
LOG.debug("Called lb_add_vip(), conf: %r, lb_id: %s, vip_dict: %r",
conf, lb_id, vip_dict)
lb_ref = db_api.loadbalancer_get(conf, lb_id, tenant_id=tenant_id)
# NOTE(akscram): server farms are really only create problems than
# they solve multiply use of the virtual IPs.
try:
sf_ref = db_api.serverfarm_get_all_by_lb_id(conf, lb_ref['id'])[0]
except IndexError:
raise exc.ServerFarmNotFound
values = db_api.virtualserver_pack_extra(vip_dict)
values['lb_id'] = lb_ref['id']
values['sf_id'] = sf_ref['id']
# XXX(akscram): Set default protocol from LoadBalancer to
# VirtualServer if it is not present.
if not values.get('extra'):
values['extra'] = {'protocol': lb_ref['protocol']}
elif 'protocol' not in values['extra']:
values['extra']['protocol'] = lb_ref['protocol']
vip_ref = db_api.virtualserver_create(conf, values)
device_driver = drivers.get_device_driver(conf, lb_ref['device_id'])
with device_driver.request_context() as ctx:
commands.create_vip(ctx, vip_ref, sf_ref)
return db_api.unpack_extra(vip_ref)
def test_create_vip_0(self, mock_f1, mock_f2):
"""No exception"""
cmd.create_vip(self.ctx, self.vip, self.server_farm)
self.assertTrue(self.ctx.device.create_virtual_ip.called)
self.assertTrue(mock_f1.called)
self.assertFalse(mock_f2.called)
mock_f1.assert_called_once_with(self.ctx.conf, self.vip['id'],
{'deployed': True})
def lb_add_vip(conf, lb_id, vip_dict):
logger.debug("Called lb_add_vip(), conf: %r, lb_id: %s, vip_dict: %r",
conf, lb_id, vip_dict)
lb_ref = db_api.loadbalancer_get(conf, lb_id)
# NOTE(akscram): server farms are really only create problems than
# they solve multiply use of the virtual IPs.
try:
sf_ref = db_api.serverfarm_get_all_by_lb_id(conf, lb_ref['id'])[0]
except IndexError:
raise exc.ServerFarmNotFound
values = db_api.virtualserver_pack_extra(vip_dict)
values['lb_id'] = lb_ref['id']
values['sf_id'] = sf_ref['id']
vip_ref = db_api.virtualserver_create(conf, values)
device_driver = drivers.get_device_driver(conf, lb_ref['device_id'])
with device_driver.request_context() as ctx:
commands.create_vip(ctx, vip_ref, sf_ref)
return db_api.unpack_extra(vip_ref)
def test_create_vip_0(self, mock_f1, mock_f2):
"""No exception"""
cmd.create_vip(self.ctx, self.vip, self.server_farm)
self.assertTrue(self.ctx.device.create_virtual_ip.called)
self.assertTrue(mock_f1.called)