def test_connect(self): self.make_mock_vpc() # With a test VPC created, we now test our own functions con = vpc.connect_to_region("ap-southeast-2") d = vpc.get_vpc_overview(con, self.new_vpc.id, "ap-southeast-2") self.assertEqual( sorted([ 'subnets', 'route_tables', 'instance_by_id', 'instances', 'zones', 'vpc' ]), sorted(d.keys())) self.assertEqual(self.new_vpc.id, d['vpc'].id) self.assertTrue(self.new_subnet_a.id in [s.id for s in d['subnets']]) self.assertTrue(self.new_subnet_b.id in [s.id for s in d['subnets']]) self.assertTrue(len(d['zones']) == 3) self.assertTrue(len(d['route_tables']) == 1) self.assertTrue(len(d['instance_by_id'].keys()) == 2) self.assertTrue(d['instance_by_id'][self.i1.id].id == self.i1.id) self.assertTrue(d['instance_by_id'][self.i2.id].id == self.i2.id) self.assertTrue( vpc.find_instance_and_eni_by_ip(d, self.i1ip)[0].id == self.i1.id) self.assertTrue( vpc.find_instance_and_eni_by_ip(d, self.i2ip)[0].id == self.i2.id)
def _prepare_mock_env(self): self.make_mock_vpc() con = vpc.connect_to_region("ap-southeast-2") d = vpc.get_vpc_overview(con, self.new_vpc.id, "ap-southeast-2") i1, eni1 = vpc.find_instance_and_eni_by_ip(d, self.i1ip) i2, eni2 = vpc.find_instance_and_eni_by_ip(d, self.i2ip) rt_id = d['route_tables'][0].id return con, d, i1, eni1, i2, eni2, rt_id
def test_connect(self): self.make_mock_vpc() # With a test VPC created, we now test our own functions # In the mocked test the meta data won't contain the info we need (vpc # and region name), because the emulated EC2 instance isn't in any # region or vpc. meta = vpc.get_ec2_meta_data() self.assertTrue(meta == {}) self.assertRaises(VpcRouteSetError, vpc.connect_to_region, "blah") con = vpc.connect_to_region("ap-southeast-2") # Error when specifying non-existent VPC self.assertRaises(VpcRouteSetError, vpc.get_vpc_overview, con, "non-existent-vpc", "ap-southeast-2") # Get the default: First VPC if no VPC is specified d = vpc.get_vpc_overview(con, None, "ap-southeast-2") self.assertEqual(d['vpc'].id, "vpc-be745e76") # Get specified VPC d = vpc.get_vpc_overview(con, self.new_vpc.id, "ap-southeast-2") self.assertEqual(d['vpc'].id, "vpc-be745e76") self.assertEqual( sorted([ 'subnets', 'route_tables', 'instance_by_id', 'ip_subnet_lookup', 'instances', 'rt_subnet_lookup', 'zones', 'vpc' ]), sorted(d.keys())) self.assertEqual(self.new_vpc.id, d['vpc'].id) self.assertTrue(self.new_subnet_a.id in [s.id for s in d['subnets']]) self.assertTrue(self.new_subnet_b.id in [s.id for s in d['subnets']]) self.assertTrue(len(d['zones']) == 3) self.assertTrue(len(d['route_tables']) == 1) self.assertTrue(len(d['instance_by_id'].keys()) == 2) self.assertTrue(d['instance_by_id'][self.i1.id].id == self.i1.id) self.assertTrue(d['instance_by_id'][self.i2.id].id == self.i2.id) self.assertRaises(VpcRouteSetError, vpc.find_instance_and_eni_by_ip, d, "9.9.9.9") # Non existent IP self.assertTrue( vpc.find_instance_and_eni_by_ip(d, self.i1ip)[0].id == self.i1.id) self.assertTrue( vpc.find_instance_and_eni_by_ip(d, self.i2ip)[0].id == self.i2.id)
def test_handle_spec(self): self.make_mock_vpc() # Need to take a peek inside the VPC so we can properly evaluate the # output later on con = vpc.connect_to_region("ap-southeast-2") d = vpc.get_vpc_overview(con, self.new_vpc.id, "ap-southeast-2") route_spec = {u"10.2.0.0/16": [self.i1ip]} d['cluster_node_subnets'] = \ vpc.make_cluster_node_subnet_list(d, route_spec) i, eni = vpc.find_instance_and_eni_by_ip(d, self.i1ip) rt_id = d['route_tables'][0].id con.associate_route_table(route_table_id=rt_id, subnet_id=self.new_subnet_a.id) con.associate_route_table(route_table_id=rt_id, subnet_id=self.new_subnet_b.id) # Test handle_spec vid = self.new_vpc.id self.lc.clear() vpc.handle_spec("ap-southeast-2", vid, route_spec, [], []) self.lc.check( ('root', 'DEBUG', 'Handle route spec'), ('root', 'DEBUG', "Connecting to AWS region 'ap-southeast-2'"), ('root', 'DEBUG', "Retrieving information for VPC '%s'" % vid), ('root', 'DEBUG', 'Route spec processing. No failed IPs.'), ('root', 'INFO', "--- adding route in RT '%s' 10.2.0.0/16 -> %s (%s, %s)" % (rt_id, self.i1ip, self.i1.id, eni.id))) # mock the get_instance_private_ip_from_route() function in vpc. Reason # being: The boto mocking library (moto) doesn't handle ENIs in routes # correctly. Therefore, a match against the information we get from the # routes will never work. So, we provide a wrapper, which fills the # instance's ENI information into the route. This means that this # function now will always match. It's good for testing the 'match' # part of the code. old_func = vpc.get_instance_private_ip_from_route def my_get_instance_private_ip_from_route(instance, route): route.interface_id = instance.interfaces[0].id return old_func(instance, route) vpc.get_instance_private_ip_from_route = \ my_get_instance_private_ip_from_route self.lc.clear() vpc.handle_spec("ap-southeast-2", vid, route_spec, [], []) vpc.get_instance_private_ip_from_route = old_func self.lc.check( ('root', 'DEBUG', 'Handle route spec'), ('root', 'DEBUG', "Connecting to AWS region 'ap-southeast-2'"), ('root', 'DEBUG', "Retrieving information for VPC '%s'" % vid), ('root', 'DEBUG', 'Route spec processing. No failed IPs.'), ('root', 'INFO', "--- route exists already in RT '%s': 10.2.0.0/16 -> " "%s (%s, %s)" % (rt_id, self.i1ip, self.i1.id, eni.id)))
def test_process_route_spec_config(self): self.make_mock_vpc() con = vpc.connect_to_region("ap-southeast-2") d = vpc.get_vpc_overview(con, self.new_vpc.id, "ap-southeast-2") i1, eni1 = vpc.find_instance_and_eni_by_ip(d, self.i1ip) i2, eni2 = vpc.find_instance_and_eni_by_ip(d, self.i2ip) rt_id = d['route_tables'][0].id route_spec = {u"10.1.0.0/16": [self.i1ip, self.i2ip]} # Process a simple route spec, a route should have been added self.lc.clear() vpc.process_route_spec_config(con, d, route_spec, []) # One of the hosts is randomly chosen. We seeded the random number # generator at in this module, so we know that it will choose the # second host in this case. self.lc.check( ('root', 'DEBUG', 'Route spec processing. No failed IPs.'), ('root', 'INFO', "--- adding route in RT '%s' " "10.1.0.0/16 -> %s (%s, %s)" % (rt_id, self.i2ip, i2.id, eni2.id))) # One of the two IPs failed, switch over d = vpc.get_vpc_overview(con, self.new_vpc.id, "ap-southeast-2") self.lc.clear() vpc.process_route_spec_config(con, d, route_spec, [self.i1ip]) self.lc.check( ('root', 'DEBUG', 'Route spec processing. Failed IPs: %s' % self.i1ip), ('root', 'INFO', "--- updating existing route in RT '%s' 10.1.0.0/16 -> " "%s (%s, %s) (old IP: None, reason: old IP failed or not " "eligible anymore)" % (rt_id, self.i2ip, i2.id, eni2.id))) # Now all IPs for a route have failed d = vpc.get_vpc_overview(con, self.new_vpc.id, "ap-southeast-2") self.lc.clear() vpc.process_route_spec_config(con, d, route_spec, [self.i1ip, self.i2ip]) self.lc.check( ('root', 'DEBUG', 'Route spec processing. Failed IPs: %s,%s' % (self.i1ip, self.i2ip)), ('root', 'WARNING', '--- cannot find available target for route update 10.1.0.0/16! ' 'Nothing I can do...')) # Add new route, remove old one route_spec = {u"10.2.0.0/16": [self.i1ip]} d = vpc.get_vpc_overview(con, self.new_vpc.id, "ap-southeast-2") self.lc.clear() vpc.process_route_spec_config(con, d, route_spec, []) self.lc.check( ('root', 'DEBUG', 'Route spec processing. No failed IPs.'), ('root', 'INFO', "--- route not in spec, deleting in RT '%s': " "10.1.0.0/16 -> ... (%s, (unknown))" % (rt_id, i2.id)), ('root', 'INFO', "--- adding route in RT '%s' " "10.2.0.0/16 -> %s (%s, %s)" % (rt_id, self.i1ip, i1.id, eni1.id)))