def setUp(self):
super(BaseTestMemberManager, self).setUp()
self.in_member = data_models.Member(id='member1')
self.member2 = data_models.Member(id='member2')
self.in_pool.members = [self.in_member, self.member2]
self.in_member.pool = self.in_pool
self.member2.pool = self.in_pool
def test_update_member(self, mmember):
loadbalancer = data_models.LoadBalancer(id='1')
pool = data_models.Pool(id='1', loadbalancer=loadbalancer,
protocol='HTTPS')
member = data_models.Member(id='1', pool=pool, weight=1)
old_member = data_models.Member(id='1', pool=pool, weight=2)
mmember.side_effect = [member, old_member]
self.mgr.update_member(mock.Mock(), old_member.to_dict(),
member.to_dict())
self.driver_mock.member.update.assert_called_once_with(old_member,
member)
self.update_statuses.assert_called_once_with(member)
def setUp(self):
super(BaseTestEdgeLbaasV2, self).setUp()
self.context = context.get_admin_context()
callbacks = mock.Mock()
callbacks.plugin = mock.Mock()
self.edge_driver = vcns_driver.VcnsDriver(callbacks)
self.lbv2_driver = mock.Mock()
self.core_plugin = mock.Mock()
base_mgr.EdgeLoadbalancerBaseManager._lbv2_driver = self.lbv2_driver
base_mgr.EdgeLoadbalancerBaseManager._core_plugin = self.core_plugin
self._patch_lb_plugin(self.lbv2_driver, self._tested_entity)
self.lb = lb_models.LoadBalancer(LB_ID, LB_TENANT_ID, 'lb-name', '',
'some-subnet', 'port-id', LB_VIP)
self.listener = lb_models.Listener(LISTENER_ID, LB_TENANT_ID,
'l-name', '', None, LB_ID,
'HTTP', protocol_port=80,
loadbalancer=self.lb)
self.pool = lb_models.Pool(POOL_ID, LB_TENANT_ID, 'pool-name', '',
None, 'HTTP', 'ROUND_ROBIN',
listener=self.listener)
self.member = lb_models.Member(MEMBER_ID, LB_TENANT_ID, POOL_ID,
MEMBER_ADDRESS, 80, 1, pool=self.pool)
self.hm = lb_models.HealthMonitor(HM_ID, LB_TENANT_ID, 'PING', 3, 3,
1, pool=self.pool)
def test_update(self):
new_member = lb_models.Member(MEMBER_ID,
LB_TENANT_ID,
POOL_ID,
MEMBER_ADDRESS,
80,
2,
pool=self.pool,
name='member-nnn-nnn')
with mock.patch.object(nsx_db, 'get_nsx_lbaas_pool_binding'
) as mock_get_pool_binding, \
mock.patch.object(self.pool_client, 'get'
) as mock_get_pool, \
mock.patch.object(lb_utils, 'get_network_from_subnet'
) as mock_get_network_from_subnet:
mock_get_pool_binding.return_value = POOL_BINDING
mock_get_pool.return_value = LB_POOL_WITH_MEMBER
mock_get_network_from_subnet.return_value = LB_NETWORK
self.edge_driver.member.update(self.context, self.member,
new_member)
mock_successful_completion = (
self.lbv2_driver.member.successful_completion)
mock_successful_completion.assert_called_with(
self.context, new_member)
def test_update_statuses_member(self):
self.update_statuses_patcher.stop()
member = data_models.Member(id='1')
pool = data_models.Pool(id='1', members=[member])
member.pool = pool
listener = data_models.Listener(id='1', default_pool=pool)
lb = data_models.LoadBalancer(id='1', listeners=[listener])
listener.loadbalancer = lb
pool.loadbalancer = lb
self.mgr._update_statuses(member)
self.assertEqual(2, self.rpc_mock.update_status.call_count)
calls = [mock.call('member', member.id,
provisioning_status=constants.ACTIVE,
operating_status=lb_const.ONLINE),
mock.call('loadbalancer', lb.id,
provisioning_status=constants.ACTIVE,
operating_status=None)]
self.rpc_mock.update_status.assert_has_calls(calls)
self.rpc_mock.update_status.reset_mock()
self.mgr._update_statuses(member, error=True)
self.assertEqual(2, self.rpc_mock.update_status.call_count)
calls = [mock.call('member', member.id,
provisioning_status=constants.ERROR,
operating_status=lb_const.OFFLINE),
mock.call('loadbalancer', lb.id,
provisioning_status=constants.ACTIVE,
operating_status=None)]
self.rpc_mock.update_status.assert_has_calls(calls)
def _create_fake_models(self):
# This id is used for all the entities.
id = 'test_id'
lb = data_models.LoadBalancer(id=id)
sni_container = data_models.SNI(listener_id=id)
listener = data_models.Listener(id=id,
loadbalancer=lb,
sni_containers=[sni_container])
pool = data_models.Pool(id=id, loadbalancer=lb)
member = data_models.Member(id=id, pool=pool)
hm = data_models.HealthMonitor(id=id, pool=pool)
l7policy = data_models.L7Policy(
id=id,
listener=listener,
redirect_pool_id=pool.id,
action=constants.L7_POLICY_ACTION_REDIRECT_TO_POOL)
l7rule = data_models.L7Rule(
id=id,
policy=l7policy,
type=constants.L7_RULE_TYPE_PATH,
compare_type=constants.L7_RULE_COMPARE_TYPE_STARTS_WITH,
value='/api')
lb.listeners = [listener]
lb.pools = [pool]
listener.default_pool = pool
listener.l7policies = [l7policy]
l7policy.rules = [l7rule]
pool.members = [member]
pool.healthmonitor = hm
return lb
def test_update(self):
new_member = lb_models.Member(MEMBER_ID, LB_TENANT_ID, POOL_ID,
MEMBER_ADDRESS, 8000, 1, True,
pool=self.pool)
with mock.patch.object(nsxv_db, 'get_nsxv_lbaas_loadbalancer_binding'
) as mock_get_lb_binding, \
mock.patch.object(nsxv_db, 'get_nsxv_lbaas_pool_binding'
) as mock_get_pool_binding, \
mock.patch.object(self.edge_driver.vcns, 'get_pool'
) as mock_get_pool, \
mock.patch.object(self.edge_driver.vcns, 'update_pool'
) as mock_update_pool:
mock_get_lb_binding.return_value = LB_BINDING
mock_get_pool_binding.return_value = POOL_BINDING
edge_pool_def = EDGE_POOL_DEF.copy()
edge_pool_def['member'] = [EDGE_MEMBER_DEF]
mock_get_pool.return_value = (None, edge_pool_def)
self.edge_driver.member.update(self.context, self.member,
new_member)
edge_member_def = EDGE_MEMBER_DEF.copy()
edge_member_def['port'] = 8000
edge_member_def['monitorPort'] = 8000
edge_member_def['condition'] = 'enabled'
edge_pool_def['member'] = [edge_member_def]
mock_update_pool.assert_called_with(
LB_EDGE_ID, EDGE_POOL_ID, edge_pool_def)
mock_successful_completion = (
self.lbv2_driver.member.successful_completion)
mock_successful_completion.assert_called_with(self.context,
new_member)
def test_delete_member(self, mmember):
loadbalancer = data_models.LoadBalancer(id='1')
pool = data_models.Pool(id='1', loadbalancer=loadbalancer,
protocol='HTTPS')
member = data_models.Member(id='1', pool=pool, weight=1)
mmember.return_value = member
self.mgr.delete_member(mock.Mock(), member.to_dict())
self.driver_mock.member.delete.assert_called_once_with(member)
def _create_fake_models(self, children=True, graph=False):
# This id is used for all the entities.
id = 'test_id'
lb = data_models.LoadBalancer(id=id)
if not children:
return lb
sni_container = data_models.SNI(listener_id=id)
listener = data_models.Listener(id=id,
loadbalancer=lb,
sni_containers=[sni_container])
pool = data_models.Pool(id=id, loadbalancer=lb)
member = data_models.Member(id=id, pool=pool)
hm = data_models.HealthMonitor(id=id, pool=pool)
sp = data_models.SessionPersistence(pool_id=pool.id, pool=pool)
l7policy = data_models.L7Policy(
id=id,
listener=listener,
listener_id=listener.id,
action=constants.L7_POLICY_ACTION_REDIRECT_TO_POOL)
l7rule = data_models.L7Rule(
id=id,
policy=l7policy,
type=constants.L7_RULE_TYPE_PATH,
compare_type=constants.L7_RULE_COMPARE_TYPE_STARTS_WITH,
value='/api')
lb.listeners = [listener]
lb.pools = [pool]
if graph:
r_pool = data_models.Pool(id=id, loadbalancer=lb)
r_member = data_models.Member(id=id, pool=r_pool)
r_pool.members = [r_member]
l7policy.redirect_pool = r_pool
l7policy.redirect_pool_id = r_pool.id
lb.pools.append(r_pool)
else:
l7policy.redirect_pool = pool
l7policy.redirect_pool_id = pool.id
listener.default_pool = pool
listener.l7_policies = [l7policy]
l7policy.rules = [l7rule]
pool.members = [member]
pool.session_persistence = sp
pool.healthmonitor = hm
return lb
def test_update(self):
new_member = lb_models.Member(MEMBER_ID, LB_TENANT_ID, POOL_ID,
MEMBER_ADDRESS, 80, 1, pool=self.pool,
name='member-nnn-nnn')
self.edge_driver.member.update(self.context, self.pool, new_member)
mock_successful_completion = (
self.lbv2_driver.member.successful_completion)
mock_successful_completion.assert_called_with(self.context, new_member)
def test_create_member_failed(self, mmember):
loadbalancer = data_models.LoadBalancer(id='1')
pool = data_models.Pool(id='1', loadbalancer=loadbalancer,
protocol='HTTPS')
member = data_models.Member(id='1', pool=pool)
mmember.return_value = member
self.driver_mock.member.create.side_effect = Exception
self.mgr.create_member(mock.Mock(), member.to_dict())
self.driver_mock.member.create.assert_called_once_with(member)
self.update_statuses.assert_called_once_with(member, error=True)
def setUp(self):
super(BaseTestEdgeLbaasV2, self).setUp()
self.context = context.get_admin_context()
callbacks = mock.Mock()
callbacks.plugin = mock.Mock()
self.edge_driver = vcns_driver.VcnsDriver(callbacks)
self.lbv2_driver = mock.Mock()
self.core_plugin = mock.Mock()
base_mgr.LoadbalancerBaseManager._lbv2_driver = self.lbv2_driver
base_mgr.LoadbalancerBaseManager._core_plugin = self.core_plugin
self._patch_lb_plugin(self.lbv2_driver, self._tested_entity)
self.lb = lb_models.LoadBalancer(LB_ID, LB_TENANT_ID, 'lb-name', '',
'some-subnet', 'port-id', LB_VIP)
self.listener = lb_models.Listener(LISTENER_ID, LB_TENANT_ID,
'l-name', '', None, LB_ID,
'HTTP', protocol_port=80,
loadbalancer=self.lb)
self.sess_persist = lb_models.SessionPersistence(type='HTTP_COOKIE')
self.pool = lb_models.Pool(POOL_ID, LB_TENANT_ID, 'pool-name', '',
None, 'HTTP', 'ROUND_ROBIN',
loadbalancer_id=LB_ID,
listener=self.listener,
listeners=[self.listener],
loadbalancer=self.lb,
session_persistence=self.sess_persist)
self.listener.default_pool = self.pool
self.member = lb_models.Member(MEMBER_ID, LB_TENANT_ID, POOL_ID,
MEMBER_ADDRESS, 80, 1, pool=self.pool)
self.hm = lb_models.HealthMonitor(HM_ID, LB_TENANT_ID, 'PING', 3, 3,
1, pool=self.pool)
self.l7policy = lb_models.L7Policy(L7POL_ID, LB_TENANT_ID,
name='policy-test',
description='policy-desc',
listener_id=LISTENER_ID,
action='REJECT',
listener=self.listener,
position=1)
self.l7rule1 = lb_models.L7Rule(L7RULE_ID1, LB_TENANT_ID,
l7policy_id=L7POL_ID,
compare_type='EQUAL_TO',
invert=False,
type='HEADER',
key='key1',
value='val1',
policy=self.l7policy)
self.l7rule2 = lb_models.L7Rule(L7RULE_ID2, LB_TENANT_ID,
l7policy_id=L7POL_ID,
compare_type='STARTS_WITH',
invert=True,
type='PATH',
value='/images',
policy=self.l7policy)
def test_create_member(self, mmember):
loadbalancer = data_models.LoadBalancer(id='1')
listener = data_models.Listener(id=1,
loadbalancer_id='1',
loadbalancer=loadbalancer)
pool = data_models.Pool(id='1', listener=listener, protocol='HTTPS')
member = data_models.Member(id='1', pool=pool)
mmember.return_value = member
self.mgr.create_member(mock.Mock(), member.to_dict())
self.driver_mock.member.create.assert_called_once_with(member)
self.update_statuses.assert_called_once_with(member)
def _create_fake_models(self):
id = 'name-001'
lb = data_models.LoadBalancer(id=id)
listener = data_models.Listener(id=id, loadbalancer=lb)
pool = data_models.Pool(id=id, listener=listener)
member = data_models.Member(id=id, pool=pool)
hm = data_models.HealthMonitor(id=id, pool=pool)
lb.listeners = [listener]
listener.default_pool = pool
pool.members = [member]
pool.healthmonitor = hm
return lb
def setUp(self):
super(BaseTestEdgeLbaasV2, self).setUp()
self.context = context.get_admin_context()
self.edge_driver = lb_driver_v2.EdgeLoadbalancerDriverV2()
self.lbv2_driver = mock.Mock()
self.core_plugin = mock.Mock()
base_mgr.LoadbalancerBaseManager._lbv2_driver = self.lbv2_driver
base_mgr.LoadbalancerBaseManager._core_plugin = self.core_plugin
self._patch_lb_plugin(self.lbv2_driver, self._tested_entity)
self._patch_nsxlib_lb_clients(self.core_plugin)
self.lb = lb_models.LoadBalancer(LB_ID, LB_TENANT_ID, 'lb1', '',
'some-subnet', 'port-id', LB_VIP)
self.listener = lb_models.Listener(LISTENER_ID, LB_TENANT_ID,
'listener1', 'Dummy', None, LB_ID,
'HTTP', protocol_port=80,
loadbalancer=self.lb)
self.https_listener = lb_models.Listener(
LISTENER_ID, LB_TENANT_ID, 'listener1', '', None, LB_ID,
'HTTPS', protocol_port=443, loadbalancer=self.lb)
self.terminated_https_listener = lb_models.Listener(
LISTENER_ID, LB_TENANT_ID, 'listener1', '', None, LB_ID,
'TERMINATED_HTTPS', protocol_port=443, loadbalancer=self.lb)
self.pool = lb_models.Pool(POOL_ID, LB_TENANT_ID, 'pool1', '',
None, 'HTTP', 'ROUND_ROBIN',
loadbalancer_id=LB_ID,
listener=self.listener,
listeners=[self.listener],
loadbalancer=self.lb)
self.member = lb_models.Member(MEMBER_ID, LB_TENANT_ID, POOL_ID,
MEMBER_ADDRESS, 80, 1, pool=self.pool,
name='member1')
self.hm = lb_models.HealthMonitor(HM_ID, LB_TENANT_ID, 'PING', 3, 3,
1, pool=self.pool, name='hm1')
self.l7policy = lb_models.L7Policy(L7POLICY_ID, LB_TENANT_ID,
name='policy-test',
description='policy-desc',
listener_id=LISTENER_ID,
action='REDIRECT_TO_POOL',
redirect_pool_id=POOL_ID,
listener=self.listener,
position=1)
self.l7rule = lb_models.L7Rule(L7RULE_ID, LB_TENANT_ID,
l7policy_id=L7POLICY_ID,
compare_type='EQUAL_TO',
invert=False,
type='HEADER',
key='key1',
value='val1',
policy=self.l7policy)
def _create_fake_models(self):
# This id is used for all the entities.
id = 'test_id'
lb = data_models.LoadBalancer(id=id)
sni_container = data_models.SNI(listener_id=id)
listener = data_models.Listener(id=id,
loadbalancer=lb,
sni_containers=[sni_container])
pool = data_models.Pool(id=id, listener=listener)
member = data_models.Member(id=id, pool=pool)
hm = data_models.HealthMonitor(id=id, pool=pool)
lb.listeners = [listener]
listener.default_pool = pool
pool.members = [member]
pool.healthmonitor = hm
return lb
def _get_faked_model(obj):
lb = data_models.LoadBalancer(id='test_lb')
if obj == 'lbaas/loadbalancer':
return lb
pool = data_models.Pool(id='test_pool_id', loadbalancer=lb)
if obj == 'lbaas/pool':
return pool
listener = data_models.Listener(id='test_listener_id', loadbalancer=lb)
if obj == 'lbaas/listener':
return listener
member = data_models.Member(id='test_member_id', pool=pool)
if obj == 'lbaas/member':
return member
hm = data_models.HealthMonitor(id='test_health_monitor_id', pool=pool)
return hm
def test_update(self):
old_member = data_models.Member(id=self.in_member.id,
address='0.0.0.0')
self.member_manager.update(old_member, self.in_member)
self.refresh.assert_called_once_with(self.in_lb)
