| # 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): |
| |
| 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( |
| [l.get_tuple() for l in balancers[0].listeners], 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] |
| ) |
| |
| |
| if __name__ == '__main__': |
| unittest.main() |