Index: tools/telemetry/third_party/gsutil/third_party/boto/tests/unit/ec2/elb/test_attribute.py |
diff --git a/tools/telemetry/third_party/gsutil/third_party/boto/tests/unit/ec2/elb/test_attribute.py b/tools/telemetry/third_party/gsutil/third_party/boto/tests/unit/ec2/elb/test_attribute.py |
deleted file mode 100644 |
index 40edecba8279ecdd9e1c7f832a9fdf6fa90cedc6..0000000000000000000000000000000000000000 |
--- a/tools/telemetry/third_party/gsutil/third_party/boto/tests/unit/ec2/elb/test_attribute.py |
+++ /dev/null |
@@ -1,205 +0,0 @@ |
-from tests.unit import unittest |
-from tests.compat import mock |
- |
-from boto.ec2.elb import ELBConnection |
-from boto.ec2.elb import LoadBalancer |
-from boto.ec2.elb.attributes import LbAttributes |
- |
-ATTRIBUTE_GET_TRUE_CZL_RESPONSE = b"""<?xml version="1.0" encoding="UTF-8"?> |
-<DescribeLoadBalancerAttributesResponse xmlns="http://elasticloadbalancing.amazonaws.com/doc/2012-06-01/"> |
- <DescribeLoadBalancerAttributesResult> |
- <LoadBalancerAttributes> |
- <CrossZoneLoadBalancing> |
- <Enabled>true</Enabled> |
- </CrossZoneLoadBalancing> |
- </LoadBalancerAttributes> |
- </DescribeLoadBalancerAttributesResult> |
-<ResponseMetadata> |
- <RequestId>83c88b9d-12b7-11e3-8b82-87b12EXAMPLE</RequestId> |
-</ResponseMetadata> |
-</DescribeLoadBalancerAttributesResponse> |
-""" |
- |
-ATTRIBUTE_GET_FALSE_CZL_RESPONSE = b"""<?xml version="1.0" encoding="UTF-8"?> |
-<DescribeLoadBalancerAttributesResponse xmlns="http://elasticloadbalancing.amazonaws.com/doc/2012-06-01/"> |
- <DescribeLoadBalancerAttributesResult> |
- <LoadBalancerAttributes> |
- <CrossZoneLoadBalancing> |
- <Enabled>false</Enabled> |
- </CrossZoneLoadBalancing> |
- </LoadBalancerAttributes> |
- </DescribeLoadBalancerAttributesResult> |
-<ResponseMetadata> |
- <RequestId>83c88b9d-12b7-11e3-8b82-87b12EXAMPLE</RequestId> |
-</ResponseMetadata> |
-</DescribeLoadBalancerAttributesResponse> |
-""" |
- |
-ATTRIBUTE_GET_CS_RESPONSE = b"""<?xml version="1.0" encoding="UTF-8"?> |
-<DescribeLoadBalancerAttributesResponse xmlns="http://elasticloadbalancing.amazonaws.com/doc/2012-06-01/"> |
- <DescribeLoadBalancerAttributesResult> |
- <LoadBalancerAttributes> |
- <ConnectionSettings> |
- <IdleTimeout>30</IdleTimeout> |
- </ConnectionSettings> |
- </LoadBalancerAttributes> |
- </DescribeLoadBalancerAttributesResult> |
-<ResponseMetadata> |
- <RequestId>83c88b9d-12b7-11e3-8b82-87b12EXAMPLE</RequestId> |
-</ResponseMetadata> |
-</DescribeLoadBalancerAttributesResponse> |
-""" |
- |
-ATTRIBUTE_SET_RESPONSE = b"""<?xml version="1.0" encoding="UTF-8"?> |
-<ModifyLoadBalancerAttributesResponse xmlns="http://elasticloadbalancing.amazonaws.com/doc/2012-06-01/"> |
-<ModifyLoadBalancerAttributesResult/> |
-<ResponseMetadata> |
- <RequestId>83c88b9d-12b7-11e3-8b82-87b12EXAMPLE</RequestId> |
-</ResponseMetadata> |
-</ModifyLoadBalancerAttributesResponse> |
-""" |
- |
-# make_request arguments for setting attributes. |
-# Format: (API_COMMAND, API_PARAMS, API_PATH, API_METHOD) |
-ATTRIBUTE_SET_CZL_TRUE_REQUEST = ( |
- 'ModifyLoadBalancerAttributes', |
- {'LoadBalancerAttributes.CrossZoneLoadBalancing.Enabled': 'true', |
- 'LoadBalancerName': 'test_elb'}, mock.ANY, mock.ANY) |
-ATTRIBUTE_SET_CZL_FALSE_REQUEST = ( |
- 'ModifyLoadBalancerAttributes', |
- {'LoadBalancerAttributes.CrossZoneLoadBalancing.Enabled': 'false', |
- 'LoadBalancerName': 'test_elb'}, mock.ANY, mock.ANY) |
- |
-# Tests to be run on an LbAttributes |
-# Format: |
-# (EC2_RESPONSE_STRING, list( (string_of_attribute_to_test, value) ) ) |
-ATTRIBUTE_TESTS = [ |
- (ATTRIBUTE_GET_TRUE_CZL_RESPONSE, |
- [('cross_zone_load_balancing.enabled', True)]), |
- (ATTRIBUTE_GET_FALSE_CZL_RESPONSE, |
- [('cross_zone_load_balancing.enabled', False)]), |
- (ATTRIBUTE_GET_CS_RESPONSE, |
- [('connecting_settings.idle_timeout', 30)]), |
-] |
- |
- |
-class TestLbAttributes(unittest.TestCase): |
- """Tests LB Attributes.""" |
- def _setup_mock(self): |
- """Sets up a mock elb request. |
- Returns: response, elb connection and LoadBalancer |
- """ |
- mock_response = mock.Mock() |
- mock_response.status = 200 |
- elb = ELBConnection(aws_access_key_id='aws_access_key_id', |
- aws_secret_access_key='aws_secret_access_key') |
- elb.make_request = mock.Mock(return_value=mock_response) |
- return mock_response, elb, LoadBalancer(elb, 'test_elb') |
- |
- def _verify_attributes(self, attributes, attr_tests): |
- """Verifies an LbAttributes object.""" |
- for attr, result in attr_tests: |
- attr_result = attributes |
- for sub_attr in attr.split('.'): |
- attr_result = getattr(attr_result, sub_attr, None) |
- self.assertEqual(attr_result, result) |
- |
- def test_get_all_lb_attributes(self): |
- """Tests getting the LbAttributes from the elb.connection.""" |
- mock_response, elb, _ = self._setup_mock() |
- |
- for response, attr_tests in ATTRIBUTE_TESTS: |
- mock_response.read.return_value = response |
- attributes = elb.get_all_lb_attributes('test_elb') |
- self.assertTrue(isinstance(attributes, LbAttributes)) |
- self._verify_attributes(attributes, attr_tests) |
- |
- def test_get_lb_attribute(self): |
- """Tests getting a single attribute from elb.connection.""" |
- mock_response, elb, _ = self._setup_mock() |
- |
- tests = [ |
- ('crossZoneLoadBalancing', True, ATTRIBUTE_GET_TRUE_CZL_RESPONSE), |
- ('crossZoneLoadBalancing', False, ATTRIBUTE_GET_FALSE_CZL_RESPONSE), |
- ] |
- |
- for attr, value, response in tests: |
- mock_response.read.return_value = response |
- status = elb.get_lb_attribute('test_elb', attr) |
- self.assertEqual(status, value) |
- |
- def test_modify_lb_attribute(self): |
- """Tests setting the attributes from elb.connection.""" |
- mock_response, elb, _ = self._setup_mock() |
- |
- tests = [ |
- ('crossZoneLoadBalancing', True, ATTRIBUTE_SET_CZL_TRUE_REQUEST), |
- ('crossZoneLoadBalancing', False, ATTRIBUTE_SET_CZL_FALSE_REQUEST), |
- ] |
- |
- for attr, value, args in tests: |
- mock_response.read.return_value = ATTRIBUTE_SET_RESPONSE |
- result = elb.modify_lb_attribute('test_elb', attr, value) |
- self.assertTrue(result) |
- elb.make_request.assert_called_with(*args) |
- |
- def test_lb_get_attributes(self): |
- """Tests the LbAttributes from the ELB object.""" |
- mock_response, _, lb = self._setup_mock() |
- |
- for response, attr_tests in ATTRIBUTE_TESTS: |
- mock_response.read.return_value = response |
- attributes = lb.get_attributes(force=True) |
- self.assertTrue(isinstance(attributes, LbAttributes)) |
- self._verify_attributes(attributes, attr_tests) |
- |
- def test_lb_is_cross_zone_load_balancing(self): |
- """Tests checking is_cross_zone_load_balancing.""" |
- mock_response, _, lb = self._setup_mock() |
- |
- tests = [ |
- # Format: (method, args, result, response) |
- # Gets a true result. |
- (lb.is_cross_zone_load_balancing, [], True, |
- ATTRIBUTE_GET_TRUE_CZL_RESPONSE), |
- # Returns the previous calls cached value. |
- (lb.is_cross_zone_load_balancing, [], True, |
- ATTRIBUTE_GET_FALSE_CZL_RESPONSE), |
- # Gets a false result. |
- (lb.is_cross_zone_load_balancing, [True], False, |
- ATTRIBUTE_GET_FALSE_CZL_RESPONSE), |
- ] |
- |
- for method, args, result, response in tests: |
- mock_response.read.return_value = response |
- self.assertEqual(method(*args), result) |
- |
- def test_lb_enable_cross_zone_load_balancing(self): |
- """Tests enabling cross zone balancing from LoadBalancer.""" |
- mock_response, elb, lb = self._setup_mock() |
- |
- mock_response.read.return_value = ATTRIBUTE_SET_RESPONSE |
- self.assertTrue(lb.enable_cross_zone_load_balancing()) |
- elb.make_request.assert_called_with(*ATTRIBUTE_SET_CZL_TRUE_REQUEST) |
- |
- def test_lb_disable_cross_zone_load_balancing(self): |
- """Tests disabling cross zone balancing from LoadBalancer.""" |
- mock_response, elb, lb = self._setup_mock() |
- |
- mock_response.read.return_value = ATTRIBUTE_SET_RESPONSE |
- self.assertTrue(lb.disable_cross_zone_load_balancing()) |
- elb.make_request.assert_called_with(*ATTRIBUTE_SET_CZL_FALSE_REQUEST) |
- |
- def test_lb_get_connection_settings(self): |
- """Tests checking connectionSettings attribute""" |
- mock_response, elb, _ = self._setup_mock() |
- |
- attrs = [('idle_timeout', 30), ] |
- mock_response.read.return_value = ATTRIBUTE_GET_CS_RESPONSE |
- attributes = elb.get_all_lb_attributes('test_elb') |
- self.assertTrue(isinstance(attributes, LbAttributes)) |
- for attr, value in attrs: |
- self.assertEqual(getattr(attributes.connecting_settings, attr), value) |
- |
-if __name__ == '__main__': |
- unittest.main() |