Jakub Ruzicka fcc2a65
From d7ca548883971b8042660af5523ba74e3eb78bf0 Mon Sep 17 00:00:00 2001
Jakub Ruzicka fcc2a65
From: Gary Kotton <gkotton@vmware.com>
Jakub Ruzicka fcc2a65
Date: Thu, 13 Jun 2013 15:25:39 +0000
Jakub Ruzicka fcc2a65
Subject: [PATCH] Nova CLI for server groups
Jakub Ruzicka fcc2a65
Jakub Ruzicka fcc2a65
CLI support for blueprint instance-group-api-extension
Jakub Ruzicka fcc2a65
Jakub Ruzicka fcc2a65
REST API support:- https://review.openstack.org/#/c/62557/
Jakub Ruzicka fcc2a65
Jakub Ruzicka fcc2a65
DocImpact
Jakub Ruzicka fcc2a65
 - supports create, list, get and delete
Jakub Ruzicka fcc2a65
 - only V2 is supported
Jakub Ruzicka fcc2a65
Jakub Ruzicka fcc2a65
Change-Id: Iaa5a2922b9a0eed9f682b7584c2acf582379b422
Jakub Ruzicka fcc2a65
---
Jakub Ruzicka fcc2a65
 novaclient/tests/v1_1/fakes.py              | 45 ++++++++++++++++++
Jakub Ruzicka fcc2a65
 novaclient/tests/v1_1/test_server_groups.py | 52 +++++++++++++++++++++
Jakub Ruzicka fcc2a65
 novaclient/v1_1/client.py                   |  2 +
Jakub Ruzicka fcc2a65
 novaclient/v1_1/server_groups.py            | 71 +++++++++++++++++++++++++++++
Jakub Ruzicka fcc2a65
 novaclient/v1_1/shell.py                    | 39 ++++++++++++++++
Jakub Ruzicka fcc2a65
 5 files changed, 209 insertions(+)
Jakub Ruzicka fcc2a65
 create mode 100644 novaclient/tests/v1_1/test_server_groups.py
Jakub Ruzicka fcc2a65
 create mode 100644 novaclient/v1_1/server_groups.py
Jakub Ruzicka fcc2a65
Jakub Ruzicka fcc2a65
diff --git a/novaclient/tests/v1_1/fakes.py b/novaclient/tests/v1_1/fakes.py
Jakub Ruzicka fcc2a65
index dd8fdcc..50658c1 100644
Jakub Ruzicka fcc2a65
--- a/novaclient/tests/v1_1/fakes.py
Jakub Ruzicka fcc2a65
+++ b/novaclient/tests/v1_1/fakes.py
Jakub Ruzicka fcc2a65
@@ -1996,3 +1996,48 @@ class FakeHTTPClient(base_client.HTTPClient):
Jakub Ruzicka fcc2a65
         return (200, {}, {'events': [
Jakub Ruzicka fcc2a65
                     {'name': 'network-changed',
Jakub Ruzicka fcc2a65
                      'server_uuid': '1234'}]})
