def test_create_and_update_pools(self, pool_create_args=None,
pool_update_args=None):
neutron_scenario = loadbalancer_v1.NeutronLoadbalancerV1(
self._get_context())
pools = [{
"pool": {
"id": "pool-id"
}
}]
updated_pool = {
"pool": {
"id": "pool-id",
"name": "updated-pool",
"admin_state_up": True
}
}
pool_data = pool_create_args or {}
pool_update_args = pool_update_args or {}
pool_update_args.update({"name": "_updated", "admin_state_up": True})
neutron_scenario._create_v1_pools = mock.Mock(return_value=pools)
neutron_scenario._update_v1_pool = mock.Mock(
return_value=updated_pool)
networks = self._get_context()["tenant"]["networks"]
neutron_scenario.create_and_update_pools(
pool_create_args=pool_data,
pool_update_args=pool_update_args)
self.assertEqual([mock.call(networks, **pool_data)],
neutron_scenario._create_v1_pools.mock_calls)
for pool in pools:
neutron_scenario._update_v1_pool.assert_called_once_with(
pool, **pool_update_args)
def _validate_create_and_update_pools_scenario(self, pool_create_args):
neutron_scenario = loadbalancer_v1.NeutronLoadbalancerV1(
self._get_context())
pool = {"pool": {"id": "pool-id"}}
updated_pool = {
"pool": {
"id": "pool-id",
"name": "updated-pool",
"admin_state_up": True
}
}
pool_update_args = {"name": "_updated", "admin_state_up": True}
neutron_scenario._create_v1_pool = mock.Mock(return_value=pool)
neutron_scenario._update_v1_pool = mock.Mock(return_value=updated_pool)
neutron_scenario.create_and_update_pools(
pool_create_args=pool_create_args,
pool_update_args=pool_update_args)
pools = []
for net in self._get_context()["tenant"]["networks"]:
for subnet_id in net["subnets"]:
pools.append(
neutron_scenario._create_v1_pool.assert_called_once_with(
subnet_id, **pool_create_args))
for pool in pools:
neutron_scenario._update_v1_pool.assert_called_once_with(
neutron_scenario._create_v1_pool.return_value,
**pool_update_args)
def test_create_and_delete_vips(self, pool_create_args=None,
vip_create_args=None):
pools = [{
"pool": {
"id": "pool-id"
}
}]
vip = {
"vip": {
"id": "vip-id"
}
}
neutron_scenario = loadbalancer_v1.NeutronLoadbalancerV1(
self._get_context())
vip_data = vip_create_args or {}
pool_data = pool_create_args or {}
networks = self._get_context()["tenant"]["networks"]
neutron_scenario._create_v1_pools = mock.Mock(return_value=pools)
neutron_scenario._create_v1_vip = mock.Mock(return_value=vip)
neutron_scenario._delete_v1_vip = mock.Mock()
neutron_scenario.create_and_delete_vips(
pool_create_args=pool_create_args,
vip_create_args=vip_create_args)
neutron_scenario._create_v1_pools.assert_called_once_with(
networks, **pool_data)
neutron_scenario._create_v1_vip.assert_has_calls(
[mock.call(pool, **vip_data) for pool in pools])
neutron_scenario._delete_v1_vip.assert_has_calls(
[mock.call(vip["vip"])])
def test_create_and_update_vips(self,
pool_create_args=None,
vip_create_args=None,
vip_update_args=None):
neutron_scenario = loadbalancer_v1.NeutronLoadbalancerV1(
self._get_context())
pools = [{
"pool": {
"id": "pool-id",
}
}]
expected_vip = {"vip": {"id": "vip-id", "name": "vip-name"}}
updated_vip = {"vip": {"id": "vip-id", "name": "updated-vip-name"}}
vips = [expected_vip]
vip_data = vip_create_args or {}
vip_update_data = vip_update_args or {}
pool_data = pool_create_args or {}
networks = self._get_context()["tenant"]["networks"]
neutron_scenario._create_v1_pools = mock.Mock(return_value=pools)
neutron_scenario._create_v1_vip = mock.Mock(return_value=expected_vip)
neutron_scenario._update_v1_vip = mock.Mock(return_value=updated_vip)
neutron_scenario.create_and_update_vips(
pool_create_args=pool_create_args,
vip_create_args=vip_create_args,
vip_update_args=vip_update_args)
neutron_scenario._create_v1_pools.assert_called_once_with(
networks, **pool_data)
neutron_scenario._create_v1_vip.assert_has_calls(
[mock.call(pool, **vip_data) for pool in pools])
neutron_scenario._update_v1_vip.assert_has_calls(
[mock.call(vip, **vip_update_data) for vip in vips])
def _validate_create_and_list_pools_scenario(self, pool_create_args):
neutron_scenario = loadbalancer_v1.NeutronLoadbalancerV1(
self._get_context())
neutron_scenario._create_v1_pool = mock.Mock()
neutron_scenario._list_v1_pools = mock.Mock()
neutron_scenario.create_and_list_pools(
pool_create_args=pool_create_args)
for net in self._get_context()["tenant"]["networks"]:
for subnet_id in net["subnets"]:
neutron_scenario._create_v1_pool.assert_called_once_with(
subnet_id, **pool_create_args)
neutron_scenario._list_v1_pools.assert_called_once_with()
def test_create_and_list_pools(self, pool_create_args=None):
neutron_scenario = loadbalancer_v1.NeutronLoadbalancerV1(
self._get_context())
pool_data = pool_create_args or {}
networks = self._get_context()["tenant"]["networks"]
neutron_scenario._create_v1_pools = mock.Mock()
neutron_scenario._list_v1_pools = mock.Mock()
neutron_scenario.create_and_list_pools(
pool_create_args=pool_create_args)
neutron_scenario._create_v1_pools.assert_called_once_with(
networks, **pool_data)
neutron_scenario._list_v1_pools.assert_called_once_with()
def test_create_and_list_healthmonitors(self,
healthmonitor_create_args=None):
neutron_scenario = loadbalancer_v1.NeutronLoadbalancerV1(
self._get_context())
hm_data = healthmonitor_create_args or {}
neutron_scenario._create_v1_healthmonitor = mock.Mock()
neutron_scenario._list_v1_healthmonitors = mock.Mock()
neutron_scenario.create_and_list_healthmonitors(
healthmonitor_create_args=healthmonitor_create_args)
neutron_scenario._create_v1_healthmonitor.assert_called_once_with(
**hm_data)
neutron_scenario._list_v1_healthmonitors.assert_called_once_with()
def test_create_and_delete_pools(self, pool_create_args=None):
neutron_scenario = loadbalancer_v1.NeutronLoadbalancerV1(
self._get_context())
pools = [{"pool": {"id": "pool-id"}}]
pool_data = pool_create_args or {}
networks = self._get_context()["tenant"]["networks"]
neutron_scenario._create_v1_pools = mock.Mock(return_value=pools)
neutron_scenario._delete_v1_pool = mock.Mock()
neutron_scenario.create_and_delete_pools(
pool_create_args=pool_create_args)
self.assertEqual([mock.call(networks, **pool_data)],
neutron_scenario._create_v1_pools.mock_calls)
for pool in pools:
self.assertEqual(1, neutron_scenario._delete_v1_pool.call_count)
def test_create_and_delete_healthmonitors(self,
healthmonitor_create_args=None):
neutron_scenario = loadbalancer_v1.NeutronLoadbalancerV1(
self._get_context())
hm = {"health_monitor": {"id": "hm-id"}}
hm_data = healthmonitor_create_args or {}
neutron_scenario._create_v1_healthmonitor = mock.Mock(return_value=hm)
neutron_scenario._delete_v1_healthmonitor = mock.Mock()
neutron_scenario.create_and_delete_healthmonitors(
healthmonitor_create_args=healthmonitor_create_args)
neutron_scenario._create_v1_healthmonitor.assert_called_once_with(
**hm_data)
neutron_scenario._delete_v1_healthmonitor.assert_called_once_with(
neutron_scenario._create_v1_healthmonitor.return_value[
"health_monitor"])
def _validate_create_and_delete_pools_scenario(self, pool_create_args):
neutron_scenario = loadbalancer_v1.NeutronLoadbalancerV1(
self._get_context())
pool = {"pool": {"id": "pool-id"}}
neutron_scenario._create_v1_pool = mock.Mock(return_value=pool)
neutron_scenario._delete_v1_pool = mock.Mock()
neutron_scenario.create_and_delete_pools(
pool_create_args=pool_create_args)
pools = []
for net in self._get_context()["tenant"]["networks"]:
for subnet_id in net["subnets"]:
self.assertEqual(
[mock.call(subnet_id=subnet_id, **pool_create_args)],
neutron_scenario._create_v1_pool.mock_calls)
for pool in pools:
self.assertEqual(1, neutron_scenario._delete_v1_pool.call_count)
def test_create_and_update_healthmonitors(self,
healthmonitor_create_args=None,
healthmonitor_update_args=None):
neutron_scenario = loadbalancer_v1.NeutronLoadbalancerV1(
self._get_context())
mock_random = loadbalancer_v1.random = mock.Mock()
hm = {"healthmonitor": {"id": "hm-id"}}
hm_data = healthmonitor_create_args or {}
hm_update_data = healthmonitor_update_args or {
"max_retries": mock_random.choice.return_value}
neutron_scenario._create_v1_healthmonitor = mock.Mock(return_value=hm)
neutron_scenario._update_v1_healthmonitor = mock.Mock()
neutron_scenario.create_and_update_healthmonitors(
healthmonitor_create_args=healthmonitor_create_args,
healthmonitor_update_args=healthmonitor_update_args)
neutron_scenario._create_v1_healthmonitor.assert_called_once_with(
**hm_data)
neutron_scenario._update_v1_healthmonitor.assert_called_once_with(
neutron_scenario._create_v1_healthmonitor.return_value,
**hm_update_data)
