def get_update_l7rule_flow(self): """Create a flow to update an L7 rule :returns: The flow for updating an L7 rule """ update_l7rule_flow = linear_flow.Flow(constants.UPDATE_L7RULE_FLOW) update_l7rule_flow.add( lifecycle_tasks.L7RuleToErrorOnRevertTask(requires=[ constants.L7RULE, constants.LISTENERS, constants.LOADBALANCER ])) update_l7rule_flow.add( database_tasks.MarkL7RulePendingUpdateInDB( requires=constants.L7RULE)) update_l7rule_flow.add( amphora_driver_tasks.ListenersUpdate( requires=[constants.LOADBALANCER, constants.LISTENERS])) update_l7rule_flow.add( database_tasks.UpdateL7RuleInDB( requires=[constants.L7RULE, constants.UPDATE_DICT])) update_l7rule_flow.add( database_tasks.MarkL7RuleActiveInDB(requires=constants.L7RULE)) update_l7rule_flow.add( database_tasks.MarkL7PolicyActiveInDB(requires=constants.L7POLICY)) update_l7rule_flow.add( database_tasks.MarkLBAndListenersActiveInDB( requires=[constants.LOADBALANCER, constants.LISTENERS])) return update_l7rule_flow
def get_delete_l7rule_flow(self): """Create a flow to delete an L7 rule :returns: The flow for deleting an L7 rule """ delete_l7rule_flow = linear_flow.Flow(constants.DELETE_L7RULE_FLOW) delete_l7rule_flow.add( lifecycle_tasks.L7RuleToErrorOnRevertTask(requires=[ constants.L7RULE, constants.LISTENERS, constants.LOADBALANCER ])) delete_l7rule_flow.add( database_tasks.MarkL7RulePendingDeleteInDB( requires=constants.L7RULE)) delete_l7rule_flow.add( model_tasks.DeleteModelObject( rebind={constants.OBJECT: constants.L7RULE})) delete_l7rule_flow.add( amphora_driver_tasks.ListenersUpdate( requires=[constants.LOADBALANCER, constants.LISTENERS])) delete_l7rule_flow.add( database_tasks.DeleteL7RuleInDB(requires=constants.L7RULE)) delete_l7rule_flow.add( database_tasks.MarkL7PolicyActiveInDB(requires=constants.L7POLICY)) delete_l7rule_flow.add( database_tasks.MarkLBAndListenersActiveInDB( requires=[constants.LOADBALANCER, constants.LISTENERS])) return delete_l7rule_flow
def get_delete_l7rule_flow(self, topology): """Create a flow to delete an L7 rule :returns: The flow for deleting an L7 rule """ delete_l7rule_flow = linear_flow.Flow(constants.DELETE_L7RULE_FLOW) delete_l7rule_flow.add( lifecycle_tasks.L7RuleToErrorOnRevertTask(requires=[ constants.L7RULE, constants.LISTENERS, constants.LOADBALANCER ])) delete_l7rule_flow.add( vthunder_tasks.VthunderInstanceBusy( requires=a10constants.COMPUTE_BUSY)) delete_l7rule_flow.add( database_tasks.MarkL7RulePendingDeleteInDB( requires=constants.L7RULE)) delete_l7rule_flow.add( model_tasks.DeleteModelObject( rebind={constants.OBJECT: constants.L7RULE})) delete_l7rule_flow.add( a10_database_tasks.GetVThunderByLoadBalancer( requires=constants.LOADBALANCER, provides=a10constants.VTHUNDER)) if topology == constants.TOPOLOGY_ACTIVE_STANDBY: delete_l7rule_flow.add( vthunder_tasks.GetMasterVThunder( name=a10constants.GET_MASTER_VTHUNDER, requires=a10constants.VTHUNDER, provides=a10constants.VTHUNDER)) delete_l7rule_flow.add( l7rule_tasks.DeleteL7Rule(requires=[ constants.L7RULE, constants.LISTENERS, a10constants.VTHUNDER ])) delete_l7rule_flow.add( database_tasks.DeleteL7RuleInDB(requires=constants.L7RULE)) delete_l7rule_flow.add( database_tasks.MarkL7PolicyActiveInDB(requires=constants.L7POLICY)) delete_l7rule_flow.add( database_tasks.MarkLBAndListenersActiveInDB( requires=[constants.LOADBALANCER, constants.LISTENERS])) delete_l7rule_flow.add( vthunder_tasks.WriteMemory(requires=a10constants.VTHUNDER)) delete_l7rule_flow.add( a10_database_tasks.SetThunderUpdatedAt( requires=a10constants.VTHUNDER)) return delete_l7rule_flow
def get_update_l7rule_flow(self, topology): """Create a flow to update an L7 rule :returns: The flow for updating an L7 rule """ update_l7rule_flow = linear_flow.Flow(constants.UPDATE_L7RULE_FLOW) update_l7rule_flow.add( lifecycle_tasks.L7RuleToErrorOnRevertTask(requires=[ constants.L7RULE, constants.LISTENERS, constants.LOADBALANCER ])) update_l7rule_flow.add( vthunder_tasks.VthunderInstanceBusy( requires=a10constants.COMPUTE_BUSY)) update_l7rule_flow.add( database_tasks.MarkL7RulePendingUpdateInDB( requires=constants.L7RULE)) update_l7rule_flow.add( a10_database_tasks.GetVThunderByLoadBalancer( requires=constants.LOADBALANCER, provides=a10constants.VTHUNDER)) if topology == constants.TOPOLOGY_ACTIVE_STANDBY: update_l7rule_flow.add( vthunder_tasks.GetMasterVThunder( requires=a10constants.VTHUNDER, provides=a10constants.VTHUNDER)) update_l7rule_flow.add( l7rule_tasks.UpdateL7Rule(requires=[ constants.L7RULE, constants.LISTENERS, a10constants.VTHUNDER, constants.UPDATE_DICT ])) update_l7rule_flow.add( database_tasks.UpdateL7RuleInDB( requires=[constants.L7RULE, constants.UPDATE_DICT])) update_l7rule_flow.add( database_tasks.MarkL7RuleActiveInDB(requires=constants.L7RULE)) update_l7rule_flow.add( database_tasks.MarkL7PolicyActiveInDB(requires=constants.L7POLICY)) update_l7rule_flow.add( database_tasks.MarkLBAndListenersActiveInDB( requires=[constants.LOADBALANCER, constants.LISTENERS])) update_l7rule_flow.add( vthunder_tasks.WriteMemory(requires=a10constants.VTHUNDER)) update_l7rule_flow.add( a10_database_tasks.SetThunderUpdatedAt( requires=a10constants.VTHUNDER)) return update_l7rule_flow
def test_L7RuleToErrorOnRevertTask(self, mock_listener_prov_status_active, mock_loadbalancer_prov_status_active, mock_l7rule_prov_status_error): l7rule_to_error_on_revert = ( lifecycle_tasks.L7RuleToErrorOnRevertTask()) # Execute l7rule_to_error_on_revert.execute(self.L7RULE, self.LISTENERS, self.LOADBALANCER) self.assertFalse(mock_l7rule_prov_status_error.called) # Revert l7rule_to_error_on_revert.revert(self.L7RULE, self.LISTENERS, self.LOADBALANCER) mock_l7rule_prov_status_error.assert_called_once_with(self.L7RULE_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)