def get_create_member_snat_pool_subflow(self): create_member_snat_subflow = linear_flow.Flow( a10constants.CREATE_MEMBER_SNAT_POOL_SUBFLOW) create_member_snat_subflow.add( server_tasks.MemberFindNatPool(requires=[ constants.MEMBER, a10constants.VTHUNDER, constants.POOL, constants.FLAVOR ], provides=a10constants.NAT_FLAVOR)) create_member_snat_subflow.add( a10_database_tasks.GetNatPoolEntry( requires=[constants.MEMBER, a10constants.NAT_FLAVOR], provides=a10constants.NAT_POOL)) create_member_snat_subflow.add( a10_network_tasks.ReserveSubnetAddressForMember( requires=[ constants.MEMBER, a10constants.NAT_FLAVOR, a10constants.NAT_POOL ], provides=a10constants.SUBNET_PORT)) create_member_snat_subflow.add( a10_database_tasks.UpdateNatPoolDB(requires=[ constants.MEMBER, a10constants.NAT_FLAVOR, a10constants.NAT_POOL, a10constants.SUBNET_PORT ])) return create_member_snat_subflow
def test_UpdateNatPoolDB_update(self): db_task = task.UpdateNatPoolDB() db_task.nat_pool_repo = self.nat_pool_repo flavor = { "pool_name": "p1", "start_address": "1.1.1.1", "end_address": "1.1.1.2" } NAT_POOL.member_ref_count = 1 db_task.execute(MEMBER_1, flavor, NAT_POOL, None) self.nat_pool_repo.update.assert_called_with(mock.ANY, mock.ANY, member_ref_count=2)
def test_UpdateNatPoolDB_create(self): db_task = task.UpdateNatPoolDB() db_task.nat_pool_repo = self.nat_pool_repo flavor = { "pool_name": "p1", "start_address": "1.1.1.1", "end_address": "1.1.1.2" } port = mock.Mock(id="1") db_task.execute(MEMBER_1, flavor, None, port) self.nat_pool_repo.create.assert_called_with( mock.ANY, id=mock.ANY, name="p1", subnet_id=MEMBER_1.subnet_id, start_address="1.1.1.1", end_address="1.1.1.2", member_ref_count=1, port_id="1")
def get_rack_vthunder_create_member_flow(self): """Create a flow to create a rack vthunder member :returns: The flow for creating a rack vthunder member """ create_member_flow = linear_flow.Flow(constants.CREATE_MEMBER_FLOW) create_member_flow.add( lifecycle_tasks.MemberToErrorOnRevertTask(requires=[ constants.MEMBER, constants.LISTENERS, constants.LOADBALANCER, constants.POOL ])) create_member_flow.add( database_tasks.MarkMemberPendingCreateInDB( requires=constants.MEMBER)) create_member_flow.add( a10_database_tasks.GetVThunderByLoadBalancer( requires=constants.LOADBALANCER, provides=a10constants.VTHUNDER)) create_member_flow.add( vthunder_tasks.SetupDeviceNetworkMap( requires=a10constants.VTHUNDER, provides=a10constants.VTHUNDER)) create_member_flow.add(self.handle_vrid_for_member_subflow()) if CONF.a10_global.network_type == 'vlan': create_member_flow.add( vthunder_tasks.TagInterfaceForMember( requires=[constants.MEMBER, a10constants.VTHUNDER])) create_member_flow.add( a10_database_tasks.CountMembersWithIP( requires=constants.MEMBER, provides=a10constants.MEMBER_COUNT_IP)) create_member_flow.add( a10_database_tasks.GetFlavorData( rebind={a10constants.LB_RESOURCE: constants.LOADBALANCER}, provides=constants.FLAVOR)) create_member_flow.add( server_tasks.MemberFindNatPool(requires=[ constants.MEMBER, a10constants.VTHUNDER, constants.POOL, constants.FLAVOR ], provides=a10constants.NAT_FLAVOR)) create_member_flow.add( a10_database_tasks.GetNatPoolEntry( requires=[constants.MEMBER, a10constants.NAT_FLAVOR], provides=a10constants.NAT_POOL)) create_member_flow.add( a10_network_tasks.ReserveSubnetAddressForMember( requires=[ constants.MEMBER, a10constants.NAT_FLAVOR, a10constants.NAT_POOL ], provides=a10constants.SUBNET_PORT)) create_member_flow.add( a10_database_tasks.UpdateNatPoolDB(requires=[ constants.MEMBER, a10constants.NAT_FLAVOR, a10constants.NAT_POOL, a10constants.SUBNET_PORT ])) create_member_flow.add( server_tasks.MemberCreate(requires=(constants.MEMBER, a10constants.VTHUNDER, constants.POOL, a10constants.MEMBER_COUNT_IP, constants.FLAVOR))) create_member_flow.add( database_tasks.MarkMemberActiveInDB(requires=constants.MEMBER)) create_member_flow.add( database_tasks.MarkPoolActiveInDB(requires=constants.POOL)) create_member_flow.add( database_tasks.MarkLBAndListenersActiveInDB( requires=(constants.LOADBALANCER, constants.LISTENERS))) create_member_flow.add( vthunder_tasks.WriteMemory(requires=a10constants.VTHUNDER)) create_member_flow.add( a10_database_tasks.SetThunderUpdatedAt( requires=a10constants.VTHUNDER)) return create_member_flow