From 534039f46517a6c2b74ec30531cef0b3f725da82 Mon Sep 17 00:00:00 2001 From: Shawn Falkner-Horine Date: Thu, 4 Sep 2014 17:07:00 -0700 Subject: [PATCH] Route Tables / Routes: Initial implementation. (Fixed modification of dict while looping said dict.) --- moto/ec2/models.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/moto/ec2/models.py b/moto/ec2/models.py index 17222f472..fd15a8ee2 100644 --- a/moto/ec2/models.py +++ b/moto/ec2/models.py @@ -1148,10 +1148,15 @@ class RouteTableBackend(object): raise InvalidRouteTableIdError(route_table_id) return deleted - def delete_route_table_for_vpc(self, vpc_id): + def get_route_table_for_vpc(self, vpc_id): for route_table in self.route_tables.values(): if route_table.vpc_id == vpc_id: - self.delete_route_table(route_table.id) + return route_table + + def delete_route_table_for_vpc(self, vpc_id): + route_table = self.get_route_table_for_vpc(vpc_id) + if route_table: + self.delete_route_table(route_table.id) class Route(object):