| Index: tools/telemetry/third_party/gsutilz/third_party/boto/tests/integration/ec2/vpc/test_connection.py
|
| diff --git a/tools/telemetry/third_party/gsutilz/third_party/boto/tests/integration/ec2/vpc/test_connection.py b/tools/telemetry/third_party/gsutilz/third_party/boto/tests/integration/ec2/vpc/test_connection.py
|
| deleted file mode 100755
|
| index 6e7b28896bf779c894a9a7cefdec9e5ff8f85bf8..0000000000000000000000000000000000000000
|
| --- a/tools/telemetry/third_party/gsutilz/third_party/boto/tests/integration/ec2/vpc/test_connection.py
|
| +++ /dev/null
|
| @@ -1,213 +0,0 @@
|
| -#!/usr/bin/env python
|
| -# Copyright (c) 2012 Amazon.com, Inc. or its affiliates. All Rights Reserved
|
| -#
|
| -# Permission is hereby granted, free of charge, to any person obtaining a
|
| -# copy of this software and associated documentation files (the
|
| -# "Software"), to deal in the Software without restriction, including
|
| -# without limitation the rights to use, copy, modify, merge, publish, dis-
|
| -# tribute, sublicense, and/or sell copies of the Software, and to permit
|
| -# persons to whom the Software is furnished to do so, subject to the fol-
|
| -# lowing conditions:
|
| -#
|
| -# The above copyright notice and this permission notice shall be included
|
| -# in all copies or substantial portions of the Software.
|
| -#
|
| -# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
|
| -# OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABIL-
|
| -# ITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT
|
| -# SHALL THE AUTHOR BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY,
|
| -# WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
| -# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS
|
| -# IN THE SOFTWARE.
|
| -#
|
| -import time
|
| -
|
| -import boto
|
| -from boto.compat import six
|
| -from tests.compat import unittest
|
| -from boto.ec2.networkinterface import NetworkInterfaceCollection
|
| -from boto.ec2.networkinterface import NetworkInterfaceSpecification
|
| -from boto.ec2.networkinterface import PrivateIPAddress
|
| -
|
| -
|
| -class TestVPCConnection(unittest.TestCase):
|
| -
|
| - def setUp(self):
|
| - # Registry of instances to be removed
|
| - self.instances = []
|
| - # Registry for cleaning up the vpc after all instances are terminated
|
| - # in the format [ ( func, (arg1, ... argn) ) ]
|
| - self.post_terminate_cleanups = []
|
| -
|
| - self.api = boto.connect_vpc()
|
| - self.vpc = self.api.create_vpc('10.0.0.0/16')
|
| -
|
| - # Need time for the VPC to be in place. :/
|
| - time.sleep(5)
|
| - self.subnet = self.api.create_subnet(self.vpc.id, '10.0.0.0/24')
|
| - # Register the subnet to be deleted after instance termination
|
| - self.post_terminate_cleanups.append((self.api.delete_subnet, (self.subnet.id,)))
|
| -
|
| - # Need time for the subnet to be in place.
|
| - time.sleep(10)
|
| -
|
| - def post_terminate_cleanup(self):
|
| - """Helper to run clean up tasks after instances are removed."""
|
| - for fn, args in self.post_terminate_cleanups:
|
| - fn(*args)
|
| - # Give things time to catch up each time
|
| - time.sleep(10)
|
| -
|
| - # Now finally delete the vpc
|
| - if self.vpc:
|
| - self.api.delete_vpc(self.vpc.id)
|
| -
|
| - def terminate_instances(self):
|
| - """Helper to remove all instances and kick off additional cleanup
|
| - once they are terminated.
|
| - """
|
| - for instance in self.instances:
|
| - self.terminate_instance(instance)
|
| - self.post_terminate_cleanup()
|
| -
|
| - def terminate_instance(self, instance):
|
| - instance.terminate()
|
| - for i in six.moves.range(300):
|
| - instance.update()
|
| - if instance.state == 'terminated':
|
| - # Give it a litle more time to settle.
|
| - time.sleep(30)
|
| - return
|
| - else:
|
| - time.sleep(10)
|
| -
|
| - def delete_elastic_ip(self, eip):
|
| - # Fetch a new copy of the eip so we're up to date
|
| - new_eip = self.api.get_all_addresses([eip.public_ip])[0]
|
| - if new_eip.association_id:
|
| - new_eip.disassociate()
|
| - new_eip.release()
|
| - time.sleep(10)
|
| -
|
| - def test_multi_ip_create(self):
|
| - interface = NetworkInterfaceSpecification(
|
| - device_index=0, subnet_id=self.subnet.id,
|
| - private_ip_address='10.0.0.21',
|
| - description="This is a test interface using boto.",
|
| - delete_on_termination=True, private_ip_addresses=[
|
| - PrivateIPAddress(private_ip_address='10.0.0.22',
|
| - primary=False),
|
| - PrivateIPAddress(private_ip_address='10.0.0.23',
|
| - primary=False),
|
| - PrivateIPAddress(private_ip_address='10.0.0.24',
|
| - primary=False)])
|
| - interfaces = NetworkInterfaceCollection(interface)
|
| -
|
| - reservation = self.api.run_instances(image_id='ami-a0cd60c9', instance_type='m1.small',
|
| - network_interfaces=interfaces)
|
| - # Give it a few seconds to start up.
|
| - time.sleep(10)
|
| - instance = reservation.instances[0]
|
| - self.addCleanup(self.terminate_instance, instance)
|
| - retrieved = self.api.get_all_reservations(instance_ids=[instance.id])
|
| - self.assertEqual(len(retrieved), 1)
|
| - retrieved_instances = retrieved[0].instances
|
| - self.assertEqual(len(retrieved_instances), 1)
|
| - retrieved_instance = retrieved_instances[0]
|
| -
|
| - self.assertEqual(len(retrieved_instance.interfaces), 1)
|
| - interface = retrieved_instance.interfaces[0]
|
| -
|
| - private_ip_addresses = interface.private_ip_addresses
|
| - self.assertEqual(len(private_ip_addresses), 4)
|
| - self.assertEqual(private_ip_addresses[0].private_ip_address,
|
| - '10.0.0.21')
|
| - self.assertEqual(private_ip_addresses[0].primary, True)
|
| - self.assertEqual(private_ip_addresses[1].private_ip_address,
|
| - '10.0.0.22')
|
| - self.assertEqual(private_ip_addresses[2].private_ip_address,
|
| - '10.0.0.23')
|
| - self.assertEqual(private_ip_addresses[3].private_ip_address,
|
| - '10.0.0.24')
|
| -
|
| - def test_associate_public_ip(self):
|
| - # Supplying basically nothing ought to work.
|
| - interface = NetworkInterfaceSpecification(
|
| - associate_public_ip_address=True,
|
| - subnet_id=self.subnet.id,
|
| - # Just for testing.
|
| - delete_on_termination=True
|
| - )
|
| - interfaces = NetworkInterfaceCollection(interface)
|
| -
|
| - reservation = self.api.run_instances(
|
| - image_id='ami-a0cd60c9',
|
| - instance_type='m1.small',
|
| - network_interfaces=interfaces
|
| - )
|
| - instance = reservation.instances[0]
|
| - self.instances.append(instance)
|
| - self.addCleanup(self.terminate_instances)
|
| -
|
| - # Give it a **LONG** time to start up.
|
| - # Because the public IP won't be there right away.
|
| - time.sleep(60)
|
| -
|
| - retrieved = self.api.get_all_reservations(
|
| - instance_ids=[
|
| - instance.id
|
| - ]
|
| - )
|
| - self.assertEqual(len(retrieved), 1)
|
| - retrieved_instances = retrieved[0].instances
|
| - self.assertEqual(len(retrieved_instances), 1)
|
| - retrieved_instance = retrieved_instances[0]
|
| -
|
| - self.assertEqual(len(retrieved_instance.interfaces), 1)
|
| - interface = retrieved_instance.interfaces[0]
|
| -
|
| - # There ought to be a public IP there.
|
| - # We can't reason about the IP itself, so just make sure it vaguely
|
| - # resembles an IP (& isn't empty/``None``)...
|
| - self.assertTrue(interface.publicIp.count('.') >= 3)
|
| -
|
| - def test_associate_elastic_ip(self):
|
| - interface = NetworkInterfaceSpecification(
|
| - associate_public_ip_address=False,
|
| - subnet_id=self.subnet.id,
|
| - # Just for testing.
|
| - delete_on_termination=True
|
| - )
|
| - interfaces = NetworkInterfaceCollection(interface)
|
| -
|
| - reservation = self.api.run_instances(
|
| - image_id='ami-a0cd60c9',
|
| - instance_type='m1.small',
|
| - network_interfaces=interfaces
|
| - )
|
| - instance = reservation.instances[0]
|
| - # Register instance to be removed
|
| - self.instances.append(instance)
|
| - # Add terminate instances helper as cleanup command
|
| - self.addCleanup(self.terminate_instances)
|
| -
|
| - # Create an internet gateway so we can attach an eip
|
| - igw = self.api.create_internet_gateway()
|
| - # Wait on gateway before attaching
|
| - time.sleep(5)
|
| - # Attach and register clean up tasks
|
| - self.api.attach_internet_gateway(igw.id, self.vpc.id)
|
| - self.post_terminate_cleanups.append((self.api.detach_internet_gateway, (igw.id, self.vpc.id)))
|
| - self.post_terminate_cleanups.append((self.api.delete_internet_gateway, (igw.id,)))
|
| -
|
| - # Allocate an elastic ip to this vpc
|
| - eip = self.api.allocate_address('vpc')
|
| - self.post_terminate_cleanups.append((self.delete_elastic_ip, (eip,)))
|
| -
|
| - # Wait on instance and eip then try to associate directly to instance
|
| - time.sleep(60)
|
| - eip.associate(instance.id)
|
| -
|
| -
|
| -if __name__ == '__main__':
|
| - unittest.main()
|
|
|