gcloud 0.0.4 → 0.0.6

data/packages/gsutil/boto/tests/integration/ec2/elb/__init__.py

@@ -0,0 +1,20 @@
+# Copyright (c) 2006-2011 Mitch Garnaat http://garnaat.org/
+#
+# 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.

data/packages/gsutil/boto/tests/integration/ec2/elb/test_cert_verification.py

@@ -0,0 +1,40 @@
+# Copyright (c) 2012 Mitch Garnaat http://garnaat.org/
+# 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.
+
+"""
+Check that all of the certs on all service endpoints validate.
+"""
+
+import unittest
+import boto.ec2.elb
+
+
+class CertVerificationTest(unittest.TestCase):
+
+    elb = True
+    ssl = True
+
+    def test_certs(self):
+        for region in boto.ec2.elb.regions():
+            c = region.connect()
+            c.get_all_load_balancers()

data/packages/gsutil/boto/tests/integration/ec2/elb/test_connection.py

@@ -0,0 +1,130 @@
+# Copyright (c) 2010 Hunter Blanks http://artifex.org/~hblanks/
+# 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.
+
+"""
+Initial, and very limited, unit tests for ELBConnection.
+"""
+
+import unittest
+from boto.ec2.elb import ELBConnection
+
+class ELBConnectionTest(unittest.TestCase):
+    ec2 = True
+
+    def tearDown(self):
+        """ Deletes all load balancers after every test. """
+        for lb in ELBConnection().get_all_load_balancers():
+            lb.delete()
+
+    def test_build_list_params(self):
+        c = ELBConnection()
+        params = {}
+        c.build_list_params(
+            params, ['thing1', 'thing2', 'thing3'], 'ThingName%d')
+        expected_params = {
+            'ThingName1': 'thing1',
+            'ThingName2': 'thing2',
+            'ThingName3': 'thing3'
+            }
+        self.assertEqual(params, expected_params)
+
+    # TODO: for these next tests, consider sleeping until our load
+    # balancer comes up, then testing for connectivity to
+    # balancer.dns_name, along the lines of the existing EC2 unit tests.
+
+    def test_create_load_balancer(self):
+        c = ELBConnection()
+        name = 'elb-boto-unit-test'
+        availability_zones = ['us-east-1a']
+        listeners = [(80, 8000, 'HTTP')]
+        balancer = c.create_load_balancer(name, availability_zones, listeners)
+        self.assertEqual(balancer.name, name)
+        self.assertEqual(balancer.availability_zones, availability_zones)
+        self.assertEqual(balancer.listeners, listeners)
+
+        balancers = c.get_all_load_balancers()
+        self.assertEqual([lb.name for lb in balancers], [name])
+
+    def test_create_load_balancer_listeners(self):
+        c = ELBConnection()
+        name = 'elb-boto-unit-test'
+        availability_zones = ['us-east-1a']
+        listeners = [(80, 8000, 'HTTP')]
+        balancer = c.create_load_balancer(name, availability_zones, listeners)
+
+        more_listeners = [(443, 8001, 'HTTP')]
+        c.create_load_balancer_listeners(name, more_listeners)
+        balancers = c.get_all_load_balancers()
+        self.assertEqual([lb.name for lb in balancers], [name])
+        self.assertEqual(
+            sorted(l.get_tuple() for l in balancers[0].listeners),
+            sorted(listeners + more_listeners)
+            )
+
+    def test_delete_load_balancer_listeners(self):
+        c = ELBConnection()
+        name = 'elb-boto-unit-test'
+        availability_zones = ['us-east-1a']
+        listeners = [(80, 8000, 'HTTP'), (443, 8001, 'HTTP')]
+        balancer = c.create_load_balancer(name, availability_zones, listeners)
+
+        balancers = c.get_all_load_balancers()
+        self.assertEqual([lb.name for lb in balancers], [name])
+        self.assertEqual(
+            sorted([l.get_tuple() for l in balancers[0].listeners]),
+            sorted(listeners))
+
+        c.delete_load_balancer_listeners(name, [443])
+        balancers = c.get_all_load_balancers()
+        self.assertEqual([lb.name for lb in balancers], [name])
+        self.assertEqual([l.get_tuple() for l in balancers[0].listeners],
+                         listeners[:1])
+
+    def test_create_load_balancer_listeners_with_policies(self):
+        c = ELBConnection()
+        name = 'elb-boto-unit-test-policy'
+        availability_zones = ['us-east-1a']
+        listeners = [(80, 8000, 'HTTP')]
+        balancer = c.create_load_balancer(name, availability_zones, listeners)
+
+        more_listeners = [(443, 8001, 'HTTP')]
+        c.create_load_balancer_listeners(name, more_listeners)
+
+        lb_policy_name = 'lb-policy'
+        c.create_lb_cookie_stickiness_policy(1000, name, lb_policy_name)
+        c.set_lb_policies_of_listener(name, listeners[0][0], lb_policy_name)
+
+        app_policy_name = 'app-policy'
+        c.create_app_cookie_stickiness_policy('appcookie', name, app_policy_name)
+        c.set_lb_policies_of_listener(name, more_listeners[0][0], app_policy_name)
+
+        balancers = c.get_all_load_balancers()
+        self.assertEqual([lb.name for lb in balancers], [name])
+        self.assertEqual(
+            sorted(l.get_tuple() for l in balancers[0].listeners),
+            sorted(listeners + more_listeners)
+            )
+        # Policy names should be checked here once they are supported
+        # in the Listener object.
+
+if __name__ == '__main__':
+    unittest.main()

