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