Jakub Ruzicka fcc2a65
+
Jakub Ruzicka fcc2a65
+    #
Jakub Ruzicka fcc2a65
+    # Server Groups
Jakub Ruzicka fcc2a65
+    #
Jakub Ruzicka fcc2a65
+
Jakub Ruzicka fcc2a65
+    def get_os_server_groups(self, *kw):
Jakub Ruzicka fcc2a65
+        return (200, {},
Jakub Ruzicka fcc2a65
+                {"server_groups": [
Jakub Ruzicka fcc2a65
+                 {"members": [], "metadata": {},
Jakub Ruzicka fcc2a65
+                  "id": "2cbd51f4-fafe-4cdb-801b-cf913a6f288b",
Jakub Ruzicka fcc2a65
+                  "policies": [], "name": "ig1"},
Jakub Ruzicka fcc2a65
+                 {"members": [], "metadata": {},
Jakub Ruzicka fcc2a65
+                  "id": "4473bb03-4370-4bfb-80d3-dc8cffc47d94",
Jakub Ruzicka fcc2a65
+                  "policies": ["anti-affinity"], "name": "ig2"},
Jakub Ruzicka fcc2a65
+                 {"members": [], "metadata": {"key": "value"},
Jakub Ruzicka fcc2a65
+                  "id": "31ab9bdb-55e1-4ac3-b094-97eeb1b65cc4",
Jakub Ruzicka fcc2a65
+                  "policies": [], "name": "ig3"},
Jakub Ruzicka fcc2a65
+                 {"members": ["2dccb4a1-02b9-482a-aa23-5799490d6f5d"],
Jakub Ruzicka fcc2a65
+                  "metadata": {},
Jakub Ruzicka fcc2a65
+                  "id": "4890bb03-7070-45fb-8453-d34556c87d94",
Jakub Ruzicka fcc2a65
+                  "policies": ["anti-affinity"], "name": "ig2"}]})
Jakub Ruzicka fcc2a65
+
Jakub Ruzicka fcc2a65
+    def _return_server_group(self):
Jakub Ruzicka fcc2a65
+        r = {'server_group':
Jakub Ruzicka fcc2a65
+             self.get_os_server_groups()[2]['server_groups'][0]}
Jakub Ruzicka fcc2a65
+        return (200, {}, r)
Jakub Ruzicka fcc2a65
+
Jakub Ruzicka fcc2a65
+    def post_os_server_groups(self, body, **kw):
Jakub Ruzicka fcc2a65
+        return self._return_server_group()
Jakub Ruzicka fcc2a65
+
Jakub Ruzicka fcc2a65
+    def get_os_server_groups_2cbd51f4_fafe_4cdb_801b_cf913a6f288b(self,
Jakub Ruzicka fcc2a65
+                                                                    **kw):
Jakub Ruzicka fcc2a65
+        return self._return_server_group()
Jakub Ruzicka fcc2a65
+
Jakub Ruzicka fcc2a65
+    def put_os_server_groups_2cbd51f4_fafe_4cdb_801b_cf913a6f288b(self,
Jakub Ruzicka fcc2a65
+                                                                    **kw):
Jakub Ruzicka fcc2a65
+        return self._return_server_group()
Jakub Ruzicka fcc2a65
+
Jakub Ruzicka fcc2a65
+    def post_os_server_groups_2cbd51f4_fafe_4cdb_801b_cf913a6f288b_action(
Jakub Ruzicka fcc2a65
+            self, body, **kw):
Jakub Ruzicka fcc2a65
+        return self._return_server_group()
Jakub Ruzicka fcc2a65
+
Jakub Ruzicka fcc2a65
+    def delete_os_server_groups_2cbd51f4_fafe_4cdb_801b_cf913a6f288b(
Jakub Ruzicka fcc2a65
+            self, **kw):
Jakub Ruzicka fcc2a65
+        return (202, {}, None)
Jakub Ruzicka fcc2a65
diff --git a/novaclient/tests/v1_1/test_server_groups.py b/novaclient/tests/v1_1/test_server_groups.py
Jakub Ruzicka fcc2a65
new file mode 100644
Jakub Ruzicka fcc2a65
index 0000000..fde5def
Jakub Ruzicka fcc2a65
--- /dev/null
Jakub Ruzicka fcc2a65
+++ b/novaclient/tests/v1_1/test_server_groups.py
Jakub Ruzicka fcc2a65
@@ -0,0 +1,52 @@
Jakub Ruzicka fcc2a65
+# Copyright (c) 2014 VMware, Inc.
Jakub Ruzicka fcc2a65
+# All Rights Reserved.
Jakub Ruzicka fcc2a65
+#
Jakub Ruzicka fcc2a65
+#    Licensed under the Apache License, Version 2.0 (the "License"); you may
Jakub Ruzicka fcc2a65
+#    not use this file except in compliance with the License. You may obtain
Jakub Ruzicka fcc2a65
+#    a copy of the License at
Jakub Ruzicka fcc2a65
+#
Jakub Ruzicka fcc2a65
+#         http://www.apache.org/licenses/LICENSE-2.0
Jakub Ruzicka fcc2a65
+#
Jakub Ruzicka fcc2a65
+#    Unless required by applicable law or agreed to in writing, software
Jakub Ruzicka fcc2a65
+#    distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
Jakub Ruzicka fcc2a65
+#    WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
Jakub Ruzicka fcc2a65
+#    License for the specific language governing permissions and limitations
Jakub Ruzicka fcc2a65
+#    under the License.
Jakub Ruzicka fcc2a65
+
Jakub Ruzicka fcc2a65
+from novaclient.tests import utils
Jakub Ruzicka fcc2a65
+from novaclient.tests.v1_1 import fakes
Jakub Ruzicka fcc2a65
+from novaclient.v1_1 import server_groups
Jakub Ruzicka fcc2a65
+
Jakub Ruzicka fcc2a65
+
Jakub Ruzicka fcc2a65
+cs = fakes.FakeClient()
Jakub Ruzicka fcc2a65
+
Jakub Ruzicka fcc2a65
+
Jakub Ruzicka fcc2a65
+class ServerGroupsTest(utils.TestCase):
Jakub Ruzicka fcc2a65
+
Jakub Ruzicka fcc2a65
+    def test_list_server_groups(self):
Jakub Ruzicka fcc2a65
+        result = cs.server_groups.list()
Jakub Ruzicka fcc2a65
+        cs.assert_called('GET', '/os-server-groups')
Jakub Ruzicka fcc2a65
+        for server_group in result:
Jakub Ruzicka fcc2a65
+            self.assertTrue(isinstance(server_group,
Jakub Ruzicka fcc2a65
+                                       server_groups.ServerGroup))
Jakub Ruzicka fcc2a65
+
Jakub Ruzicka fcc2a65
+    def test_create_server_group(self):
Jakub Ruzicka fcc2a65
+        kwargs = {'name': 'ig1',
Jakub Ruzicka fcc2a65
+                  'policies': ['anti-affinity']}
Jakub Ruzicka fcc2a65
+        server_group = cs.server_groups.create(**kwargs)
Jakub Ruzicka fcc2a65
+        body = {'server_group': kwargs}
Jakub Ruzicka fcc2a65
+        cs.assert_called('POST', '/os-server-groups', body)
Jakub Ruzicka fcc2a65
+        self.assertTrue(isinstance(server_group,
Jakub Ruzicka fcc2a65
+                                   server_groups.ServerGroup))
Jakub Ruzicka fcc2a65
+
Jakub Ruzicka fcc2a65
+    def test_get_server_group(self):
Jakub Ruzicka fcc2a65
+        id = '2cbd51f4-fafe-4cdb-801b-cf913a6f288b'
Jakub Ruzicka fcc2a65
+        server_group = cs.server_groups.get(id)
Jakub Ruzicka fcc2a65
+        cs.assert_called('GET', '/os-server-groups/%s' % id)
Jakub Ruzicka fcc2a65
+        self.assertTrue(isinstance(server_group,
Jakub Ruzicka fcc2a65
+                                   server_groups.ServerGroup))
Jakub Ruzicka fcc2a65
+
Jakub Ruzicka fcc2a65
+    def test_delete_server_group(self):
Jakub Ruzicka fcc2a65
+        id = '2cbd51f4-fafe-4cdb-801b-cf913a6f288b'
Jakub Ruzicka fcc2a65
+        cs.server_groups.delete(id)
Jakub Ruzicka fcc2a65
+        cs.assert_called('DELETE', '/os-server-groups/%s' % id)
Jakub Ruzicka fcc2a65
diff --git a/novaclient/v1_1/client.py b/novaclient/v1_1/client.py
Jakub Ruzicka fcc2a65
index ce4aea1..9291270 100644
Jakub Ruzicka fcc2a65
--- a/novaclient/v1_1/client.py
Jakub Ruzicka fcc2a65
+++ b/novaclient/v1_1/client.py
Jakub Ruzicka fcc2a65
@@ -37,6 +37,7 @@ from novaclient.v1_1 import quota_classes
Jakub Ruzicka fcc2a65
 from novaclient.v1_1 import quotas
