[catapult] - Copy Telemetry's gsutilz over to third_party.

third_party/gsutil/third_party/boto/tests/integration/ec2/elb/test_connection.py

Index: third_party/gsutil/third_party/boto/tests/integration/ec2/elb/test_connection.py
diff --git a/third_party/gsutil/third_party/boto/tests/integration/ec2/elb/test_connection.py b/third_party/gsutil/third_party/boto/tests/integration/ec2/elb/test_connection.py
new file mode 100644
index 0000000000000000000000000000000000000000..7f1ca80bb84acaa111d826158ab641cd5f4fbc2d
--- /dev/null
+++ b/third_party/gsutil/third_party/boto/tests/integration/ec2/elb/test_connection.py
@@ -0,0 +1,297 @@
+# 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 boto
+import time
+from tests.compat import unittest
+from boto.ec2.elb import ELBConnection
+import boto.ec2.elb
+
+
+class ELBConnectionTest(unittest.TestCase):
+    ec2 = True
+
+    def setUp(self):
+        """Creates a named load balancer that can be safely
+        deleted at the end of each test"""
+        self.conn = ELBConnection()
+        self.name = 'elb-boto-unit-test'
+        self.availability_zones = ['us-east-1a']
+        self.listeners = [(80, 8000, 'HTTP')]
+        self.balancer = self.conn.create_load_balancer(
+            self.name, self.availability_zones, self.listeners)
+
+        # S3 bucket for log tests
+        self.s3 = boto.connect_s3()
+        self.timestamp = str(int(time.time()))
+        self.bucket_name = 'boto-elb-%s' % self.timestamp
+        self.bucket = self.s3.create_bucket(self.bucket_name)
+        self.bucket.set_canned_acl('public-read-write')
+        self.addCleanup(self.cleanup_bucket, self.bucket)
+
+    def cleanup_bucket(self, bucket):
+        for key in bucket.get_all_keys():
+            key.delete()
+        bucket.delete()
+
+    def tearDown(self):
+        """ Deletes the test load balancer after every test.
+        It does not delete EVERY load balancer in your account"""
+        self.balancer.delete()
+
+    def test_build_list_params(self):
+        params = {}
+        self.conn.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):
+        self.assertEqual(self.balancer.name, self.name)
+        self.assertEqual(self.balancer.availability_zones,
+                         self.availability_zones)
+        self.assertEqual(self.balancer.listeners, self.listeners)
+
+        balancers = self.conn.get_all_load_balancers()
+        self.assertEqual([lb.name for lb in balancers], [self.name])
+
+    def test_create_load_balancer_listeners(self):
+        more_listeners = [(443, 8001, 'HTTP')]
+        self.conn.create_load_balancer_listeners(self.name, more_listeners)
+        balancers = self.conn.get_all_load_balancers()
+        self.assertEqual([lb.name for lb in balancers], [self.name])
+        self.assertEqual(
+            sorted(l.get_tuple() for l in balancers[0].listeners),
+            sorted(self.listeners + more_listeners)
+        )
+
+    def test_delete_load_balancer_listeners(self):
+        mod_listeners = [(80, 8000, 'HTTP'), (443, 8001, 'HTTP')]
+        mod_name = self.name + "-mod"
+        self.mod_balancer = self.conn.create_load_balancer(
+            mod_name, self.availability_zones, mod_listeners)
+
+        mod_balancers = self.conn.get_all_load_balancers(
+            load_balancer_names=[mod_name])
+        self.assertEqual([lb.name for lb in mod_balancers], [mod_name])
+        self.assertEqual(
+            sorted([l.get_tuple() for l in mod_balancers[0].listeners]),
+            sorted(mod_listeners))
+
+        self.conn.delete_load_balancer_listeners(self.mod_balancer.name, [443])
+        mod_balancers = self.conn.get_all_load_balancers(
+            load_balancer_names=[mod_name])
+        self.assertEqual([lb.name for lb in mod_balancers], [mod_name])
+        self.assertEqual([l.get_tuple() for l in mod_balancers[0].listeners],
+                         mod_listeners[:1])
+        self.mod_balancer.delete()
+
+    def test_create_load_balancer_listeners_with_policies(self):
+        more_listeners = [(443, 8001, 'HTTP')]
+        self.conn.create_load_balancer_listeners(self.name, more_listeners)
+
+        lb_policy_name = 'lb-policy'
+        self.conn.create_lb_cookie_stickiness_policy(
+            1000, self.name, lb_policy_name)
+        self.conn.set_lb_policies_of_listener(
+            self.name, self.listeners[0][0], lb_policy_name)
+
+        app_policy_name = 'app-policy'
+        self.conn.create_app_cookie_stickiness_policy(
+            'appcookie', self.name, app_policy_name)
+        self.conn.set_lb_policies_of_listener(
+            self.name, more_listeners[0][0], app_policy_name)
+
+        balancers = self.conn.get_all_load_balancers(
+            load_balancer_names=[self.name])
+        self.assertEqual([lb.name for lb in balancers], [self.name])
+        self.assertEqual(
+            sorted(l.get_tuple() for l in balancers[0].listeners),
+            sorted(self.listeners + more_listeners)
+        )
+        # Policy names should be checked here once they are supported
+        # in the Listener object.
+
+    def test_create_load_balancer_backend_with_policies(self):
+        other_policy_name = 'enable-proxy-protocol'
+        backend_port = 8081
+        self.conn.create_lb_policy(
+            self.name, other_policy_name,
+            'ProxyProtocolPolicyType', {'ProxyProtocol': True})
+        self.conn.set_lb_policies_of_backend_server(
+            self.name, backend_port, [other_policy_name])
+
+        balancers = self.conn.get_all_load_balancers(
+            load_balancer_names=[self.name])
+        self.assertEqual([lb.name for lb in balancers], [self.name])
+        self.assertEqual(len(balancers[0].policies.other_policies), 1)
+        self.assertEqual(balancers[0].policies.other_policies[0].policy_name,
+                         other_policy_name)
+        self.assertEqual(len(balancers[0].backends), 1)
+        self.assertEqual(balancers[0].backends[0].instance_port, backend_port)
+        self.assertEqual(balancers[0].backends[0].policies[0].policy_name,
+                         other_policy_name)
+
+        self.conn.set_lb_policies_of_backend_server(self.name, backend_port,
+                                                    [])
+
+        balancers = self.conn.get_all_load_balancers(
+            load_balancer_names=[self.name])
+        self.assertEqual([lb.name for lb in balancers], [self.name])
+        self.assertEqual(len(balancers[0].policies.other_policies), 1)
+        self.assertEqual(len(balancers[0].backends), 0)
+
+    def test_create_load_balancer_complex_listeners(self):
+        complex_listeners = [
+            (8080, 80, 'HTTP', 'HTTP'),
+            (2525, 25, 'TCP', 'TCP'),
+        ]
+
+        self.conn.create_load_balancer_listeners(
+            self.name,
+            complex_listeners=complex_listeners
+        )
+
+        balancers = self.conn.get_all_load_balancers(
+            load_balancer_names=[self.name]
+        )
+        self.assertEqual([lb.name for lb in balancers], [self.name])
+        self.assertEqual(
+            sorted(l.get_complex_tuple() for l in balancers[0].listeners),
+            # We need an extra 'HTTP' here over what ``self.listeners`` uses.
+            sorted([(80, 8000, 'HTTP', 'HTTP')] + complex_listeners)
+        )
+
+    def test_load_balancer_access_log(self):
+        attributes = self.balancer.get_attributes()
+
+        self.assertEqual(False, attributes.access_log.enabled)
+
+        attributes.access_log.enabled = True
+        attributes.access_log.s3_bucket_name = self.bucket_name
+        attributes.access_log.s3_bucket_prefix = 'access-logs'
+        attributes.access_log.emit_interval = 5
+
+        self.conn.modify_lb_attribute(self.balancer.name, 'accessLog',
+                                      attributes.access_log)
+
+        new_attributes = self.balancer.get_attributes()
+
+        self.assertEqual(True, new_attributes.access_log.enabled)
+        self.assertEqual(self.bucket_name,
+                         new_attributes.access_log.s3_bucket_name)
+        self.assertEqual('access-logs',
+                         new_attributes.access_log.s3_bucket_prefix)
+        self.assertEqual(5, new_attributes.access_log.emit_interval)
+
+    def test_load_balancer_get_attributes(self):
+        attributes = self.balancer.get_attributes()
+        connection_draining = self.conn.get_lb_attribute(self.balancer.name,
+                                                         'ConnectionDraining')
+        self.assertEqual(connection_draining.enabled,
+                         attributes.connection_draining.enabled)
+        self.assertEqual(connection_draining.timeout,
+                         attributes.connection_draining.timeout)
+
+        access_log = self.conn.get_lb_attribute(self.balancer.name,
+                                                'AccessLog')
+        self.assertEqual(access_log.enabled, attributes.access_log.enabled)
+        self.assertEqual(access_log.s3_bucket_name,
+                         attributes.access_log.s3_bucket_name)
+        self.assertEqual(access_log.s3_bucket_prefix,
+                         attributes.access_log.s3_bucket_prefix)
+        self.assertEqual(access_log.emit_interval,
+                         attributes.access_log.emit_interval)
+
+        cross_zone_load_balancing = self.conn.get_lb_attribute(
+            self.balancer.name, 'CrossZoneLoadBalancing')
+        self.assertEqual(cross_zone_load_balancing,
+                         attributes.cross_zone_load_balancing.enabled)
+
+    def change_and_verify_load_balancer_connection_draining(
+            self, enabled, timeout=None):
+        attributes = self.balancer.get_attributes()
+
+        attributes.connection_draining.enabled = enabled
+        if timeout is not None:
+            attributes.connection_draining.timeout = timeout
+
+        self.conn.modify_lb_attribute(
+            self.balancer.name, 'ConnectionDraining',
+            attributes.connection_draining)
+
+        attributes = self.balancer.get_attributes()
+        self.assertEqual(enabled, attributes.connection_draining.enabled)
+        if timeout is not None:
+            self.assertEqual(timeout, attributes.connection_draining.timeout)
+
+    def test_load_balancer_connection_draining_config(self):
+        self.change_and_verify_load_balancer_connection_draining(True, 128)
+        self.change_and_verify_load_balancer_connection_draining(True, 256)
+        self.change_and_verify_load_balancer_connection_draining(False)
+        self.change_and_verify_load_balancer_connection_draining(True, 64)
+
+    def test_set_load_balancer_policies_of_listeners(self):
+        more_listeners = [(443, 8001, 'HTTP')]
+        self.conn.create_load_balancer_listeners(self.name, more_listeners)
+
+        lb_policy_name = 'lb-policy'
+        self.conn.create_lb_cookie_stickiness_policy(
+            1000,
+            self.name,
+            lb_policy_name
+        )
+        self.conn.set_lb_policies_of_listener(
+            self.name,
+            self.listeners[0][0],
+            lb_policy_name
+        )
+
+        # Try to remove the policy by passing empty list.
+        # http://docs.aws.amazon.com/ElasticLoadBalancing/latest/APIReference/API_SetLoadBalancerPoliciesOfListener.html
+        # documents this as the way to remove policies.
+        self.conn.set_lb_policies_of_listener(
+            self.name,
+            self.listeners[0][0],
+            []
+        )
+
+    def test_can_make_sigv4_call(self):
+        connection = boto.ec2.elb.connect_to_region('eu-central-1')
+        lbs = connection.get_all_load_balancers()
+        self.assertTrue(isinstance(lbs, list))
+
+
+if __name__ == '__main__':
+    unittest.main()