NSXv: remove LBaaSv1 code

LBaaSv1 has been removed in the N cycle and therefore we can drop the
driver code.

Change-Id: I47c928ede14695b4a793aaf3beedc88ff27186a1
This commit is contained in:
Kobi Samoray 2016-08-25 12:46:02 +03:00
parent 38c163b3e1
commit 0397bf4b61
4 changed files with 0 additions and 1172 deletions

View File

@ -23,7 +23,6 @@ from vmware_nsx.plugins.nsx_v.vshield import edge_appliance_driver
from vmware_nsx.plugins.nsx_v.vshield import edge_firewall_driver from vmware_nsx.plugins.nsx_v.vshield import edge_firewall_driver
from vmware_nsx.plugins.nsx_v.vshield.tasks import tasks from vmware_nsx.plugins.nsx_v.vshield.tasks import tasks
from vmware_nsx.plugins.nsx_v.vshield import vcns from vmware_nsx.plugins.nsx_v.vshield import vcns
from vmware_nsx.services.lbaas.nsx_v.v1 import edge_loadbalancer_driver
from vmware_nsx.services.lbaas.nsx_v.v2 import ( from vmware_nsx.services.lbaas.nsx_v.v2 import (
edge_loadbalancer_driver_v2 as lbaas_v2) edge_loadbalancer_driver_v2 as lbaas_v2)
@ -31,7 +30,6 @@ LOG = logging.getLogger(__name__)
class VcnsDriver(edge_appliance_driver.EdgeApplianceDriver, class VcnsDriver(edge_appliance_driver.EdgeApplianceDriver,
edge_loadbalancer_driver.EdgeLbDriver,
lbaas_v2.EdgeLoadbalancerDriverV2, lbaas_v2.EdgeLoadbalancerDriverV2,
edge_firewall_driver.EdgeFirewallDriver): edge_firewall_driver.EdgeFirewallDriver):

View File

