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