Jakub Ruzicka fcc2a65
 from novaclient.v1_1 import security_group_rules
Jakub Ruzicka fcc2a65
 from novaclient.v1_1 import security_groups
Jakub Ruzicka fcc2a65
+from novaclient.v1_1 import server_groups
Jakub Ruzicka fcc2a65
 from novaclient.v1_1 import servers
Jakub Ruzicka fcc2a65
 from novaclient.v1_1 import services
Jakub Ruzicka fcc2a65
 from novaclient.v1_1 import usage
Jakub Ruzicka fcc2a65
@@ -131,6 +132,7 @@ class Client(object):
Jakub Ruzicka fcc2a65
         self.os_cache = os_cache or not no_cache
Jakub Ruzicka fcc2a65
         self.availability_zones = \
Jakub Ruzicka fcc2a65
             availability_zones.AvailabilityZoneManager(self)
Jakub Ruzicka fcc2a65
+        self.server_groups = server_groups.ServerGroupsManager(self)
Jakub Ruzicka fcc2a65
 
Jakub Ruzicka fcc2a65
         # Add in any extensions...
Jakub Ruzicka fcc2a65
         if extensions:
Jakub Ruzicka fcc2a65
diff --git a/novaclient/v1_1/server_groups.py b/novaclient/v1_1/server_groups.py
Jakub Ruzicka fcc2a65
new file mode 100644
Jakub Ruzicka fcc2a65
index 0000000..be6ff8e
Jakub Ruzicka fcc2a65
--- /dev/null
Jakub Ruzicka fcc2a65
+++ b/novaclient/v1_1/server_groups.py
Jakub Ruzicka fcc2a65
@@ -0,0 +1,71 @@
Jakub Ruzicka fcc2a65
+# Copyright (c) 2014 VMware, Inc.
Jakub Ruzicka fcc2a65
+# All Rights Reserved.
Jakub Ruzicka fcc2a65
+#
Jakub Ruzicka fcc2a65
+#    Licensed under the Apache License, Version 2.0 (the "License"); you may
Jakub Ruzicka fcc2a65
+#    not use this file except in compliance with the License. You may obtain
Jakub Ruzicka fcc2a65
+#    a copy of the License at
Jakub Ruzicka fcc2a65
+#
Jakub Ruzicka fcc2a65
+#         http://www.apache.org/licenses/LICENSE-2.0
Jakub Ruzicka fcc2a65
+#
Jakub Ruzicka fcc2a65
+#    Unless required by applicable law or agreed to in writing, software
Jakub Ruzicka fcc2a65
+#    distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
Jakub Ruzicka fcc2a65
+#    WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
Jakub Ruzicka fcc2a65
+#    License for the specific language governing permissions and limitations
Jakub Ruzicka fcc2a65
+#    under the License.
Jakub Ruzicka fcc2a65
+
Jakub Ruzicka fcc2a65
+"""
Jakub Ruzicka fcc2a65
+Server group interface.
Jakub Ruzicka fcc2a65
+"""
Jakub Ruzicka fcc2a65
+
Jakub Ruzicka fcc2a65
+from novaclient import base
Jakub Ruzicka fcc2a65
+
Jakub Ruzicka fcc2a65
+
Jakub Ruzicka fcc2a65
+class ServerGroup(base.Resource):
Jakub Ruzicka fcc2a65
+    """
Jakub Ruzicka fcc2a65
+    A server group.
Jakub Ruzicka fcc2a65
+    """
Jakub Ruzicka fcc2a65
+    NAME_ATTR = 'server_group_name'
Jakub Ruzicka fcc2a65
+
Jakub Ruzicka fcc2a65
+    def __repr__(self):
Jakub Ruzicka fcc2a65
+        return '<ServerGroup: %s>' % self.id
Jakub Ruzicka fcc2a65
+
Jakub Ruzicka fcc2a65
+    def delete(self):
Jakub Ruzicka fcc2a65
+        self.manager.delete(self)
Jakub Ruzicka fcc2a65
+
Jakub Ruzicka fcc2a65
+
Jakub Ruzicka fcc2a65
+class ServerGroupsManager(base.ManagerWithFind):
Jakub Ruzicka fcc2a65
+    """
Jakub Ruzicka fcc2a65
+    Manage :class:`ServerGroup` resources.
Jakub Ruzicka fcc2a65
+    """
Jakub Ruzicka fcc2a65
+    resource_class = ServerGroup
Jakub Ruzicka fcc2a65
+
Jakub Ruzicka fcc2a65
+    def list(self):
Jakub Ruzicka fcc2a65
+        """Get a list of all server groups.
Jakub Ruzicka fcc2a65
+
Jakub Ruzicka fcc2a65
+        :rtype: list of :class:`ServerGroup`.
Jakub Ruzicka fcc2a65
+        """
Jakub Ruzicka fcc2a65
+        return self._list('/os-server-groups', 'server_groups')
Jakub Ruzicka fcc2a65
+
Jakub Ruzicka fcc2a65
+    def get(self, id):
Jakub Ruzicka fcc2a65
+        """Get a specific server group.
Jakub Ruzicka fcc2a65
+
Jakub Ruzicka fcc2a65
+        :param id: The ID of the :class:`ServerGroup` to get.
Jakub Ruzicka fcc2a65
+        :rtype: :class:`ServerGroup`
Jakub Ruzicka fcc2a65
+        """
Jakub Ruzicka fcc2a65
+        return self._get('/os-server-groups/%s' % id,
Jakub Ruzicka fcc2a65
+                         'server_group')
Jakub Ruzicka fcc2a65
+
Jakub Ruzicka fcc2a65
+    def delete(self, id):
Jakub Ruzicka fcc2a65
+        """Delete a specific server group.
Jakub Ruzicka fcc2a65
+
Jakub Ruzicka fcc2a65
+        :param id: The ID of the :class:`ServerGroup` to delete.
Jakub Ruzicka fcc2a65
+        """
Jakub Ruzicka fcc2a65
+        self._delete('/os-server-groups/%s' % id)
Jakub Ruzicka fcc2a65
+
Jakub Ruzicka fcc2a65
+    def create(self, **kwargs):
Jakub Ruzicka fcc2a65
+        """Create (allocate) a server group.
Jakub Ruzicka fcc2a65
+
Jakub Ruzicka fcc2a65
+        :rtype: list of :class:`ServerGroup`
Jakub Ruzicka fcc2a65
+        """
Jakub Ruzicka fcc2a65
+        body = {'server_group': kwargs}
Jakub Ruzicka fcc2a65
+        return self._create('/os-server-groups', body, 'server_group')
Jakub Ruzicka fcc2a65
diff --git a/novaclient/v1_1/shell.py b/novaclient/v1_1/shell.py
Jakub Ruzicka fcc2a65
index a043d3c..c1754c9 100644
Jakub Ruzicka fcc2a65
--- a/novaclient/v1_1/shell.py
Jakub Ruzicka fcc2a65
+++ b/novaclient/v1_1/shell.py
Jakub Ruzicka fcc2a65
@@ -3512,3 +3512,42 @@ def do_availability_zone_list(cs, _args):
Jakub Ruzicka fcc2a65
     _translate_availability_zone_keys(result)