@ -1,627 +0,0 @@
# Copyright 2015 VMware, Inc.
# All Rights Reserved
#
# Licensed under the Apache License, Version 2.0 (the "License"); you may
# not use this file except in compliance with the License. You may obtain
# a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
# License for the specific language governing permissions and limitations
# under the License.
from neutron import manager
from neutron.plugins.common import constants
from neutron_lib import exceptions as n_exc
from oslo_log import log as logging
from oslo_utils import excutils
from vmware_nsx._i18n import _, _LE
from vmware_nsx.common import locking
from vmware_nsx.plugins.nsx_v.vshield.common import exceptions as nsxv_exc
from vmware_nsx.services.lbaas.nsx_v import lbaas_common as lb_common
from vmware_nsx.services.lbaas.nsx_v import lbaas_const as lb_const
LOG = logging.getLogger(__name__)
def convert_lbaas_pool(lbaas_pool):
"""
Transform OpenStack pool dict to NSXv pool dict.
"""
edge_pool = {
'name': 'pool_' + lbaas_pool['id'],
'description': lbaas_pool.get('description',
lbaas_pool.get('name')),
'algorithm': lb_const.BALANCE_MAP.get(
lbaas_pool.get('lb_method'), 'round-robin'),
'transparent': False
}
return edge_pool
def convert_lbaas_app_profile(name, sess_persist, protocol):
"""
Create app profile dict for lbaas VIP.
Neutron-lbaas VIP objects breaks into an application profile object, and
a virtual server object in NSXv.
"""
vcns_app_profile = {
'insertXForwardedFor': False,
'name': name,
'serverSslEnabled': False,
'sslPassthrough': False,
'template': protocol,
}
# Since SSL Termination is not supported right now, so just use
# sslPassthrough method if the protocol is HTTPS.
if protocol == lb_const.LB_PROTOCOL_HTTPS:
vcns_app_profile['sslPassthrough'] = True
if sess_persist:
persist_type = sess_persist.get('type')
if persist_type:
# If protocol is not HTTP, only source_ip is supported
if (protocol != lb_const.LB_PROTOCOL_HTTP and
persist_type != lb_const.LB_SESSION_PERSISTENCE_SOURCE_IP):
msg = (_('Invalid %(protocol)s persistence method: %(type)s') %
{'protocol': protocol,
'type': persist_type})
raise n_exc.BadRequest(resource='edge-lbaas', msg=msg)
persistence = {
'method':
lb_const.SESSION_PERSISTENCE_METHOD_MAP.get(persist_type)}
if persist_type in lb_const.SESSION_PERSISTENCE_COOKIE_MAP:
persistence.update({
'cookieName': sess_persist.get('cookie_name',
'default_cookie_name'),
'cookieMode':
lb_const.SESSION_PERSISTENCE_COOKIE_MAP[persist_type]})
vcns_app_profile['persistence'] = persistence
return vcns_app_profile
def convert_lbaas_vip(vip, app_profile_id, pool_mapping):
"""
Transform OpenStack VIP dict to NSXv virtual server dict.
"""
pool_id = pool_mapping['edge_pool_id']
return {
'name': 'vip_' + vip['id'],
'description': vip['description'],
'ipAddress': vip['address'],
'protocol': vip.get('protocol'),
'port': vip['protocol_port'],
'connectionLimit': max(0, vip.get('connection_limit')),
'defaultPoolId': pool_id,
'applicationProfileId': app_profile_id}
def convert_lbaas_member(member):
"""
Transform OpenStack pool member dict to NSXv pool member dict.
"""
return {
'ipAddress': member['address'],
'weight': member['weight'],
'port': member['protocol_port'],
'monitorPort': member['protocol_port'],
'name': lb_common.get_member_id(member['id']),
'condition': 'enabled' if member['admin_state_up'] else 'disabled'}
def convert_lbaas_monitor(monitor):
"""
Transform OpenStack health monitor dict to NSXv health monitor dict.
"""
mon = {
'type': lb_const.HEALTH_MONITOR_MAP.get(monitor['type'], 'icmp'),
'interval': monitor['delay'],
'timeout': monitor['timeout'],
'maxRetries': monitor['max_retries'],
'name': monitor['id']}
if monitor.get('http_method'):
mon['method'] = monitor['http_method']
if monitor.get('url_path'):
mon['url'] = monitor['url_path']
return mon
class EdgeLbDriver(object):
def __init__(self):
super(EdgeLbDriver, self).__init__()
LOG.debug('Initializing Edge loadbalancer')
# self.vcns is initialized by subclass
self.vcns = None
self._fw_section_id = None
self._lb_plugin = None
self._lbv1_driver_prop = None
def _get_lb_plugin(self):
if not self._lb_plugin:
loaded_plugins = manager.NeutronManager.get_service_plugins()
self._lb_plugin = loaded_plugins[constants.LOADBALANCER]
return self._lb_plugin
@property
def lbv1_driver(self):
if not self._lbv1_driver_prop:
plugin = self._get_lb_plugin()
self._lbv1_driver_prop = plugin.drivers['vmwareedge']
return self._lbv1_driver_prop
def _get_lbaas_fw_section_id(self):
if not self._fw_section_id:
self._fw_section_id = lb_common.get_lbaas_fw_section_id(self.vcns)
return self._fw_section_id
def _get_pool_member_ips(self, context, pool_id, operation, address):
plugin = self._get_lb_plugin()
members = plugin.get_members(
context,
filters={'pool_id': [pool_id]},
fields=['address'])
member_ips = {member['address'] for member in members}
if operation == 'add':
member_ips.add(address)
elif operation == 'del' and address in member_ips:
member_ips.remove(address)
return list(member_ips)
def create_pool(self, context, pool):
LOG.debug('Creating pool %s', pool)
edge_id = lb_common.get_lbaas_edge_id_for_subnet(
context, self.callbacks.plugin, pool['subnet_id'],
pool['tenant_id'])
if edge_id is None:
self.lbv1_driver.pool_failed(context, pool)
msg = _(
'No suitable Edge found for subnet %s') % pool['subnet_id']
raise n_exc.BadRequest(resource='edge-lbaas', msg=msg)
# If Edge appliance is used for the 1st time for LB,
# enable LB acceleration
if not self.is_edge_in_use(context,
edge_id):
lb_common.enable_edge_acceleration(self.vcns, edge_id)
edge_pool = convert_lbaas_pool(pool)
try:
with locking.LockManager.get_lock(edge_id):
h = self.vcns.create_pool(edge_id, edge_pool)[0]
edge_pool_id = lb_common.extract_resource_id(h['location'])
self.lbv1_driver.create_pool_successful(
context, pool, edge_id, edge_pool_id)
except nsxv_exc.VcnsApiException:
with excutils.save_and_reraise_exception():
self.lbv1_driver.pool_failed(context, pool)
LOG.error(_LE('Failed to create pool %s'), pool['id'])
def update_pool(self, context, old_pool, pool, pool_mapping):
LOG.debug('Updating pool %s to %s', old_pool, pool)
edge_pool = convert_lbaas_pool(pool)
try:
with locking.LockManager.get_lock(pool_mapping['edge_id']):
curr_pool = self.vcns.get_pool(pool_mapping['edge_id'],
pool_mapping['edge_pool_id'])[1]
curr_pool.update(edge_pool)
self.vcns.update_pool(pool_mapping['edge_id'],
pool_mapping['edge_pool_id'],
curr_pool)
self.lbv1_driver.pool_successful(context, pool)
except nsxv_exc.VcnsApiException:
with excutils.save_and_reraise_exception():
self.lbv1_driver.pool_failed(context, pool)
LOG.error(_LE('Failed to update pool %s'), pool['id'])
def delete_pool(self, context, pool, pool_mapping):
LOG.debug('Deleting pool %s', pool)
if pool_mapping:
try:
with locking.LockManager.get_lock(pool_mapping['edge_id']):
self.vcns.delete_pool(pool_mapping['edge_id'],
pool_mapping['edge_pool_id'])
except nsxv_exc.VcnsApiException:
LOG.error(_LE('Failed to delete pool %s'), pool['id'])
else:
LOG.error(_LE('No mapping found for pool %s'), pool['id'])
self.lbv1_driver.delete_pool_successful(context, pool)
def create_vip(self, context, vip, pool_mapping):
LOG.debug('Create VIP %s', vip)
app_profile = convert_lbaas_app_profile(
vip['id'], vip.get('session_persistence', {}),
vip.get('protocol'))
if not pool_mapping:
msg = _('Pool %s in not mapped to any Edge appliance') % (
vip['pool_id'])
raise n_exc.BadRequest(resource='edge-lbaas', msg=msg)
edge_id = pool_mapping['edge_id']
try:
with locking.LockManager.get_lock(edge_id):
h = (self.vcns.create_app_profile(edge_id, app_profile))[0]
app_profile_id = lb_common.extract_resource_id(h['location'])
except nsxv_exc.VcnsApiException:
with excutils.save_and_reraise_exception():
self.lbv1_driver.vip_failed(context, vip)
LOG.error(_LE('Failed to create app profile on edge: %s'),
edge_id)
edge_vip = convert_lbaas_vip(vip, app_profile_id, pool_mapping)
try:
lb_common.add_vip_as_secondary_ip(self.vcns, edge_id,
vip['address'])
with locking.LockManager.get_lock(edge_id):
h = self.vcns.create_vip(edge_id, edge_vip)[0]
edge_vip_id = lb_common.extract_resource_id(h['location'])
edge_fw_rule_id = lb_common.add_vip_fw_rule(self.vcns,
edge_id, vip['id'],
vip['address'])
self.lbv1_driver.create_vip_successful(
context, vip, edge_id, app_profile_id, edge_vip_id,
edge_fw_rule_id)
except nsxv_exc.VcnsApiException:
with excutils.save_and_reraise_exception():
self.lbv1_driver.vip_failed(context, vip)
LOG.error(_LE('Failed to create vip on Edge: %s'), edge_id)
with locking.LockManager.get_lock(edge_id):
self.vcns.delete_app_profile(edge_id, app_profile_id)
def update_vip(self, context, old_vip, vip, pool_mapping, vip_mapping):
LOG.debug('Updating VIP %s to %s', old_vip, vip)
edge_id = vip_mapping['edge_id']
edge_vip_id = vip_mapping['edge_vse_id']
app_profile_id = vip_mapping['edge_app_profile_id']
app_profile = convert_lbaas_app_profile(
vip['name'], vip.get('session_persistence', {}),
vip.get('protocol'))
try:
with locking.LockManager.get_lock(edge_id):
self.vcns.update_app_profile(edge_id, app_profile_id,
app_profile)
except nsxv_exc.VcnsApiException:
with excutils.save_and_reraise_exception():
self.lbv1_driver.vip_failed(context, vip)
LOG.error(_LE('Failed to update app profile on edge: %s'),
edge_id)
edge_vip = convert_lbaas_vip(vip, app_profile_id, pool_mapping)
try:
with locking.LockManager.get_lock(edge_id):
self.vcns.update_vip(edge_id, edge_vip_id, edge_vip)
self.lbv1_driver.vip_successful(context, vip)
except nsxv_exc.VcnsApiException:
with excutils.save_and_reraise_exception():
self.lbv1_driver.vip_failed(context, vip)
LOG.error(_LE('Failed to update vip on edge: %s'), edge_id)
def delete_vip(self, context, vip, vip_mapping):
LOG.debug('Deleting VIP %s', vip)
if not vip_mapping:
LOG.error(_LE('No mapping found for vip %s'), vip['id'])
else:
edge_id = vip_mapping['edge_id']
edge_vse_id = vip_mapping['edge_vse_id']
app_profile_id = vip_mapping['edge_app_profile_id']
try:
with locking.LockManager.get_lock(edge_id):
self.vcns.delete_vip(edge_id, edge_vse_id)
except Exception as e:
LOG.error(_LE('Failed to delete VIP from edge %(edge)s. '
'Exception is %(exc)s'),
{'edge': edge_id, 'exc': e})
try:
lb_common.del_vip_as_secondary_ip(self.vcns, edge_id,
vip['address'])
except Exception as e:
LOG.error(_LE('Failed to delete secondary IP from edge '
'%(edge)s. Exception is %(exc)s'),
{'edge': edge_id, 'exc': e})
try:
lb_common.del_vip_fw_rule(self.vcns, edge_id,
vip_mapping['edge_fw_rule_id'])
except Exception as e:
LOG.error(_LE('Failed to delete VIP FW rule from edge '
'%(edge)s. Exception is %(exc)s'),
{'edge': edge_id, 'exc': e})
try:
with locking.LockManager.get_lock(edge_id):
self.vcns.delete_app_profile(edge_id, app_profile_id)
except Exception as e:
LOG.error(_LE('Failed to delete app profile from edge '
'%(edge)s. Exception is %(exc)s'),
{'edge': edge_id, 'exc': e})
self.lbv1_driver.delete_vip_successful(context, vip)
def create_member(self, context, member, pool_mapping):
LOG.debug('Creating member %s', member)
with locking.LockManager.get_lock(pool_mapping['edge_id']):
edge_pool = self.vcns.get_pool(pool_mapping['edge_id'],
pool_mapping['edge_pool_id'])[1]
edge_member = convert_lbaas_member(member)
if edge_pool['member']:
edge_pool['member'].append(edge_member)
else:
edge_pool['member'] = [edge_member]
try:
self.vcns.update_pool(
pool_mapping['edge_id'],
pool_mapping['edge_pool_id'],
edge_pool)
member_ips = self._get_pool_member_ips(
context, member['pool_id'], 'add', member['address'])
lb_common.update_pool_fw_rule(
self.vcns, member['pool_id'], pool_mapping['edge_id'],
self._get_lbaas_fw_section_id(), member_ips)
self.lbv1_driver.member_successful(context, member)
except nsxv_exc.VcnsApiException:
with excutils.save_and_reraise_exception():
self.lbv1_driver.member_failed(context, member)
LOG.error(_LE('Failed to create member on edge: %s'),
pool_mapping['edge_id'])
def update_member(self, context, old_member, member, pool_mapping):
LOG.debug('Updating member %s to %s', old_member, member)
with locking.LockManager.get_lock(pool_mapping['edge_id']):
edge_pool = self.vcns.get_pool(pool_mapping['edge_id'],
pool_mapping['edge_pool_id'])[1]
edge_member = convert_lbaas_member(member)
for i, m in enumerate(edge_pool['member']):
if m['name'] == lb_common.get_member_id(member['id']):
edge_pool['member'][i] = edge_member
break
try:
self.vcns.update_pool(pool_mapping['edge_id'],
pool_mapping['edge_pool_id'],
edge_pool)
self.lbv1_driver.member_successful(context, member)
except nsxv_exc.VcnsApiException:
with excutils.save_and_reraise_exception():
self.lbv1_driver.member_failed(context, member)
LOG.error(_LE('Failed to update member on edge: %s'),
pool_mapping['edge_id'])
def delete_member(self, context, member, pool_mapping):
LOG.debug('Deleting member %s', member)
if pool_mapping:
with locking.LockManager.get_lock(pool_mapping['edge_id']):
edge_pool = self.vcns.get_pool(
pool_mapping['edge_id'],
pool_mapping['edge_pool_id'])[1]
for i, m in enumerate(edge_pool['member']):
if m['name'] == lb_common.get_member_id(member['id']):
edge_pool['member'].pop(i)
break
try:
self.vcns.update_pool(pool_mapping['edge_id'],
pool_mapping['edge_pool_id'],
edge_pool)
member_ips = self._get_pool_member_ips(
context, member['pool_id'], 'del', member['address'])
lb_common.update_pool_fw_rule(
self.vcns, member['pool_id'], pool_mapping['edge_id'],
self._get_lbaas_fw_section_id(), member_ips)
except nsxv_exc.VcnsApiException:
with excutils.save_and_reraise_exception():
self.lbv1_driver.member_failed(context, member)
LOG.error(_LE('Failed to update member on edge: %s'),
pool_mapping['edge_id'])
lb_plugin = self._get_lb_plugin()
lb_plugin._delete_db_member(context, member['id'])
def create_pool_health_monitor(self, context, health_monitor, pool_id,
pool_mapping, mon_mappings):
LOG.debug('Create HM %s', health_monitor)
edge_mon_id = None
with locking.LockManager.get_lock(pool_mapping['edge_id']):
# 1st, we find if we already have a pool with the same monitor, on
# the same Edge appliance.
# If there is no pool on this Edge which is already associated with
# this monitor, create this monitor on Edge
if mon_mappings:
edge_mon_id = mon_mappings['edge_monitor_id']
else:
edge_monitor = convert_lbaas_monitor(health_monitor)
try:
h = self.vcns.create_health_monitor(
pool_mapping['edge_id'], edge_monitor)[0]
edge_mon_id = lb_common.extract_resource_id(h['location'])
except nsxv_exc.VcnsApiException:
self.lbv1_driver.pool_health_monitor_failed(context,
health_monitor,
pool_id)
with excutils.save_and_reraise_exception():
LOG.error(
_LE('Failed to associate monitor on edge: %s'),
pool_mapping['edge_id'])
try:
# Associate monitor with Edge pool
edge_pool = self.vcns.get_pool(pool_mapping['edge_id'],
pool_mapping['edge_pool_id'])[1]
if edge_pool['monitorId']:
edge_pool['monitorId'].append(edge_mon_id)
else:
edge_pool['monitorId'] = [edge_mon_id]
self.vcns.update_pool(pool_mapping['edge_id'],
pool_mapping['edge_pool_id'],
edge_pool)
except nsxv_exc.VcnsApiException:
with excutils.save_and_reraise_exception():
self.lbv1_driver.pool_health_monitor_failed(context,
health_monitor,
pool_id)
LOG.error(
_LE('Failed to associate monitor on edge: %s'),
pool_mapping['edge_id'])
self.lbv1_driver.create_pool_health_monitor_successful(
context, health_monitor, pool_id, pool_mapping['edge_id'],
edge_mon_id)
def update_pool_health_monitor(self, context, old_health_monitor,
health_monitor, pool_id, mon_mapping):
LOG.debug('Update HM %s to %s', old_health_monitor, health_monitor)
edge_monitor = convert_lbaas_monitor(health_monitor)
try:
with locking.LockManager.get_lock(mon_mapping['edge_id']):
self.vcns.update_health_monitor(
mon_mapping['edge_id'],
mon_mapping['edge_monitor_id'],
edge_monitor)
except nsxv_exc.VcnsApiException:
with excutils.save_and_reraise_exception():
self.lbv1_driver.pool_health_monitor_failed(context,
health_monitor,
pool_id)
LOG.error(
_LE('Failed to update monitor on edge: %s'),
mon_mapping['edge_id'])
self.lbv1_driver.pool_health_monitor_successful(context,
health_monitor,
pool_id)
def delete_pool_health_monitor(self, context, health_monitor, pool_id,
pool_mapping, mon_mapping):
LOG.debug('Deleting HM %s', health_monitor)
edge_id = pool_mapping['edge_id']
if not mon_mapping:
return
with locking.LockManager.get_lock(pool_mapping['edge_id']):
edge_pool = self.vcns.get_pool(edge_id,
pool_mapping['edge_pool_id'])[1]
edge_pool['monitorId'].remove(mon_mapping['edge_monitor_id'])
try:
self.vcns.update_pool(edge_id,
pool_mapping['edge_pool_id'],
edge_pool)
except nsxv_exc.VcnsApiException:
with excutils.save_and_reraise_exception():
self.lbv1_driver.pool_health_monitor_failed(context,
health_monitor,
pool_id)
LOG.error(
_LE('Failed to delete monitor mapping on edge: %s'),
mon_mapping['edge_id'])
# If this monitor is not used on this edge anymore, delete it
if not edge_pool['monitorId']:
try:
self.vcns.delete_health_monitor(
mon_mapping['edge_id'],
mon_mapping['edge_monitor_id'])
except nsxv_exc.VcnsApiException:
with excutils.save_and_reraise_exception():
self.lbv1_driver.pool_health_monitor_failed(
context, health_monitor, pool_id)
LOG.error(
_LE('Failed to delete monitor on edge: %s'),
mon_mapping['edge_id'])
self.lbv1_driver.delete_pool_health_monitor_successful(
context, health_monitor, pool_id, mon_mapping)
def stats(self, context, pool_id, pool_mapping):
LOG.debug('Retrieving stats for pool %s', pool_id)
try:
lb_stats = self.vcns.get_loadbalancer_statistics(
pool_mapping['edge_id'])
except nsxv_exc.VcnsApiException:
with excutils.save_and_reraise_exception():
LOG.error(
_LE('Failed to read load balancer statistics, edge: %s'),
pool_mapping['edge_id'])
pools_stats = lb_stats[1].get('pool', [])
plugin = self._get_lb_plugin()
members = plugin.get_members(
context,
filters={'pool_id': [pool_id]},
fields=['id', 'status'])
member_map = {m['id']: m['status'] for m in members}
for pool_stats in pools_stats:
if pool_stats['poolId'] == pool_mapping['edge_pool_id']:
stats = {'bytes_in': pool_stats.get('bytesIn', 0),
'bytes_out': pool_stats.get('bytesOut', 0),
'active_connections':
pool_stats.get('curSessions', 0),
'total_connections':
pool_stats.get('totalSessions', 0)}
member_stats = {}
for member in pool_stats.get('member', []):
member_id = member['name'][len(lb_common.MEMBER_ID_PFX):]
if member_map[member_id] != 'ERROR':
member_stats[member_id] = {
'status': ('INACTIVE'
if member['status'] == 'DOWN'
else 'ACTIVE')}
stats['members'] = member_stats
return stats
return {'bytes_in': 0,
'bytes_out': 0,
'active_connections': 0,
'total_connections': 0}
def is_edge_in_use(self, context, edge_id):
return self.lbv1_driver.is_edge_in_use(context, edge_id)
def is_subnet_in_use(self, context, subnet_id):
plugin = self._get_lb_plugin()
if plugin:
pools = plugin.get_pools(context,
filters={'subnet_id': [subnet_id]})
if pools:
return True

