def handle_vrid_for_member_subflow(self): handle_vrid_for_member_subflow = linear_flow.Flow( a10constants.HANDLE_VRID_MEMBER_SUBFLOW) handle_vrid_for_member_subflow.add( a10_network_tasks.GetLBResourceSubnet( rebind={a10constants.LB_RESOURCE: constants.MEMBER}, provides=constants.SUBNET)) handle_vrid_for_member_subflow.add( a10_database_tasks.GetChildProjectsOfParentPartition( rebind={a10constants.LB_RESOURCE: constants.MEMBER}, provides=a10constants.PARTITION_PROJECT_LIST)) handle_vrid_for_member_subflow.add( a10_database_tasks.GetVRIDForLoadbalancerResource( requires=a10constants.PARTITION_PROJECT_LIST, provides=a10constants.VRID_LIST)) handle_vrid_for_member_subflow.add( a10_network_tasks.HandleVRIDFloatingIP( requires=[ a10constants.VTHUNDER, a10constants.VRID_LIST, constants.SUBNET ], rebind={a10constants.LB_RESOURCE: constants.MEMBER}, provides=a10constants.VRID_LIST)) handle_vrid_for_member_subflow.add( a10_database_tasks.UpdateVRIDForLoadbalancerResource( requires=a10constants.VRID_LIST, rebind={a10constants.LB_RESOURCE: constants.MEMBER})) return handle_vrid_for_member_subflow
def get_delete_member_vrid_internal_subflow(self): delete_member_vrid_subflow = linear_flow.Flow( a10constants.DELETE_MEMBER_VRID_INTERNAL_SUBFLOW) delete_member_vrid_subflow.add( a10_database_tasks.GetChildProjectsOfParentPartition( rebind={a10constants.LB_RESOURCE: constants.POOL}, provides=a10constants.PARTITION_PROJECT_LIST)) delete_member_vrid_subflow.add( a10_database_tasks.GetSubnetForDeletionInPool( requires=[ a10constants.MEMBER_LIST, a10constants.PARTITION_PROJECT_LIST ], provides=a10constants.SUBNET_LIST)) delete_member_vrid_subflow.add( a10_database_tasks.GetVRIDForLoadbalancerResource( requires=a10constants.PARTITION_PROJECT_LIST, provides=a10constants.VRID_LIST)) delete_member_vrid_subflow.add( a10_network_tasks.DeleteMultipleVRIDPort( requires=[ a10constants.VTHUNDER, a10constants.VRID_LIST, a10constants.SUBNET_LIST ], provides=a10constants.VRID_LIST)) delete_member_vrid_subflow.add( a10_database_tasks.DeleteMultiVRIDEntry( requires=a10constants.VRID_LIST)) return delete_member_vrid_subflow
def get_delete_member_vrid_internal_subflow(self, pool, pool_members): delete_member_vrid_subflow = linear_flow.Flow( a10constants.DELETE_MEMBER_VRID_INTERNAL_SUBFLOW) delete_member_vrid_subflow.add( a10_database_tasks.GetChildProjectsOfParentPartition( name='get_child_project_of_parent_partition' + pool, rebind={a10constants.LB_RESOURCE: pool}, provides=a10constants.PARTITION_PROJECT_LIST )) delete_member_vrid_subflow.add( a10_database_tasks.GetSubnetForDeletionInPool( name='get_subnet_for_deletion_in_pool' + pool, requires=[a10constants.MEMBER_LIST, a10constants.PARTITION_PROJECT_LIST], rebind={a10constants.MEMBER_LIST: pool_members}, provides=a10constants.SUBNET_LIST)) delete_member_vrid_subflow.add( a10_database_tasks.GetVRIDForLoadbalancerResource( name='get_vrid_for_loadbalancer_resource' + pool, requires=a10constants.PARTITION_PROJECT_LIST, provides=a10constants.VRID_LIST)) delete_member_vrid_subflow.add( a10_network_tasks.DeleteMultipleVRIDPort( name='delete_multiple_vrid_port' + pool, requires=[ a10constants.VTHUNDER, a10constants.VRID_LIST, a10constants.SUBNET_LIST], rebind={a10constants.LB_RESOURCE: pool}, provides=a10constants.VRID_LIST)) delete_member_vrid_subflow.add(a10_database_tasks.DeleteMultiVRIDEntry( name='delete_multi_vrid_entry' + pool, requires=a10constants.VRID_LIST)) return delete_member_vrid_subflow
def test_get_child_projects_for_partition(self): vthunder = copy.deepcopy(VTHUNDER) vthunder.partition_name = "mock-partition-name" vthunder.ip_address = "mock-ip-addr" mock_get_projects = task.GetChildProjectsOfParentPartition() mock_get_projects.vthunder_repo.get_project_list_using_partition = mock.Mock( ) mock_get_projects.execute(MEMBER_1, vthunder) mock_get_projects.vthunder_repo.get_project_list_using_partition.\ assert_called_once_with(mock.ANY, partition_name='mock-partition-name', ip_address="mock-ip-addr")
def get_delete_member_vrid_subflow(self): delete_member_vrid_subflow = linear_flow.Flow( a10constants.DELETE_MEMBER_VRID_SUBFLOW) delete_member_vrid_subflow.add( a10_network_tasks.GetLBResourceSubnet( rebind={a10constants.LB_RESOURCE: constants.MEMBER}, provides=constants.SUBNET)) delete_member_vrid_subflow.add( a10_database_tasks.GetChildProjectsOfParentPartition( rebind={a10constants.LB_RESOURCE: constants.MEMBER}, provides=a10constants.PARTITION_PROJECT_LIST)) delete_member_vrid_subflow.add( a10_database_tasks.CountLoadbalancersInProjectBySubnet( requires=[ constants.SUBNET, a10constants.PARTITION_PROJECT_LIST, a10constants.USE_DEVICE_FLAVOR ], provides=a10constants.LB_COUNT_SUBNET)) delete_member_vrid_subflow.add( a10_database_tasks.CountLoadbalancersOnThunderBySubnet( requires=[ a10constants.VTHUNDER, constants.SUBNET, a10constants.USE_DEVICE_FLAVOR ], provides=a10constants.LB_COUNT_THUNDER)) delete_member_vrid_subflow.add( a10_database_tasks.CountMembersInProjectBySubnet( requires=[ constants.SUBNET, a10constants.PARTITION_PROJECT_LIST ], provides=a10constants.MEMBER_COUNT)) delete_member_vrid_subflow.add( a10_database_tasks.GetVRIDForLoadbalancerResource( requires=[ a10constants.PARTITION_PROJECT_LIST, a10constants.VTHUNDER, constants.LOADBALANCER ], provides=a10constants.VRID_LIST)) delete_member_vrid_subflow.add( a10_network_tasks.DeleteVRIDPort( requires=[ a10constants.VTHUNDER, a10constants.VRID_LIST, constants.SUBNET, a10constants.USE_DEVICE_FLAVOR, a10constants.LB_COUNT_SUBNET, a10constants.MEMBER_COUNT, a10constants.LB_COUNT_THUNDER, a10constants.MEMBER_COUNT_THUNDER ], rebind={a10constants.LB_RESOURCE: constants.MEMBER}, provides=(a10constants.VRID, a10constants.DELETE_VRID))) delete_member_vrid_subflow.add( a10_database_tasks.DeleteVRIDEntry( requires=[a10constants.VRID, a10constants.DELETE_VRID])) return delete_member_vrid_subflow
def _get_vthunder_for_amphora_subflow(self, prefix, role): """Subflow to create lb in existing vThunder.""" sf_name = prefix + '-' + a10constants.LB_TO_VTHUNDER_SUBFLOW vthunder_for_amphora_subflow = linear_flow.Flow(sf_name) vthunder_for_amphora_subflow.add( database_tasks.CreateAmphoraInDB(name=sf_name + '-' + constants.CREATE_AMPHORA_INDB, provides=constants.AMPHORA_ID)) vthunder_for_amphora_subflow.add( a10_database_tasks.ValidateComputeForProject( name=sf_name + '-' + a10constants.VALIDATE_COMPUTE_FOR_PROJECT, requires=constants.LOADBALANCER, inject={"role": role}, provides=constants.COMPUTE_ID)) vthunder_for_amphora_subflow.add( database_tasks.UpdateAmphoraComputeId( name=sf_name + '-' + constants.UPDATE_AMPHORA_COMPUTEID, requires=(constants.AMPHORA_ID, constants.COMPUTE_ID))) vthunder_for_amphora_subflow.add( compute_tasks.ComputeActiveWait( name=sf_name + '-' + constants.COMPUTE_WAIT, requires=(constants.COMPUTE_ID, constants.AMPHORA_ID), provides=constants.COMPUTE_OBJ)) vthunder_for_amphora_subflow.add( database_tasks.UpdateAmphoraInfo( name=sf_name + '-' + constants.UPDATE_AMPHORA_INFO, requires=(constants.AMPHORA_ID, constants.COMPUTE_OBJ), provides=constants.AMPHORA)) # create vThunder entry in custom DB vthunder_for_amphora_subflow.add( a10_database_tasks.CreateVThunderEntry( name=sf_name + '-' + a10constants.CREATE_VTHUNDER_ENTRY, requires=(constants.AMPHORA, constants.LOADBALANCER), inject={ "role": role, "status": constants.PENDING_CREATE })) # Get VThunder details from database vthunder_for_amphora_subflow.add( a10_database_tasks.GetVThunderByLoadBalancer( name=sf_name + '-' + a10constants.VTHUNDER_BY_LB, requires=constants.LOADBALANCER, provides=a10constants.VTHUNDER)) vthunder_for_amphora_subflow.add( database_tasks.ReloadLoadBalancer( name=sf_name + '-' + constants.RELOADLOAD_BALANCER, requires=constants.LOADBALANCER_ID, provides=constants.LOADBALANCER)) vthunder_for_amphora_subflow.add( a10_network_tasks.GetLBResourceSubnet( name=sf_name + '-' + a10constants.GET_LB_RESOURCE, rebind={a10constants.LB_RESOURCE: constants.LOADBALANCER}, provides=constants.SUBNET)) vthunder_for_amphora_subflow.add( a10_database_tasks.GetChildProjectsOfParentPartition( name=sf_name + '-' + a10constants.GET_PROJECT_COUNT, requires=[a10constants.VTHUNDER], rebind={a10constants.LB_RESOURCE: constants.LOADBALANCER}, provides=a10constants.PARTITION_PROJECT_LIST)) vthunder_for_amphora_subflow.add( a10_database_tasks.CountLoadbalancersInProjectBySubnet( name=sf_name + '-' + a10constants.GET_LB_COUNT_SUBNET, requires=[ constants.SUBNET, a10constants.PARTITION_PROJECT_LIST ], provides=a10constants.LB_COUNT_SUBNET)) vthunder_for_amphora_subflow.add( a10_network_tasks.AllocateVIP( name=sf_name + '-' + a10constants.ALLOCATE_VIP, requires=[ constants.LOADBALANCER, a10constants.LB_COUNT_SUBNET ], provides=constants.VIP)) vthunder_for_amphora_subflow.add( database_tasks.UpdateVIPAfterAllocation( name=sf_name + '-' + a10constants.UPDATE_VIP_AFTER_ALLOCATION, requires=(constants.LOADBALANCER_ID, constants.VIP), provides=constants.LOADBALANCER)) vthunder_for_amphora_subflow.add( database_tasks.MarkAmphoraAllocatedInDB( name=sf_name + '-' + constants.MARK_AMPHORA_ALLOCATED_INDB, requires=(constants.AMPHORA, constants.LOADBALANCER_ID))) vthunder_for_amphora_subflow.add( database_tasks.ReloadAmphora(name=sf_name + '-' + constants.RELOAD_AMPHORA, requires=constants.AMPHORA_ID, provides=constants.AMPHORA)) vthunder_for_amphora_subflow.add( vthunder_tasks.AllowL2DSR( name=sf_name + '-' + a10constants.ALLOW_L2DSR, requires=(constants.SUBNET, constants.AMPHORA))) if role == constants.ROLE_MASTER: vthunder_for_amphora_subflow.add( database_tasks.MarkAmphoraMasterInDB( name=sf_name + '-' + constants.MARK_AMP_MASTER_INDB, requires=constants.AMPHORA)) elif role == constants.ROLE_BACKUP: vthunder_for_amphora_subflow.add( database_tasks.MarkAmphoraBackupInDB( name=sf_name + '-' + constants.MARK_AMP_BACKUP_INDB, requires=constants.AMPHORA)) elif role == constants.ROLE_STANDALONE: vthunder_for_amphora_subflow.add( database_tasks.MarkAmphoraStandAloneInDB( name=sf_name + '-' + constants.MARK_AMP_STANDALONE_INDB, requires=constants.AMPHORA)) return vthunder_for_amphora_subflow
def _get_vthunder_for_amphora_subflow(self, prefix, role): """Subflow to create lb in existing vThunder.""" sf_name = prefix + '-' + a10constants.LB_TO_VTHUNDER_SUBFLOW vthunder_for_amphora_subflow = linear_flow.Flow(sf_name) vthunder_for_amphora_subflow.add( database_tasks.CreateAmphoraInDB(name=sf_name + '-' + constants.CREATE_AMPHORA_INDB, provides=constants.AMPHORA_ID)) vthunder_for_amphora_subflow.add( a10_database_tasks.ValidateComputeForProject( name=sf_name + '-' + a10constants.VALIDATE_COMPUTE_FOR_PROJECT, requires=constants.LOADBALANCER, inject={"role": role}, provides=constants.COMPUTE_ID)) # if no compute, use spare vThunder and prepare network for spare vThunder vthunder_for_amphora_subflow.add( a10_database_tasks.GetSpareComputeForProject( name=sf_name + '-' + a10constants.GET_SPARE_COMPUTE_FOR_PROJECT, requires=constants.COMPUTE_ID, provides=(constants.COMPUTE_ID, a10constants.SPARE_VTHUNDER))) vthunder_for_amphora_subflow.add( a10_network_tasks.PlugVipNetworkOnSpare( name=sf_name + '-' + a10constants.PLUG_VIP_NETWORK_ON_SPARE, requires=(a10constants.SPARE_VTHUNDER, constants.LOADBALANCER), provides=a10constants.ADDED_NETWORK)) vthunder_for_amphora_subflow.add( vthunder_tasks.SparePostNetworkPlug( name=sf_name + '-' + a10constants.POST_SPARE_PLUG_NETWORK, rebind={a10constants.VTHUNDER: a10constants.SPARE_VTHUNDER}, requires=(a10constants.ADDED_NETWORK))) vthunder_for_amphora_subflow.add( a10_database_tasks.GetVThunderAmphora( name=sf_name + '-' + a10constants.GET_VTHUNDER_AMPHORA, rebind={a10constants.VTHUNDER: a10constants.SPARE_VTHUNDER}, provides=constants.AMPHORA)) vthunder_for_amphora_subflow.add( vthunder_tasks.VThunderComputeConnectivityWait( name=sf_name + '-' + constants.AMP_COMPUTE_CONNECTIVITY_WAIT, rebind={a10constants.VTHUNDER: a10constants.SPARE_VTHUNDER}, requires=(constants.AMPHORA))) vthunder_for_amphora_subflow.add( vthunder_tasks.EnableInterfaceOnSpare( name=sf_name + '-' + a10constants.ENABLE_VTHUNDER_INTERFACE, rebind={a10constants.VTHUNDER: a10constants.SPARE_VTHUNDER}, requires=(a10constants.ADDED_NETWORK))) vthunder_for_amphora_subflow.add( database_tasks.UpdateAmphoraComputeId( name=sf_name + '-' + constants.UPDATE_AMPHORA_COMPUTEID, requires=(constants.AMPHORA_ID, constants.COMPUTE_ID))) vthunder_for_amphora_subflow.add( compute_tasks.ComputeActiveWait( name=sf_name + '-' + constants.COMPUTE_WAIT, requires=(constants.COMPUTE_ID, constants.AMPHORA_ID), provides=constants.COMPUTE_OBJ)) vthunder_for_amphora_subflow.add( database_tasks.UpdateAmphoraInfo( name=sf_name + '-' + constants.UPDATE_AMPHORA_INFO, requires=(constants.AMPHORA_ID, constants.COMPUTE_OBJ), provides=constants.AMPHORA)) # create vThunder entry in custom DB vthunder_for_amphora_subflow.add( a10_database_tasks.CreateVThunderEntry( name=sf_name + '-' + a10constants.CREATE_VTHUNDER_ENTRY, requires=(constants.AMPHORA, constants.LOADBALANCER), inject={ "role": role, "status": constants.PENDING_CREATE })) # Get VThunder details from database vthunder_for_amphora_subflow.add( a10_database_tasks.GetVThunderByLoadBalancer( name=sf_name + '-' + a10constants.VTHUNDER_BY_LB, requires=constants.LOADBALANCER, provides=a10constants.VTHUNDER)) vthunder_for_amphora_subflow.add( database_tasks.ReloadLoadBalancer( name=sf_name + '-' + a10constants.RELOADLOAD_BALANCER, requires=constants.LOADBALANCER_ID, provides=constants.LOADBALANCER)) vthunder_for_amphora_subflow.add( a10_network_tasks.GetLBResourceSubnet( name=sf_name + '-' + a10constants.GET_LB_RESOURCE, rebind={a10constants.LB_RESOURCE: constants.LOADBALANCER}, provides=constants.SUBNET)) vthunder_for_amphora_subflow.add( a10_database_tasks.GetChildProjectsOfParentPartition( name=sf_name + '-' + a10constants.GET_PROJECT_COUNT, requires=[a10constants.VTHUNDER], rebind={a10constants.LB_RESOURCE: constants.LOADBALANCER}, provides=a10constants.PARTITION_PROJECT_LIST)) vthunder_for_amphora_subflow.add( a10_database_tasks.CountLoadbalancersInProjectBySubnet( name=sf_name + '-' + a10constants.GET_LB_COUNT_SUBNET, requires=[ constants.SUBNET, a10constants.PARTITION_PROJECT_LIST ], provides=a10constants.LB_COUNT_SUBNET)) vthunder_for_amphora_subflow.add( a10_network_tasks.AllocateVIP( name=sf_name + '-' + a10constants.ALLOCATE_VIP, requires=[ constants.LOADBALANCER, a10constants.LB_COUNT_SUBNET ], provides=constants.VIP)) vthunder_for_amphora_subflow.add( database_tasks.UpdateVIPAfterAllocation( name=sf_name + '-' + a10constants.UPDATE_VIP_AFTER_ALLOCATION, requires=(constants.LOADBALANCER_ID, constants.VIP), provides=constants.LOADBALANCER)) vthunder_for_amphora_subflow.add( database_tasks.MarkAmphoraAllocatedInDB( name=sf_name + '-' + constants.MARK_AMPHORA_ALLOCATED_INDB, requires=(constants.AMPHORA, constants.LOADBALANCER_ID))) vthunder_for_amphora_subflow.add( database_tasks.ReloadAmphora(name=sf_name + '-' + constants.RELOAD_AMPHORA, requires=constants.AMPHORA_ID, provides=constants.AMPHORA)) if role == constants.ROLE_MASTER: vthunder_for_amphora_subflow.add( database_tasks.MarkAmphoraMasterInDB( name=sf_name + '-' + constants.MARK_AMP_MASTER_INDB, requires=constants.AMPHORA)) vthunder_for_amphora_subflow.add( vthunder_tasks.UpdateAcosVersionInVthunderEntry( name=sf_name + '-' + a10constants.UPDATE_ACOS_VERSION_FOR_BACKUP_VTHUNDER, requires=(a10constants.VTHUNDER))) elif role == constants.ROLE_BACKUP: vthunder_for_amphora_subflow.add( database_tasks.MarkAmphoraBackupInDB( name=sf_name + '-' + constants.MARK_AMP_BACKUP_INDB, requires=constants.AMPHORA)) vthunder_for_amphora_subflow.add( a10_database_tasks.GetBackupVThunderByLoadBalancer( name=sf_name + '-' + a10constants.BACKUP_VTHUNDER, requires=constants.LOADBALANCER, provides=a10constants.BACKUP_VTHUNDER)) vthunder_for_amphora_subflow.add( vthunder_tasks.UpdateAcosVersionInVthunderEntry( name=sf_name + '-' + a10constants.UPDATE_ACOS_VERSION_FOR_BACKUP_VTHUNDER, rebind={ a10constants.VTHUNDER: a10constants.BACKUP_VTHUNDER })) elif role == constants.ROLE_STANDALONE: vthunder_for_amphora_subflow.add( database_tasks.MarkAmphoraStandAloneInDB( name=sf_name + '-' + constants.MARK_AMP_STANDALONE_INDB, requires=constants.AMPHORA)) vthunder_for_amphora_subflow.add( vthunder_tasks.UpdateAcosVersionInVthunderEntry( name=sf_name + '-' + a10constants.UPDATE_ACOS_VERSION_FOR_BACKUP_VTHUNDER, requires=(a10constants.VTHUNDER))) # If spare vThunder is used, remove spare vThunder the database vthunder_for_amphora_subflow.add( a10_database_tasks.DeleteStaleSpareVThunder( name=sf_name + '-' + a10constants.DELETE_STALE_SPARE_VTHUNDER, requires=(a10constants.SPARE_VTHUNDER))) return vthunder_for_amphora_subflow