Fixed #151 -- deis flavors:update working with tests.

Author: mboersma

api/models.py

@@ -206,11 +206,6 @@ 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

@@ -87,6 +87,8 @@ def test_flavor_update(self):
         url = "/api/flavors/{flavor_id}".format(**locals())
         response = self.client.get(url)
         self.assertEqual(response.status_code, 200)
+        uuid = response.data['uuid']
+        self.assertRegexpMatches(uuid, r'^\w{8}-\w{4}-\w{4}-\w{4}-\w{12}$')
         params = json.loads(response.data['params'])
         self.assertEqual(params['region'], 'us-west-2')
         self.assertEqual(params['zone'], 'any')
@@ -106,8 +108,9 @@ def test_flavor_update(self):
         url = "/api/flavors/{flavor_id}".format(**locals())
         response = self.client.get(url)
         self.assertEqual(response.status_code, 200)
+        self.assertEqual(response.data['uuid'], uuid)
         params = json.loads(response.data['params'])
-        self.assertEqual(params['region'], 'us-west-2')
-        self.assertEqual(params['zone'], 'any')
+        self.assertNotIn('region', params)
+        self.assertNotIn('zone', params)
         self.assertEqual(params['size'], 'c1.xlarge')
         self.assertEqual(params['image'], 'ami-c98d1bf9')

client/deis.py

@@ -626,6 +626,7 @@ def flavors(self, args):
         Valid commands for flavors:
 
         flavors:create        create a new node flavor
+        flavors:update        update an existing node flavor
         flavors:info          print information about a node flavor
         flavors:list          list available flavors
         flavors:delete        delete a node flavor
@@ -658,6 +659,30 @@ def flavors_create(self, args):
         else:
             print('Error!', response.text)
 
+    def flavors_update(self, args):
+        """
+        Create an existing node flavor
+
+        Usage: deis flavors:update <id> --params=<params> [options]
+
+        Options:
+
+        --params=PARAMS    provider-specific parameters (size, region, zone, etc.)
+        --init=INIT        override Ubuntu cloud-init with custom YAML
+        """
+        flavor = args.get('<id>')
+        body = {'id': flavor}
+        fields = ('params', 'init')
+        for fld in fields:
+            opt = args.get('--' + fld)
+            if opt:
+                body.update({fld: opt})
+        response = self._dispatch('patch', '/api/flavors/{}'.format(flavor), json.dumps(body))
+        if response.status_code == requests.codes.ok:  # @UndefinedVariable
+            print("{0[id]}".format(response.json()))
+        else:
+            print('Error!', response.text)
+
     def flavors_delete(self, args):
         """
         Delete a node flavor