View File

@ -1,543 +0,0 @@
# Copyright 2015 VMware, Inc.
# All Rights Reserved
#
# Licensed under the Apache License, Version 2.0 (the "License"); you may
# not use this file except in compliance with the License. You may obtain
# a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
# License for the specific language governing permissions and limitations
# under the License.
import mock
from neutron import context
from neutron.tests import base
from vmware_nsx.plugins.nsx_v.vshield import vcns_driver
from vmware_nsx.services.lbaas.nsx_v import lbaas_common as lb_common
EDGE_PROVIDER = ('LOADBALANCER:vmwareedge:neutron.services.'
'loadbalancer.drivers.vmware.edge_driver.'
'EdgeLoadbalancerDriver:default')
HEALTHMON_ID = 'cb297614-66c9-4048-8838-7e87231569ae'
POOL_ID = 'b3dfb476-6fdf-4ddd-b6bd-e86ae78dc30b'
TENANT_ID = 'f9135d3a908842bd8d785816c2c90d36'
SUBNET_ID = 'c8924d77-ff57-406f-a13c-a8c5def01fc9'
VIP_ID = 'f6393b95-34b0-4299-9001-cbc21e32bf03'
VIP_PORT_ID = '49c547e3-6775-42ea-a607-91e8f1a07432'
MEMBER_ID = '90dacafd-9c11-4af7-9d89-234e2d1fedb1'
EDGE_ID = 'edge-x'
EDGE_POOL_ID = '111'
EDGE_VSE_ID = '222'
APP_PROFILE_ID = '333'
EDGE_MON_ID = '444'
EDGE_FW_RULE_ID = '555'
def lbaas_pool_maker(**kwargs):
lbaas_dict = {
'status': 'PENDING_CREATE',
'lb_method': 'ROUND_ROBIN',
'protocol': 'HTTP',
'description': '',
'health_monitors': [],
'members': [],
'status_description': None,
'id': POOL_ID,
'vip_id': None,
'name': 'testpool',
'admin_state_up': True,
'subnet_id': SUBNET_ID,
'tenant_id': TENANT_ID,
'health_monitors_status': [],
'provider': 'vmwareedge'}
lbaas_dict.update(kwargs)
return lbaas_dict
def lbaas_vip_maker(**kwargs):
lbaas_vip = {
'status': 'PENDING_CREATE',
'protocol': 'HTTP',
'description': '',
'address': '10.0.0.8',
'protocol_port': 555,
'port_id': VIP_PORT_ID,
'id': VIP_ID,
'status_description': None,
'name': 'testvip1',
'admin_state_up': True,
'subnet_id': SUBNET_ID,
'tenant_id': TENANT_ID,
'connection_limit': -1,
'pool_id': POOL_ID,
'session_persistence': {'type': 'SOURCE_IP'}}
lbaas_vip.update(kwargs)
return lbaas_vip
def lbaas_member_maker(**kwargs):
lbaas_member = {
'admin_state_up': True,
'status': 'PENDING_CREATE',
'status_description': None,
'weight': 5,
'address': '10.0.0.4',
'tenant_id': TENANT_ID,
'protocol_port': 555,
'id': MEMBER_ID,
'pool_id': POOL_ID}
lbaas_member.update(kwargs)
return lbaas_member
def lbaas_hmon_maker(**kwargs):
hmon = {
'admin_state_up': True,
'tenant_id': TENANT_ID,
'delay': 5,
'max_retries': 5,
'timeout': 5,
'pools': [{'status': 'PENDING_CREATE', 'status_description': None,
'pool_id': POOL_ID}],
'type': 'PING',
'id': HEALTHMON_ID}
hmon.update(kwargs)
return hmon
class TestEdgeLbDriver(base.BaseTestCase):
def setUp(self):
super(TestEdgeLbDriver, self).setUp()
self.context = context.get_admin_context()
callbacks = mock.Mock()
callbacks.plugin = mock.Mock()
self.edge_driver = vcns_driver.VcnsDriver(callbacks)
self.edge_driver._lbv1_driver_prop = mock.Mock()
self._temp_get_lbaas_edge_id_for_subnet = (
lb_common.get_lbaas_edge_id_for_subnet)
self._temp_update_pool_fw_rule = lb_common.update_pool_fw_rule
self._temp_add_vip_as_secondary_ip = lb_common.add_vip_as_secondary_ip
self._temp_add_vip_fw_rule = lb_common.add_vip_fw_rule
self._temp_del_vip_as_secondary_ip = lb_common.del_vip_as_secondary_ip
self._temp_del_vip_fw_rule = lb_common.del_vip_fw_rule
lb_common.get_lbaas_edge_id_for_subnet = mock.Mock()
lb_common.update_pool_fw_rule = mock.Mock()
lb_common.add_vip_as_secondary_ip = mock.Mock()
lb_common.add_vip_fw_rule = mock.Mock()
lb_common.del_vip_as_secondary_ip = mock.Mock()
lb_common.del_vip_fw_rule = mock.Mock()
def tearDown(self):
super(TestEdgeLbDriver, self).tearDown()
lb_common.get_lbaas_edge_id_for_subnet = (
self._temp_get_lbaas_edge_id_for_subnet)
lb_common.update_pool_fw_rule = self._temp_update_pool_fw_rule
lb_common.add_vip_as_secondary_ip = self._temp_add_vip_as_secondary_ip
lb_common.add_vip_fw_rule = self._temp_add_vip_fw_rule
lb_common.del_vip_as_secondary_ip = self._temp_del_vip_as_secondary_ip
lb_common.del_vip_fw_rule = self._temp_del_vip_fw_rule
def _mock_edge_driver(self, attr):
return mock.patch.object(self.edge_driver, attr)
def _mock_edge_driver_vcns(self, attr):
return mock.patch.object(self.edge_driver.vcns, attr)
def _mock_edge_lbv1_driver(self, attr):
return mock.patch.object(self.edge_driver.lbv1_driver, attr)
def test_create_pool(self):
lbaas_pool = lbaas_pool_maker()
edge_pool = {
'transparent': False, 'name': 'pool_' + POOL_ID,
'algorithm': 'round-robin', 'description': ''}
with self._mock_edge_lbv1_driver(
'create_pool_successful') as create_pool_successful, \
self._mock_edge_driver_vcns('create_pool') as create_pool:
lb_common.get_lbaas_edge_id_for_subnet.return_value = EDGE_ID
create_pool.return_value = ({'location': 'x/' + EDGE_POOL_ID},
None)
self.edge_driver.create_pool(self.context, lbaas_pool)
create_pool.assert_called_with(EDGE_ID, edge_pool)
create_pool_successful.assert_called_with(
self.context, lbaas_pool, EDGE_ID, EDGE_POOL_ID)
def test_update_pool(self):
from_pool = lbaas_pool_maker(status='ACTIVE')
to_pool = lbaas_pool_maker(status='PENDING_UPDATE',
lb_method='LEAST_CONNECTIONS')
edge_pool = {
'transparent': False, 'name': 'pool_' + POOL_ID,
'algorithm': 'leastconn', 'description': ''}
pool_mapping = {'edge_id': EDGE_ID, 'edge_pool_id': EDGE_POOL_ID}
with self._mock_edge_lbv1_driver('pool_successful') as pool_successful,\
self._mock_edge_driver_vcns('get_pool') as get_pool, \
self._mock_edge_driver_vcns('update_pool') as update_pool:
get_pool.return_value = (None, {})
self.edge_driver.update_pool(
self.context, from_pool, to_pool, pool_mapping)
update_pool.assert_called_with(EDGE_ID, EDGE_POOL_ID, edge_pool)
pool_successful.assert_called_with(self.context, to_pool)
def test_delete_pool(self):
lbaas_pool = lbaas_pool_maker()
pool_mapping = {'edge_id': EDGE_ID, 'edge_pool_id': EDGE_POOL_ID}
with self._mock_edge_driver_vcns('delete_pool'),\
self._mock_edge_lbv1_driver(
'delete_pool_successful') as mock_delete_successful:
self.edge_driver.delete_pool(
self.context, lbaas_pool, pool_mapping)
mock_delete_successful.assert_called_with(self.context, lbaas_pool)
def test_create_vip(self):
lbaas_vip = lbaas_vip_maker()
edge_app_prof = {
'name': VIP_ID, 'insertXForwardedFor': False,
'serverSslEnabled': False, 'template': 'HTTP',
'sslPassthrough': False, 'persistence': {'method': 'sourceip'}}
edge_vip = {
'protocol': 'HTTP', 'name': 'vip_' + VIP_ID, 'connectionLimit': 0,
'defaultPoolId': EDGE_POOL_ID, 'ipAddress': '10.0.0.8',
'port': 555, 'applicationProfileId': APP_PROFILE_ID,
'description': ''}
pool_mapping = {'edge_id': EDGE_ID, 'edge_pool_id': EDGE_POOL_ID}
with self._mock_edge_driver_vcns(
'create_app_profile') as mock_create_app_profile,\
self._mock_edge_driver_vcns('create_vip') as mock_create_vip,\
self._mock_edge_lbv1_driver(
'create_vip_successful') as mock_vip_successful:
mock_create_app_profile.return_value = (
{'location': 'x/' + APP_PROFILE_ID}, None)
mock_create_vip.return_value = (
{'location': 'x/' + EDGE_VSE_ID}, None)
lb_common.add_vip_fw_rule.return_value = EDGE_FW_RULE_ID
self.edge_driver.create_vip(self.context, lbaas_vip, pool_mapping)
mock_create_app_profile.assert_called_with(EDGE_ID, edge_app_prof)
lb_common.add_vip_fw_rule.assert_called_with(
self.edge_driver.vcns, EDGE_ID, VIP_ID, '10.0.0.8')
mock_create_vip.assert_called_with(EDGE_ID, edge_vip)
mock_vip_successful.assert_called_with(
self.context, lbaas_vip, EDGE_ID, APP_PROFILE_ID, EDGE_VSE_ID,
EDGE_FW_RULE_ID)
def test_update_vip(self):
vip_from = lbaas_vip_maker(status='ACTIVE')
vip_to = lbaas_vip_maker(status='PENDING_UPDATE',
session_persistence={'type': 'HTTP_COOKIE'})
edge_app_prof = {
'name': 'testvip1', 'insertXForwardedFor': False,
'serverSslEnabled': False, 'template': 'HTTP',
'sslPassthrough': False,
'persistence': {'cookieName': 'default_cookie_name',
'method': 'cookie', 'cookieMode': 'insert'}}
edge_vip = {
'protocol': 'HTTP', 'name': 'vip_' + VIP_ID, 'connectionLimit': 0,
'defaultPoolId': EDGE_POOL_ID, 'ipAddress': '10.0.0.8',
'port': 555, 'applicationProfileId': '333', 'description': ''}
pool_mapping = {'edge_pool_id': '111'}
vip_mapping = {'edge_id': EDGE_ID, 'edge_vse_id': EDGE_VSE_ID,
'edge_app_profile_id': APP_PROFILE_ID}
with self._mock_edge_driver_vcns('update_vip') as mock_upd_vip,\
self._mock_edge_driver_vcns(
'update_app_profile') as mock_upd_app_prof,\
self._mock_edge_lbv1_driver(
'vip_successful') as mock_vip_successful:
self.edge_driver.update_vip(self.context, vip_from, vip_to,
pool_mapping, vip_mapping)
mock_upd_app_prof.assert_called_with(EDGE_ID, APP_PROFILE_ID,
edge_app_prof)
mock_upd_vip.assert_called_with(EDGE_ID, EDGE_VSE_ID, edge_vip)
mock_vip_successful.assert_called_with(self.context, vip_to)
def test_delete_vip(self):
lbaas_vip = lbaas_vip_maker(status='PENDING_DELETE')
vip_mapping = {'edge_id': EDGE_ID, 'edge_vse_id': EDGE_VSE_ID,
'edge_app_profile_id': APP_PROFILE_ID,
'edge_fw_rule_id': EDGE_FW_RULE_ID}
with self._mock_edge_driver_vcns(
'delete_app_profile') as mock_del_app_profile,\
self._mock_edge_driver_vcns('delete_vip') as mock_del_vip,\
self._mock_edge_lbv1_driver(
'delete_vip_successful') as mock_del_successful:
self.edge_driver.delete_vip(self.context, lbaas_vip, vip_mapping)
mock_del_app_profile.assert_called_with(EDGE_ID, APP_PROFILE_ID)
mock_del_vip.assert_called_with(EDGE_ID, EDGE_VSE_ID)
lb_common.del_vip_fw_rule.assert_called_with(
self.edge_driver.vcns, EDGE_ID, EDGE_FW_RULE_ID)
mock_del_successful.assert_called_with(self.context, lbaas_vip)
def test_create_member(self):
lbaas_member = lbaas_member_maker()
edge_pool = {
'monitorId': [], 'name': POOL_ID, 'applicationRuleId': [],
'member': [], 'poolId': 'pool-1', 'algorithm': 'round-robin',
'transparent': False}
edge_member = {
'condition': 'enabled', 'ipAddress': '10.0.0.4', 'port': 555,
'weight': 5, 'name': 'member-' + MEMBER_ID}
pool_mapping = {'edge_id': EDGE_ID, 'edge_pool_id': EDGE_POOL_ID}
with self._mock_edge_driver_vcns('update_pool'),\
self._mock_edge_driver_vcns('get_pool') as mock_get_pool,\
self._mock_edge_driver(
'_get_pool_member_ips') as mock_get_ips, \
self._mock_edge_driver(
'_get_lbaas_fw_section_id') as mock_fw_sect, \
self._mock_edge_lbv1_driver(
'member_successful') as mock_member_successful:
mock_get_pool.return_value = (None, edge_pool)
mock_get_ips.return_value = ['10.0.0.4']
mock_fw_sect.return_value = 10010
self.edge_driver.create_member(self.context, lbaas_member,
pool_mapping)
edge_pool['member'].append(edge_member)
mock_member_successful.assert_called_with(self.context,
lbaas_member)
def test_update_member(self):
member_from = lbaas_member_maker(status='ACTIVE')
member_to = lbaas_member_maker(status='PENDING_UPDATE', weight=10)
edge_pool = {
'monitorId': [], 'name': POOL_ID, 'applicationRuleId': [],
'member': [
{'condition': 'enabled', 'ipAddress': '10.0.0.4', 'port': 555,
'weight': 5, 'name': 'member-' + MEMBER_ID}],
'poolId': 'pool-1', 'algorithm': 'round-robin',
'transparent': False}
pool_mapping = {'edge_id': EDGE_ID, 'edge_pool_id': EDGE_POOL_ID}
with self._mock_edge_driver_vcns('get_pool') as mock_get_pool,\
self._mock_edge_driver_vcns(
'update_pool') as mock_update_pool,\
self._mock_edge_lbv1_driver(
'member_successful') as mock_member_successful:
mock_get_pool.return_value = (None, edge_pool)
self.edge_driver.update_member(self.context, member_from,
member_to, pool_mapping)
edge_pool['member'][0]['weight'] = 6
mock_update_pool.assert_called_with(EDGE_ID, EDGE_POOL_ID,
edge_pool)
mock_member_successful.assert_called_with(self.context, member_to)
def test_delete_member(self):
def _del_member(context, member_id):
self.assertEqual(context, self.context)
self.assertEqual(member_id, MEMBER_ID)
lbaas_member = lbaas_member_maker(status='PENDING_DELETE')
edge_pool = {
'monitorId': [], 'name': POOL_ID, 'applicationRuleId': [],
'member': [
{'condition': 'enabled', 'ipAddress': '10.0.0.4', 'port': 555,
'weight': 5, 'name': 'member-' + MEMBER_ID}],
'poolId': 'pool-1', 'algorithm': 'round-robin',
'transparent': False}
pool_mapping = {'edge_id': EDGE_ID, 'edge_pool_id': EDGE_POOL_ID}
mock_lb_plugin = mock.Mock()
mock_lb_plugin._delete_db_member = _del_member
with self._mock_edge_driver('_get_lb_plugin') as mock_get_lb_plugin,\
self._mock_edge_driver_vcns('get_pool') as mock_get_pool,\
self._mock_edge_driver(
'_get_lbaas_fw_section_id') as mock_fw_sect, \
self._mock_edge_driver_vcns(
'update_pool') as mock_update_pool:
mock_get_pool.return_value = (None, edge_pool)
self.edge_driver.delete_member(self.context, lbaas_member,
pool_mapping)
mock_fw_sect.return_value = 10010
mock_get_lb_plugin.return_value = mock_lb_plugin
edge_pool['member'] = []
mock_update_pool.assert_called_with(EDGE_ID, EDGE_POOL_ID,
edge_pool)
def test_create_pool_health_monitor(self):
hmon = lbaas_hmon_maker()
edge_hm = {'maxRetries': 5, 'interval': 5, 'type': 'icmp',
'name': HEALTHMON_ID, 'timeout': 5}
edge_pool = {'monitorId': [], 'name': POOL_ID,
'applicationRuleId': [], 'member': [],
'poolId': 'pool-1', 'algorithm': 'round-robin',
'transparent': False}
pool_mapping = {'edge_id': EDGE_ID, 'edge_pool_id': EDGE_POOL_ID}
with self._mock_edge_driver_vcns('update_pool') as mock_update_pool,\
self._mock_edge_driver_vcns('get_pool') as mock_get_pool,\
self._mock_edge_driver_vcns(
'create_health_monitor') as mock_create_mon,\
self._mock_edge_lbv1_driver(
'create_pool_health_monitor_successful') as (
mock_create_successful):
mock_create_mon.return_value = ({'location': 'x/' + HEALTHMON_ID},
None)
mock_get_pool.return_value = (None, edge_pool)
self.edge_driver.create_pool_health_monitor(
self.context, hmon, POOL_ID, pool_mapping, None)
mock_create_mon.assert_called_with(EDGE_ID, edge_hm)
edge_pool['monitorId'].append(HEALTHMON_ID)
mock_update_pool.assert_called_with(EDGE_ID, EDGE_POOL_ID,
edge_pool)
mock_create_successful.assert_called_with(
self.context, hmon, POOL_ID, EDGE_ID, HEALTHMON_ID)
def test_update_pool_health_monitor(self):
from_hmon = lbaas_hmon_maker(status='ACTIVE')
to_hmon = lbaas_hmon_maker(status='PENDING_UPDATE',
max_retries=10)
edge_hmon = {'maxRetries': 10, 'interval': 5, 'type': 'icmp',
'name': HEALTHMON_ID, 'timeout': 5}
mon_mapping = {'edge_id': EDGE_ID, 'edge_monitor_id': EDGE_MON_ID}
with self._mock_edge_driver_vcns(
'update_health_monitor') as mock_update_mon,\
self._mock_edge_lbv1_driver(
'pool_health_monitor_successful') as mock_hmon_successful:
self.edge_driver.update_pool_health_monitor(
self.context, from_hmon, to_hmon, POOL_ID, mon_mapping)
mock_update_mon.assert_called_with(EDGE_ID, EDGE_MON_ID, edge_hmon)
mock_hmon_successful.assert_called_with(self.context, to_hmon,
POOL_ID,)
def test_delete_pool_health_monitor(self):
hmon = lbaas_hmon_maker(status='PENDING_DELETE')
edge_pool = {'monitorId': [EDGE_MON_ID], 'name': POOL_ID,
'applicationRuleId': [], 'member': [],
'poolId': 'pool-1', 'algorithm': 'round-robin',
'transparent': False}
pool_mapping = {'edge_id': EDGE_ID, 'edge_pool_id': EDGE_POOL_ID}
mon_mapping = {'edge_id': EDGE_ID, 'edge_monitor_id': EDGE_MON_ID}
with self._mock_edge_driver_vcns('update_pool') as mock_update_pool,\
self._mock_edge_driver_vcns('get_pool') as mock_get_pool,\
self._mock_edge_driver_vcns(
'delete_health_monitor') as mock_del_mon,\
self._mock_edge_lbv1_driver(
'delete_pool_health_monitor_successful') as (
mock_del_successful):
mock_get_pool.return_value = (None, edge_pool)
self.edge_driver.delete_pool_health_monitor(
self.context, hmon, POOL_ID, pool_mapping, mon_mapping)
edge_pool['monitorId'] = []
mock_update_pool.assert_called_with(EDGE_ID, EDGE_POOL_ID,
edge_pool)
mock_del_mon.assert_called_with(EDGE_ID, EDGE_MON_ID)
mock_del_successful.assert_called_with(self.context, hmon, POOL_ID,
mon_mapping)
def test_stats(self):
pool_mapping = {'edge_id': EDGE_ID, 'edge_pool_id': EDGE_POOL_ID}
pool_stats = (
{
'status': '200',
'content-location': '',
'transfer-encoding': 'chunked',
'expires': 'Thu, 01 Jan 1970 00:00:00 GMT',
'server': '',
'cache-control': 'private, no-cache',
'date': 'Thu, 30 Jul 2015 08:59:27 GMT',
'content-type': 'application/json'},
{
'timeStamp': 1427358733,
'virtualServer': [
{'name': 'MdSrv',
'virtualServerId': 'virtualServer-1',
'bytesIn': 0,
'bytesOut': 0,
'totalSessions': 0,
'ipAddress': '169.254.128.2',
'curSessions': 0}],
'pool': [
{'status': 'UP',
'totalSessions': 10000,
'rateMax': 0,
'name': 'MDSrvPool',
'bytesOut': 100000,
'rateLimit': 0,
'member': [
{'status': 'UP',
'name': 'member-xxx-xxx-xxx-xxx',
'bytesOut': 0,
'memberId': 'member-1',
'totalSessions': 20000,
'ipAddress': '192.168.55.101',
'httpReqRateMax': 0,
'curSessions': 0,
'bytesIn': 0},
{'status': 'UP',
'name': 'member-yyy-yyy-yyy-yyy',
'bytesOut': 0,
'memberId': 'member-2',
'totalSessions': 20000,
'ipAddress': '192.168.55.102',
'httpReqRateMax': 0,
'curSessions': 0,
'bytesIn': 0}],
'poolId': EDGE_POOL_ID,
'maxSessions': 10000,
'httpReqRateMax': 0,
'curSessions': 5000,
'bytesIn': 1000000}]})
expected_stats = {'active_connections': 5000,
'bytes_in': 1000000,
'bytes_out': 100000,
'total_connections': 10000,
'members': {'xxx-xxx-xxx-xxx': {'status': 'ACTIVE'}}}
members = [{'id': 'xxx-xxx-xxx-xxx', 'status': 'ACTIVE'},
{'id': 'yyy-yyy-yyy-yyy', 'status': 'ERROR'}]
mock_lb_plugin = mock.Mock()
with mock.patch.object(self.edge_driver.vcns,
'get_loadbalancer_statistics',
return_value=pool_stats), \
mock.patch.object(self.edge_driver,
'_get_lb_plugin',
return_value=mock_lb_plugin), \
mock.patch.object(mock_lb_plugin, 'get_members',
return_value=members):
stats = self.edge_driver.stats(self.context, POOL_ID, pool_mapping)
self.assertEqual(stats, expected_stats)