Added tests for Flavor updating.

Author: mboersma

api/models.py

@@ -206,6 +206,11 @@ class Flavor(UuidAuditedModel):
         'us-west-2': 'ami-ea1e82da',
     }
 
+    def save(self, *args, **kwargs):
+        print args, kwargs
+        print self.params
+        return super(Flavor, self).save(*args, **kwargs)
+
     class Meta:
         unique_together = (('owner', 'id'),)
 

api/tests/flavor.py

@@ -69,3 +69,45 @@ def test_flavor_contents(self):
         self.assertEqual(params['zone'], 'any')
         self.assertEqual(params['size'], 'm1.medium')
         self.assertTrue(params['image'])
+
+    def test_flavor_update(self):
+        """Tests that flavors can be updated by the client."""
+        url = '/api/flavors'
+        params = {
+            'region': 'us-west-2',
+            'size': 't1.micro',
+        }
+        body = {'id': 'autotest', 'provider': 'autotest', 'params': json.dumps(params)}
+        response = self.client.post(url, json.dumps(body), content_type='application/json')
+        self.assertEqual(response.status_code, 201)
+        flavor_id = response.data['id']
+        response = self.client.get('/api/flavors')
+        self.assertEqual(response.status_code, 200)
+        self.assertEqual(len(response.data['results']), 1)
+        url = "/api/flavors/{flavor_id}".format(**locals())
+        response = self.client.get(url)
+        self.assertEqual(response.status_code, 200)
+        params = json.loads(response.data['params'])
+        self.assertEqual(params['region'], 'us-west-2')
+        self.assertEqual(params['zone'], 'any')
+        self.assertEqual(params['size'], 't1.micro')
+        self.assertTrue(params['image'])
+        params = {
+            'size': 'c1.xlarge',
+            'image': 'ami-c98d1bf9',
+        }
+        body = {'id': flavor_id, 'params': json.dumps(params)}
+        response = self.client.patch(url, json.dumps(body), content_type='application/json')
+        self.assertEqual(response.status_code, 200)
+        flavor_id = response.data['id']
+        response = self.client.get('/api/flavors')
+        self.assertEqual(response.status_code, 200)
+        self.assertEqual(len(response.data['results']), 1)
+        url = "/api/flavors/{flavor_id}".format(**locals())
+        response = self.client.get(url)
+        self.assertEqual(response.status_code, 200)
+        params = json.loads(response.data['params'])
+        self.assertEqual(params['region'], 'us-west-2')
+        self.assertEqual(params['zone'], 'any')
+        self.assertEqual(params['size'], 'c1.xlarge')
+        self.assertEqual(params['image'], 'ami-c98d1bf9')