Now cluster-template-update works for "labels"

Now we are handling the label values converting them as a dictionary.
But as magnum-api only accepts the input as string. We are converting
the dictionary into string and passing to magnum-api. It is parsing
the string to dictionary and storing the label value a dictionary.

Change-Id: I2cec19e24c6dd32e209ca65a26a2f1999d8289e0
Depends-on: I4d64da78dc4ed4d5599533b54861b65bce609c28
Closes-Bug: #1659189

2 files changed, 16 insertions, 1 deletions

diff --git a/magnumclient/common/utils.py b/magnumclient/common/utils.py
index 57ce02e..af5642f 100644
--- a/magnumclient/common/utils.py
+++ b/magnumclient/common/utils.py
@@ -78,7 +78,13 @@ def args_array_to_patch(op, attributes):
             attr = '/' + attr
         if op in ['add', 'replace']:
             path, value = split_and_deserialize(attr)
-            patch.append({'op': op, 'path': path, 'value': value})
+            if path == "/labels":
+                a = []
+                a.append(value)
+                value = str(handle_labels(a))
+                patch.append({'op': op, 'path': path, 'value': value})
+            else:
+                patch.append({'op': op, 'path': path, 'value': value})
 
         elif op == "remove":
             # For remove only the key is needed
diff --git a/magnumclient/tests/v1/test_clustertemplates_shell.py b/magnumclient/tests/v1/test_clustertemplates_shell.py
index bb53ff6..67bce87 100644
--- a/magnumclient/tests/v1/test_clustertemplates_shell.py
+++ b/magnumclient/tests/v1/test_clustertemplates_shell.py
@@ -681,6 +681,15 @@ class ShellTest(shell_test_base.TestCommandLineArgument):
 
     @mock.patch(
         'magnumclient.v1.cluster_templates.ClusterTemplateManager.update')
+    def test_cluster_template_update_label(self, mock_update):
+        self._test_arg_success('cluster-template-update test '
+                               'replace labels=key1=val1')
+        patch = [{'op': 'replace', 'path': '/labels',
+                 'value': "{'key1': 'val1'}"}]
+        mock_update.assert_called_once_with('test', patch)
+
+    @mock.patch(
+        'magnumclient.v1.cluster_templates.ClusterTemplateManager.update')
     def test_cluster_template_update_failure_wrong_op(self, mock_update):
         _error_msg = [
             '.*?^usage: magnum cluster-template-update ',