data/packages/gsutil/boto/tests/integration/ec2/test_cert_verification.py

@@ -0,0 +1,40 @@
+# Copyright (c) 2012 Mitch Garnaat http://garnaat.org/
+# 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.
+
+"""
+Check that all of the certs on all service endpoints validate.
+"""
+
+import unittest
+import boto.ec2
+
+
+class CertVerificationTest(unittest.TestCase):
+
+    ec2 = True
+    ssl = True
+
+    def test_certs(self):
+        for region in boto.ec2.regions():
+            c = region.connect()
+            c.get_all_instances()

data/packages/gsutil/boto/tests/integration/ec2/test_connection.py

@@ -0,0 +1,192 @@
+# Copyright (c) 2006-2010 Mitch Garnaat http://garnaat.org/
+# Copyright (c) 2009, Eucalyptus Systems, Inc.
+# 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.
+
+"""
+Some unit tests for the EC2Connection
+"""
+
+import unittest
+import time
+import telnetlib
+import socket
+
+from nose.plugins.attrib import attr
+from boto.ec2.connection import EC2Connection
+
+
+class EC2ConnectionTest (unittest.TestCase):
+    ec2 = True
+
+    @attr('notdefault')
+    def test_launch_permissions(self):
+        # this is my user_id, if you want to run these tests you should
+        # replace this with yours or they won't work
+        user_id = '963068290131'
+        print '--- running EC2Connection tests ---'
+        c = EC2Connection()
+        # get list of private AMI's
+        rs = c.get_all_images(owners=[user_id])
+        assert len(rs) > 0
+        # now pick the first one
+        image = rs[0]
+        # temporarily make this image runnable by everyone
+        status = image.set_launch_permissions(group_names=['all'])
+        assert status
+        d = image.get_launch_permissions()
+        assert 'groups' in d
+        assert len(d['groups']) > 0
+        # now remove that permission
+        status = image.remove_launch_permissions(group_names=['all'])
+        assert status
+        time.sleep(10)
+        d = image.get_launch_permissions()
+        assert 'groups' not in d
+
+    def test_1_basic(self):
+        # create 2 new security groups
+        c = EC2Connection()
+        group1_name = 'test-%d' % int(time.time())
+        group_desc = 'This is a security group created during unit testing'
+        group1 = c.create_security_group(group1_name, group_desc)
+        time.sleep(2)
+        group2_name = 'test-%d' % int(time.time())
+        group_desc = 'This is a security group created during unit testing'
+        group2 = c.create_security_group(group2_name, group_desc)
+        # now get a listing of all security groups and look for our new one
+        rs = c.get_all_security_groups()
+        found = False
+        for g in rs:
+            if g.name == group1_name:
+                found = True
+        assert found
+        # now pass arg to filter results to only our new group
+        rs = c.get_all_security_groups([group1_name])
+        assert len(rs) == 1
+        # try some group to group authorizations/revocations
+        # first try the old style
+        status = c.authorize_security_group(group1.name,
+                                            group2.name,
+                                            group2.owner_id)
+        assert status
+        status = c.revoke_security_group(group1.name,
+                                         group2.name,
+                                         group2.owner_id)
+        assert status
+        # now try specifying a specific port
+        status = c.authorize_security_group(group1.name,
+                                            group2.name,
+                                            group2.owner_id,
+                                            'tcp', 22, 22)
+        assert status
+        status = c.revoke_security_group(group1.name,
+                                         group2.name,
+                                         group2.owner_id,
+                                         'tcp', 22, 22)
+        assert status
+
+        # now delete the second security group
+        status = c.delete_security_group(group2_name)
+        # now make sure it's really gone
+        rs = c.get_all_security_groups()
+        found = False
+        for g in rs:
+            if g.name == group2_name:
+                found = True
+        assert not found
+
+        group = group1
+
+        # now try to launch apache image with our new security group
+        rs = c.get_all_images()
+        img_loc = 'ec2-public-images/fedora-core4-apache.manifest.xml'
+        for image in rs:
+            if image.location == img_loc:
+                break
+        reservation = image.run(security_groups=[group.name])
+        instance = reservation.instances[0]
+        while instance.state != 'running':
+            print '\tinstance is %s' % instance.state
+            time.sleep(30)
+            instance.update()
+        # instance in now running, try to telnet to port 80
+        t = telnetlib.Telnet()
+        try:
+            t.open(instance.dns_name, 80)
+        except socket.error:
+            pass
+        # now open up port 80 and try again, it should work
+        group.authorize('tcp', 80, 80, '0.0.0.0/0')
+        t.open(instance.dns_name, 80)
+        t.close()
+        # now revoke authorization and try again
+        group.revoke('tcp', 80, 80, '0.0.0.0/0')
+        try:
+            t.open(instance.dns_name, 80)
+        except socket.error:
+            pass
+        # now kill the instance and delete the security group
+        instance.terminate()
+
+        # check that state and previous_state have updated
+        assert instance.state == 'shutting-down'
+        assert instance.state_code == 32
+        assert instance.previous_state == 'running'
+        assert instance.previous_state_code == 16
+
+        # unfortunately, I can't delete the sg within this script
+        #sg.delete()
+
+        # create a new key pair
+        key_name = 'test-%d' % int(time.time())
+        status = c.create_key_pair(key_name)
+        assert status
+        # now get a listing of all key pairs and look for our new one
+        rs = c.get_all_key_pairs()
+        found = False
+        for k in rs:
+            if k.name == key_name:
+                found = True
+        assert found
+        # now pass arg to filter results to only our new key pair
+        rs = c.get_all_key_pairs([key_name])
+        assert len(rs) == 1
+        key_pair = rs[0]
+        # now delete the key pair
+        status = c.delete_key_pair(key_name)
+        # now make sure it's really gone
+        rs = c.get_all_key_pairs()
+        found = False
+        for k in rs:
+            if k.name == key_name:
+                found = True
+        assert not found
+
+        # short test around Paid AMI capability
+        demo_paid_ami_id = 'ami-bd9d78d4'
+        demo_paid_ami_product_code = 'A79EC0DB'
+        l = c.get_all_images([demo_paid_ami_id])
+        assert len(l) == 1
+        assert len(l[0].product_codes) == 1
+        assert l[0].product_codes[0] == demo_paid_ami_product_code
+
+        print '--- tests completed ---'

data/packages/gsutil/boto/tests/integration/ec2/vpc/test_connection.py

@@ -0,0 +1,95 @@
+#!/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 unittest
+import time
+
+import boto
+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):
+        self.api = boto.connect_vpc()
+        vpc = self.api.create_vpc('10.0.0.0/16')
+        self.addCleanup(self.api.delete_vpc, vpc.id)
+
+        self.subnet = self.api.create_subnet(vpc.id, '10.0.0.0/24')
+        self.addCleanup(self.api.delete_subnet, self.subnet.id)
+
+    def terminate_instance(self, instance):
+        instance.terminate()
+        for i in xrange(300):
+            instance.update()
+            if instance.state == 'terminated':
+                # Give it a litle more time to settle.
+                time.sleep(10)
+                return
+            else:
+                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_instances(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')
+
+
+if __name__ == '__main__':
+    unittest.main()