Jakub Ruzicka fcc2a65
     utils.print_list(result, ['Name', 'Status'],
Jakub Ruzicka fcc2a65
                      sortby_index=None)
Jakub Ruzicka fcc2a65
+
Jakub Ruzicka fcc2a65
+
Jakub Ruzicka fcc2a65
+def _print_server_group_details(server_group):
Jakub Ruzicka fcc2a65
+    columns = ['Id', 'Name', 'Policies', 'Members', 'Metadata']
Jakub Ruzicka fcc2a65
+    utils.print_list(server_group, columns)
Jakub Ruzicka fcc2a65
+
Jakub Ruzicka fcc2a65
+
Jakub Ruzicka fcc2a65
+def do_server_group_list(cs, args):
Jakub Ruzicka fcc2a65
+    """Print a list of all server groups."""
Jakub Ruzicka fcc2a65
+    server_groups = cs.server_groups.list()
Jakub Ruzicka fcc2a65
+    _print_server_group_details(server_groups)
Jakub Ruzicka fcc2a65
+
Jakub Ruzicka fcc2a65
+
Jakub Ruzicka fcc2a65
+@utils.arg('name', metavar='<name>', help='Server group name.')
Jakub Ruzicka fcc2a65
+@utils.arg('--policy', metavar='<policy>', action='append',
Jakub Ruzicka fcc2a65
+           dest='policies', default=[], type=str,
Jakub Ruzicka fcc2a65
+           help='Policies for the server groups')
Jakub Ruzicka fcc2a65
+def do_server_group_create(cs, args):
Jakub Ruzicka fcc2a65
+    """Create a new server group with the specified details."""
Jakub Ruzicka fcc2a65
+    kwargs = {'name': args.name,
Jakub Ruzicka fcc2a65
+              'policies': args.policies}
Jakub Ruzicka fcc2a65
+    server_group = cs.server_groups.create(**kwargs)
Jakub Ruzicka fcc2a65
+    _print_server_group_details([server_group])
Jakub Ruzicka fcc2a65
+
Jakub Ruzicka fcc2a65
+
Jakub Ruzicka fcc2a65
+@utils.arg('id', metavar='<id>',
Jakub Ruzicka fcc2a65
+           help="Unique ID of the server group to delete")
Jakub Ruzicka fcc2a65
+def do_server_group_delete(cs, args):
Jakub Ruzicka fcc2a65
+    """Delete a specific server group."""
Jakub Ruzicka fcc2a65
+    cs.server_groups.delete(args.id)
Jakub Ruzicka fcc2a65
+    print("Instance group %s has been successfully deleted." % args.id)
Jakub Ruzicka fcc2a65
+
Jakub Ruzicka fcc2a65
+
Jakub Ruzicka fcc2a65
+@utils.arg('id', metavar='<id>',
Jakub Ruzicka fcc2a65
+           help="Unique ID of the server group to get")
Jakub Ruzicka fcc2a65
+def do_server_group_get(cs, args):
Jakub Ruzicka fcc2a65
+    """Get a specific server group."""
Jakub Ruzicka fcc2a65
+    server_group = cs.server_groups.get(args.id)
Jakub Ruzicka fcc2a65
+    _print_server_group_details([server_group])