Some unit tests

1 files changed, 6 insertions, 3 deletions

diff --git a/nova/compute/api.py b/nova/compute/api.py
index 1393c01d5..0dc2bb3d3 100644
--- a/nova/compute/api.py
+++ b/nova/compute/api.py
@@ -460,10 +460,13 @@ class API(base.Base):
     def resize(self, context, instance_id, flavor_id):
         """Resize a running instance."""
         instance = self.db.instance_get(context, instance_id)
-        current_instance_type = self.db.instance_type_get_by_flavor_id(
-            context, instance['flavor_id'])
+        current_instance_type = self.db.instance_type_get_by_name(
+            context, instance['instance_type'])
+        
         new_instance_type = self.db.instance_type_get_by_flavor_id(
-            context, flavor_id)
+                context, flavor_id)
+        if not new_instance_type:
+            raise exception.ApiError(_("Requested flavor does not exist"))
 
         if current_instance_type.memory_mb > new_instance_typ.memory_mb:
             raise exception.ApiError(_("Invalid flavor: cannot downsize"