| Index: third_party/gsutil/boto/tests/unit/ec2/test_networkinterface.py
|
| diff --git a/third_party/gsutil/boto/tests/unit/ec2/test_networkinterface.py b/third_party/gsutil/boto/tests/unit/ec2/test_networkinterface.py
|
| new file mode 100644
|
| index 0000000000000000000000000000000000000000..b23f6c3601b53e7badfb3a5edd29e06822770b5b
|
| --- /dev/null
|
| +++ b/third_party/gsutil/boto/tests/unit/ec2/test_networkinterface.py
|
| @@ -0,0 +1,140 @@
|
| +#!/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.
|
| +#
|
| +
|
| +from tests.unit import unittest
|
| +
|
| +
|
| +from boto.ec2.networkinterface import NetworkInterfaceCollection
|
| +from boto.ec2.networkinterface import NetworkInterfaceSpecification
|
| +from boto.ec2.networkinterface import PrivateIPAddress
|
| +
|
| +
|
| +class TestNetworkInterfaceCollection(unittest.TestCase):
|
| + maxDiff = None
|
| +
|
| + def setUp(self):
|
| + self.private_ip_address1 = PrivateIPAddress(
|
| + private_ip_address='10.0.0.10', primary=False)
|
| + self.private_ip_address2 = PrivateIPAddress(
|
| + private_ip_address='10.0.0.11', primary=False)
|
| + self.network_interfaces_spec1 = NetworkInterfaceSpecification(
|
| + device_index=1, subnet_id='subnet_id',
|
| + description='description1',
|
| + private_ip_address='10.0.0.54', delete_on_termination=False,
|
| + private_ip_addresses=[self.private_ip_address1,
|
| + self.private_ip_address2])
|
| +
|
| + self.private_ip_address3 = PrivateIPAddress(
|
| + private_ip_address='10.0.1.10', primary=False)
|
| + self.private_ip_address4 = PrivateIPAddress(
|
| + private_ip_address='10.0.1.11', primary=False)
|
| + self.network_interfaces_spec2 = NetworkInterfaceSpecification(
|
| + device_index=2, subnet_id='subnet_id2',
|
| + description='description2',
|
| + groups=['group_id1', 'group_id2'],
|
| + private_ip_address='10.0.1.54', delete_on_termination=False,
|
| + private_ip_addresses=[self.private_ip_address3,
|
| + self.private_ip_address4])
|
| +
|
| + def test_param_serialization(self):
|
| + collection = NetworkInterfaceCollection(self.network_interfaces_spec1,
|
| + self.network_interfaces_spec2)
|
| + params = {}
|
| + collection.build_list_params(params)
|
| + self.assertDictEqual(params, {
|
| + 'NetworkInterface.1.DeviceIndex': '1',
|
| + 'NetworkInterface.1.DeleteOnTermination': 'false',
|
| + 'NetworkInterface.1.Description': 'description1',
|
| + 'NetworkInterface.1.PrivateIpAddress': '10.0.0.54',
|
| + 'NetworkInterface.1.SubnetId': 'subnet_id',
|
| + 'NetworkInterface.1.PrivateIpAddresses.1.Primary': 'false',
|
| + 'NetworkInterface.1.PrivateIpAddresses.1.PrivateIpAddress':
|
| + '10.0.0.10',
|
| + 'NetworkInterface.1.PrivateIpAddresses.2.Primary': 'false',
|
| + 'NetworkInterface.1.PrivateIpAddresses.2.PrivateIpAddress':
|
| + '10.0.0.11',
|
| + 'NetworkInterface.2.DeviceIndex': '2',
|
| + 'NetworkInterface.2.Description': 'description2',
|
| + 'NetworkInterface.2.DeleteOnTermination': 'false',
|
| + 'NetworkInterface.2.PrivateIpAddress': '10.0.1.54',
|
| + 'NetworkInterface.2.SubnetId': 'subnet_id2',
|
| + 'NetworkInterface.2.SecurityGroupId.1': 'group_id1',
|
| + 'NetworkInterface.2.SecurityGroupId.2': 'group_id2',
|
| + 'NetworkInterface.2.PrivateIpAddresses.1.Primary': 'false',
|
| + 'NetworkInterface.2.PrivateIpAddresses.1.PrivateIpAddress':
|
| + '10.0.1.10',
|
| + 'NetworkInterface.2.PrivateIpAddresses.2.Primary': 'false',
|
| + 'NetworkInterface.2.PrivateIpAddresses.2.PrivateIpAddress':
|
| + '10.0.1.11',
|
| + })
|
| +
|
| + def test_add_prefix_to_serialization(self):
|
| + return
|
| + collection = NetworkInterfaceCollection(self.network_interfaces_spec1,
|
| + self.network_interfaces_spec2)
|
| + params = {}
|
| + collection.build_list_params(params, prefix='LaunchSpecification.')
|
| + # We already tested the actual serialization previously, so
|
| + # we're just checking a few keys to make sure we get the proper
|
| + # prefix.
|
| + self.assertDictEqual(params, {
|
| + 'LaunchSpecification.NetworkInterface.1.DeviceIndex': '1',
|
| + 'LaunchSpecification.NetworkInterface.1.DeleteOnTermination':
|
| + 'false',
|
| + 'LaunchSpecification.NetworkInterface.1.Description':
|
| + 'description1',
|
| + 'LaunchSpecification.NetworkInterface.1.PrivateIpAddress':
|
| + '10.0.0.54',
|
| + 'LaunchSpecification.NetworkInterface.1.SubnetId': 'subnet_id',
|
| + 'LaunchSpecification.NetworkInterface.1.PrivateIpAddresses.1.Primary':
|
| + 'false',
|
| + 'LaunchSpecification.NetworkInterface.1.PrivateIpAddresses.1.PrivateIpAddress':
|
| + '10.0.0.10',
|
| + 'LaunchSpecification.NetworkInterface.1.PrivateIpAddresses.2.Primary': 'false',
|
| + 'LaunchSpecification.NetworkInterface.1.PrivateIpAddresses.2.PrivateIpAddress':
|
| + '10.0.0.11',
|
| + 'LaunchSpecification.NetworkInterface.2.DeviceIndex': '2',
|
| + 'LaunchSpecification.NetworkInterface.2.Description':
|
| + 'description2',
|
| + 'LaunchSpecification.NetworkInterface.2.DeleteOnTermination':
|
| + 'false',
|
| + 'LaunchSpecification.NetworkInterface.2.PrivateIpAddress':
|
| + '10.0.1.54',
|
| + 'LaunchSpecification.NetworkInterface.2.SubnetId': 'subnet_id2',
|
| + 'LaunchSpecification.NetworkInterface.2.SecurityGroupId.1':
|
| + 'group_id1',
|
| + 'LaunchSpecification.NetworkInterface.2.SecurityGroupId.2':
|
| + 'group_id2',
|
| + 'LaunchSpecification.NetworkInterface.2.PrivateIpAddresses.1.Primary':
|
| + 'false',
|
| + 'LaunchSpecification.NetworkInterface.2.PrivateIpAddresses.1.PrivateIpAddress':
|
| + '10.0.1.10',
|
| + 'LaunchSpecification.NetworkInterface.2.PrivateIpAddresses.2.Primary':
|
| + 'false',
|
| + 'LaunchSpecification.NetworkInterface.2.PrivateIpAddresses.2.PrivateIpAddress':
|
| + '10.0.1.11',
|
| + })
|
| +
|
| +
|
| +if __name__ == '__main__':
|
| + unittest.main()
|
|
|