def get_delete_l7policy_flow(self):
"""Create a flow to delete an L7 policy
:returns: The flow for deleting an L7 policy
"""
delete_l7policy_flow = linear_flow.Flow(constants.DELETE_L7POLICY_FLOW)
delete_l7policy_flow.add(
lifecycle_tasks.L7PolicyToErrorOnRevertTask(requires=[
constants.L7POLICY, constants.LISTENERS, constants.LOADBALANCER
]))
delete_l7policy_flow.add(
database_tasks.MarkL7PolicyPendingDeleteInDB(
requires=constants.L7POLICY))
delete_l7policy_flow.add(
model_tasks.DeleteModelObject(
rebind={constants.OBJECT: constants.L7POLICY}))
#delete_l7policy_flow.add(amphora_driver_tasks.ListenersUpdate(
# requires=[constants.LOADBALANCER, constants.LISTENERS]))
delete_l7policy_flow.add(
a10_database_tasks.GetVThunderByLoadBalancer(
requires=constants.LOADBALANCER,
provides=a10constants.VTHUNDER))
delete_l7policy_flow.add(
handler_l7policy.DeleteL7Policy(
requires=[constants.L7POLICY, a10constants.VTHUNDER]))
delete_l7policy_flow.add(
database_tasks.DeleteL7PolicyInDB(requires=constants.L7POLICY))
delete_l7policy_flow.add(
database_tasks.MarkLBAndListenersActiveInDB(
requires=[constants.LOADBALANCER, constants.LISTENERS]))
return delete_l7policy_flow
def get_create_l7policy_flow(self):
"""Create a flow to create an L7 policy
:returns: The flow for creating an L7 policy
"""
create_l7policy_flow = linear_flow.Flow(constants.CREATE_L7POLICY_FLOW)
create_l7policy_flow.add(
lifecycle_tasks.L7PolicyToErrorOnRevertTask(requires=[
constants.L7POLICY, constants.LISTENERS, constants.LOADBALANCER
]))
create_l7policy_flow.add(
database_tasks.MarkL7PolicyPendingCreateInDB(
requires=constants.L7POLICY))
#create_l7policy_flow.add(amphora_driver_tasks.ListenersUpdate(
# requires=[constants.LOADBALANCER, constants.LISTENERS]))
create_l7policy_flow.add(
a10_database_tasks.GetVThunderByLoadBalancer(
requires=constants.LOADBALANCER,
provides=a10constants.VTHUNDER))
create_l7policy_flow.add(
handler_l7policy.CreateL7Policy(requires=[
constants.L7POLICY, constants.LISTENERS, a10constants.VTHUNDER
]))
create_l7policy_flow.add(
database_tasks.MarkL7PolicyActiveInDB(requires=constants.L7POLICY))
create_l7policy_flow.add(
database_tasks.MarkLBAndListenersActiveInDB(
requires=[constants.LOADBALANCER, constants.LISTENERS]))
return create_l7policy_flow
def get_update_l7policy_flow(self):
"""Create a flow to update an L7 policy
:returns: The flow for updating an L7 policy
"""
update_l7policy_flow = linear_flow.Flow(constants.UPDATE_L7POLICY_FLOW)
update_l7policy_flow.add(
lifecycle_tasks.L7PolicyToErrorOnRevertTask(requires=[
constants.L7POLICY, constants.LISTENERS, constants.LOADBALANCER
]))
update_l7policy_flow.add(
database_tasks.MarkL7PolicyPendingUpdateInDB(
requires=constants.L7POLICY))
update_l7policy_flow.add(
amphora_driver_tasks.ListenersUpdate(
requires=[constants.LOADBALANCER, constants.LISTENERS]))
update_l7policy_flow.add(
database_tasks.UpdateL7PolicyInDB(
requires=[constants.L7POLICY, constants.UPDATE_DICT]))
update_l7policy_flow.add(
database_tasks.MarkL7PolicyActiveInDB(requires=constants.L7POLICY))
update_l7policy_flow.add(
database_tasks.MarkLBAndListenersActiveInDB(
requires=[constants.LOADBALANCER, constants.LISTENERS]))
return update_l7policy_flow
def get_delete_l7policy_flow(self):
"""Create a flow to delete an L7 policy
:returns: The flow for deleting an L7 policy
"""
delete_l7policy_flow = linear_flow.Flow(constants.DELETE_L7POLICY_FLOW)
delete_l7policy_flow.add(
lifecycle_tasks.L7PolicyToErrorOnRevertTask(requires=[
constants.L7POLICY, constants.LISTENERS,
constants.LOADBALANCER_ID
]))
delete_l7policy_flow.add(
database_tasks.MarkL7PolicyPendingDeleteInDB(
requires=constants.L7POLICY))
delete_l7policy_flow.add(
amphora_driver_tasks.ListenersUpdate(
requires=constants.LOADBALANCER_ID))
delete_l7policy_flow.add(
database_tasks.DeleteL7PolicyInDB(requires=constants.L7POLICY))
delete_l7policy_flow.add(
database_tasks.DecrementL7policyQuota(requires=constants.L7POLICY))
delete_l7policy_flow.add(
database_tasks.MarkLBAndListenersActiveInDB(
requires=(constants.LOADBALANCER_ID, constants.LISTENERS)))
return delete_l7policy_flow
def test_L7PolicyToErrorOnRevertTask(self,
mock_listener_prov_status_active,
mock_loadbalancer_prov_status_active,
mock_l7policy_prov_status_error):
l7policy_to_error_on_revert = (
lifecycle_tasks.L7PolicyToErrorOnRevertTask())
# Execute
l7policy_to_error_on_revert.execute(
{constants.L7POLICY_ID: self.L7POLICY_ID}, self.LISTENERS,
self.LOADBALANCER_ID)
self.assertFalse(mock_l7policy_prov_status_error.called)
# Revert
l7policy_to_error_on_revert.revert(
{constants.L7POLICY_ID: self.L7POLICY_ID}, self.LISTENERS,
self.LOADBALANCER_ID)
mock_l7policy_prov_status_error.assert_called_once_with(
self.L7POLICY_ID)
mock_loadbalancer_prov_status_active.assert_called_once_with(
self.LOADBALANCER_ID)
mock_listener_prov_status_active.assert_called_once_with(
self.LISTENER_ID)
def get_create_l7policy_flow(self):
"""Create a flow to create an L7 policy
:returns: The flow for creating an L7 policy
"""
create_l7policy_flow = linear_flow.Flow(constants.CREATE_L7POLICY_FLOW)
create_l7policy_flow.add(
lifecycle_tasks.L7PolicyToErrorOnRevertTask(requires=[
constants.L7POLICY, constants.LISTENERS, constants.LOADBALANCER
]))
create_l7policy_flow.add(
database_tasks.MarkL7PolicyPendingCreateInDB(
requires=constants.L7POLICY))
create_l7policy_flow.add(
amphora_driver_tasks.ListenersUpdate(
requires=constants.LOADBALANCER_ID))
create_l7policy_flow.add(
database_tasks.MarkL7PolicyActiveInDB(requires=constants.L7POLICY))
create_l7policy_flow.add(
database_tasks.MarkLBAndListenersActiveInDB(
requires=(constants.LOADBALANCER_ID, constants.LISTENERS)))
return create_l7policy_flow