def test_associate_floating_ip_server(self, flow_mock, sync_point_mock, ensure_floating_ip_mock): fixed_ip_binding = "fixed-ip-{}".format(self.test_instance_uuid) flow = floating_ip.associate_floating_ip_server( self.context, self.test_address, self.fixed_ip_info, self.test_instance_uuid) flow_mock.assert_called_once_with("associate-floating-ip-{}-server-{}" .format(self.test_address, self.test_instance_uuid)) self.assertEqual(flow.add.call_args_list, [call(sync_point_mock()), call(ensure_floating_ip_mock())])
def restore_floating_ips(context, server_info): flow = unordered_flow.Flow("post-migration-{}".format(server_info["id"])) addresses = server_info["addresses"] for label in addresses: fixed_ip = addresses[label][0] for floating_ip in [addr["addr"] for addr in addresses[label] if addr['OS-EXT-IPS:type'] == 'floating']: fip_retrieve = "floating-ip-{}-retrieve".format(floating_ip) if fip_retrieve in context.store: fip_flow = fip_tasks.associate_floating_ip_server( context, floating_ip, fixed_ip, server_info["id"]) flow.add(fip_flow) else: raise exceptions.NotFound() return flow