From dd0958860d5df5b7da9be50a294912a47e016087 Mon Sep 17 00:00:00 2001
From: Ricardo Rocha <rocha.porto@gmail.com>
Date: Tue, 1 Aug 2017 14:44:52 +0000
Subject: [PATCH] OSC: Add --flavor to coe cluster create

Add a flavor override parameter on osc cluster create to override
the value present on the cluster template.

Partial-Bug: #1699245
Depends-On: Ib60c05cce1cf2639ca4740abdd264403033433f9
Change-Id: Ib00514b66ee2d0a2ee4ddafb5b8c2ff8fa82de77
---
 magnumclient/osc/v1/clusters.py                 | 7 +++++++
 magnumclient/tests/osc/unit/v1/fakes.py         | 1 +
 magnumclient/tests/osc/unit/v1/test_clusters.py | 1 +
 magnumclient/v1/clusters.py                     | 1 +
 4 files changed, 10 insertions(+)

diff --git a/magnumclient/osc/v1/clusters.py b/magnumclient/osc/v1/clusters.py
index 27d3caef..3a92acd4 100644
--- a/magnumclient/osc/v1/clusters.py
+++ b/magnumclient/osc/v1/clusters.py
@@ -44,6 +44,7 @@ CLUSTER_ATTRIBUTES = [
     'container_version',
     'name',
     'master_flavor_id',
+    'flavor_id',
 ]
 
 
@@ -108,6 +109,11 @@ class CreateCluster(command.Command):
             metavar='<master-flavor>',
             help=_('The nova flavor name or UUID to use when launching the '
                    'master node of the Cluster.'))
+        parser.add_argument(
+            '--flavor',
+            metavar='<flavor>',
+            help=_('The nova flavor name or UUID to use when launching the '
+                   'Cluster.'))
 
         return parser
 
@@ -125,6 +131,7 @@ class CreateCluster(command.Command):
             'name': parsed_args.name,
             'node_count': parsed_args.node_count,
             'master_flavor_id': parsed_args.master_flavor,
+            'flavor_id': parsed_args.flavor,
         }
 
         if parsed_args.labels is not None:
diff --git a/magnumclient/tests/osc/unit/v1/fakes.py b/magnumclient/tests/osc/unit/v1/fakes.py
index 034dfe76..703e3c43 100644
--- a/magnumclient/tests/osc/unit/v1/fakes.py
+++ b/magnumclient/tests/osc/unit/v1/fakes.py
@@ -203,6 +203,7 @@ class FakeCluster(object):
             'container_version': None,
             'name': 'fake-cluster',
             'master_flavor_id': None,
+            'flavor_id': 'm1.medium',
         }
 
         # Overwrite default attributes.
diff --git a/magnumclient/tests/osc/unit/v1/test_clusters.py b/magnumclient/tests/osc/unit/v1/test_clusters.py
index 065f8ee2..62f76718 100644
--- a/magnumclient/tests/osc/unit/v1/test_clusters.py
+++ b/magnumclient/tests/osc/unit/v1/test_clusters.py
@@ -44,6 +44,7 @@ class TestClusterCreate(TestCluster):
             'discovery_url': None,
             'docker_volume_size': None,
             'labels': {},
+            'flavor_id': None,
             'keypair': None,
             'master_count': 1,
             'name': 'fake-cluster-1',
diff --git a/magnumclient/v1/clusters.py b/magnumclient/v1/clusters.py
index df5ebfee..cdb599a5 100644
--- a/magnumclient/v1/clusters.py
+++ b/magnumclient/v1/clusters.py
@@ -22,6 +22,7 @@ CREATION_ATTRIBUTES.append('keypair')
 CREATION_ATTRIBUTES.append('docker_volume_size')
 CREATION_ATTRIBUTES.append('labels')
 CREATION_ATTRIBUTES.append('master_flavor_id')
+CREATION_ATTRIBUTES.append('flavor_id')
 
 
 class Cluster(baseunit.BaseTemplate):