def test_delete(self):
members = [[], [fake_objs.FakeMember()]]
hms = [None, fake_objs.FakeHM('PING')]
persistences = [None, 'SOURCE_IP', 'HTTP_COOKIE']
listeners = [False, True]
for m in members:
for hm in hms:
for pers in persistences:
for lst in listeners:
self.a.reset_mocks()
pool = fake_objs.FakePool('TCP',
'ROUND_ROBIN',
pers,
lst,
members=m,
hm=hm)
self.a.pool.neutron.member_count.return_value = 1
self.a.pool.delete(None, pool)
self.print_mocks()
(self.a.last_client.slb.service_group.delete.
assert_called_with(pool.id))
if pers == 'SOURCE_IP':
(self.a.last_client.slb.template.
src_ip_persistence.delete.assert_called_with(
pool.id))
elif pers == 'HTTP_COOKIE':
(self.a.last_client.slb.template.
cookie_persistence.delete.assert_called_with(
pool.id))
def test_delete(self):
m = fake_objs.FakeMember(False, pool=mock.MagicMock())
ip = self.a.member.neutron.member_get_ip(None, m, True)
self.set_count_1()
self.a.member.delete(None, m)
self.a.last_client.slb.server.delete(ip)
def test_delete_count_gt_one(self):
m = fake_objs.FakeMember(False, pool=mock.MagicMock())
ip = self.a.member.neutron.member_get_ip(None, m, True)
name = self.a.member._get_name(m, ip)
self.set_count_2()
self.a.member.delete(None, m)
self.a.last_client.slb.service_group.member.delete.assert_called_with(
m.pool_id, name, m.protocol_port)
def test_update_down(self):
m = fake_objs.FakeMember(False, pool=mock.MagicMock())
ip = self.a.member.neutron.member_get_ip(None, m, True)
name = self.a.member._get_name(m, ip)
self.a.member.update(None, m, m)
self.a.last_client.slb.service_group.member.update.assert_called_with(
m.pool.id,
name,
m.protocol_port,
self.a.last_client.slb.DOWN,
axapi_args={'member': {}})
def test_create_with_template(self, ):
template = {"server": {"template-server": "sg1"}}
expect = {'template-server': 'sg1'}
for k, v in self.a.config.get_devices().items():
v['templates'] = template
m = fake_objs.FakeMember(admin_state_up=True, pool=mock.MagicMock())
self.a.member.create(None, m)
self.a.last_client.slb.server.create.assert_called_with(
mock.ANY,
mock.ANY,
status=mock.ANY,
config_defaults=mock.ANY,
server_templates=expect,
axapi_args={'server': {}})
def test_create_empty_name_noexception(self):
self.a.config.get_member_expressions = self._get_expressions_mock
admin_state = self.a.last_client.slb.UP
m = fake_objs.FakeMember(admin_state_up=admin_state,
pool=mock.MagicMock())
m.name = None
handler = self.a.member
handler.create(None, m)
self.a.last_client.slb.server.create.assert_called_with(
mock.ANY,
mock.ANY,
status=mock.ANY,
config_defaults={},
axapi_args={'server': {}})
def _test_create(self,
admin_state_up=True,
uuid_name=False,
conn_limit=None,
conn_resume=None):
if uuid_name:
old = self.a.config.get('member_name_use_uuid')
self.a.config._config.member_name_use_uuid = True
m = fake_objs.FakeMember(admin_state_up=admin_state_up,
pool=mock.MagicMock())
ip = self.a.member.neutron.member_get_ip(None, m, True)
if uuid_name:
name = m.id
else:
name = self.a.member._get_name(m, ip)
self.a.member.create(None, m)
if admin_state_up:
status = self.a.last_client.slb.UP
else:
status = self.a.last_client.slb.DOWN
server_args = {}
if conn_limit is not None:
if conn_limit > 0 and conn_limit <= 8000000:
server_args['conn-limit'] = conn_limit
if conn_resume is not None:
if conn_resume > 0 and conn_resume <= 1000000:
server_args['conn-resume'] = conn_resume
self.a.last_client.slb.server.create.assert_called_with(
name,
ip,
status=status,
server_templates=None,
config_defaults=mock.ANY,
axapi_args={'server': server_args})
self.a.last_client.slb.service_group.member.create.assert_called_with(
m.pool.id,
name,
m.protocol_port,
status=status,
axapi_args={'member': {}})
if uuid_name:
self.a.config._config.member_name_use_uuid = old
def _test_create_expressions(self, os_name, pattern, expressions=None):
self.a.config.get_member_expressions = self._get_expressions_mock
expressions = expressions or self.a.config.get_member_expressions()
expected = expressions.get(pattern, {}).get("json", {})
admin_state = self.a.last_client.slb.UP
m = fake_objs.FakeMember(admin_state_up=admin_state,
pool=mock.MagicMock())
m.name = os_name
handler = self.a.member
handler.create(None, m)
# s = str(self.a.last_client.mock_calls)
self.a.last_client.slb.server.create.assert_called_with(
mock.ANY,
mock.ANY,
status=mock.ANY,
config_defaults=expected,
axapi_args={'server': {}})
def test_count(self):
self.a.member.neutron.member_count(
None, fake_objs.FakeMember(pool=mock.MagicMock()))
def test_get_name(self):
m = fake_objs.FakeMember(pool=mock.MagicMock())
z = self.a.member._get_name(m, '1.1.1.1')
self.assertEqual(z, '_get-o_1_1_1_1_neutron')
def test_get_ip(self):
m = fake_objs.FakeMember(pool=mock.MagicMock())
self.a.member.neutron.member_get_ip(None, m, False)
self.print_mocks()
self.a.neutron.member_get_ip.assert_called_with(None, m, False)