From b3ff95eb4a45d47176049065fb6440b529e68f45 Mon Sep 17 00:00:00 2001
From: Fei Long Wang <flwang@catalyst.net.nz>
Date: Thu, 16 Feb 2017 13:43:42 +1300
Subject: [PATCH] Support purge queue

Change-Id: Ic20c9ad65d2a77c84afc38d7fcb4535cbaa1d01d
---
 setup.cfg                          |  1 +
 zaqarclient/queues/v2/api.py       |  8 ++++++++
 zaqarclient/queues/v2/cli.py       | 26 ++++++++++++++++++++++++++
 zaqarclient/queues/v2/core.py      | 21 +++++++++++++++++++++
 zaqarclient/queues/v2/queues.py    |  5 +++++
 zaqarclient/tests/queues/queues.py | 24 ++++++++++++++++++++++++
 6 files changed, 85 insertions(+)

diff --git a/setup.cfg b/setup.cfg
index ddd83506..ac7b6aa3 100644
--- a/setup.cfg
+++ b/setup.cfg
@@ -79,6 +79,7 @@ openstack.messaging.v2 =
     queue_stats = zaqarclient.queues.v2.cli:GetQueueStats
     queue_set_metadata = zaqarclient.queues.v2.cli:SetQueueMetadata
     queue_get_metadata = zaqarclient.queues.v2.cli:GetQueueMetadata
+    queue_purge = zaqarclient.queues.v2.cli:PurgeQueue
     pool_create = zaqarclient.queues.v2.cli:CreatePool
     pool_show = zaqarclient.queues.v2.cli:ShowPool
     pool_update = zaqarclient.queues.v2.cli:UpdatePool
diff --git a/zaqarclient/queues/v2/api.py b/zaqarclient/queues/v2/api.py
index 53de3f77..df560e4d 100644
--- a/zaqarclient/queues/v2/api.py
+++ b/zaqarclient/queues/v2/api.py
@@ -22,6 +22,14 @@ class V2(api.V1_1):
 
 
 V2.schema.update({
+    'queue_purge': {
+        'ref': 'queues/{queue_name}/purge',
+        'method': 'POST',
+        'required': ['queue_name'],
+        'properties': {
+            'queue_name': {'type': 'string'}
+        }
+    },
     'signed_url_create': {
         'ref': 'queues/{queue_name}/share',
         'method': 'POST',
diff --git a/zaqarclient/queues/v2/cli.py b/zaqarclient/queues/v2/cli.py
index 7d693099..3879fe62 100644
--- a/zaqarclient/queues/v2/cli.py
+++ b/zaqarclient/queues/v2/cli.py
@@ -57,6 +57,32 @@ class GetQueueMetadata(cli.GetQueueMetadata):
     pass
 
 
+class PurgeQueue(command.Command):
+    """Purge a queue"""
+
+    _description = _("Purge a queue")
+    log = logging.getLogger(__name__ + ".PurgeQueue")
+
+    def get_parser(self, prog_name):
+        parser = super(PurgeQueue, self).get_parser(prog_name)
+        parser.add_argument(
+            "queue_name",
+            metavar="<queue_name>",
+            help="Name of the queue")
+        parser.add_argument(
+            "--resource_types",
+            metavar="<resource_types>",
+            choices=['messages', 'subscriptions'],
+            help="Resource types want to be purged.")
+        return parser
+
+    def take_action(self, parsed_args):
+        client = _get_client(self, parsed_args)
+        queue_name = parsed_args.queue_name
+        client.queue(queue_name).purge(
+            resource_types=parsed_args.resource_types)
+
+
 class CreatePool(cli.CreatePool):
     """Create a pool"""
     pass
diff --git a/zaqarclient/queues/v2/core.py b/zaqarclient/queues/v2/core.py
index 7d5540e7..7475217e 100644
--- a/zaqarclient/queues/v2/core.py
+++ b/zaqarclient/queues/v2/core.py
@@ -88,6 +88,27 @@ def queue_update(transport, request, name, metadata, callback=None):
     return resp.deserialized_content
 
 
+def queue_purge(transport, request, name, resource_types=None):
+    """Purge resources under a queue
+
+    :param transport: Transport instance to use
+    :type transport: `transport.base.Transport`
+    :param request: Request instance ready to be sent.
+    :type request: `transport.request.Request`
+    :param name: Queue reference name.
+    :type name: `six.text_type`
+    :param resource_types: Resource types will be purged
+    :type resource_types: `list`
+    """
+    request.operation = 'queue_purge'
+    request.params['queue_name'] = name
+    if resource_types:
+        request.content = json.dumps({'resource_types': [resource_types]})
+
+    resp = transport.send(request)
+    return resp.deserialized_content
+
+
 def signed_url_create(transport, request, queue_name, paths=None,
                       ttl_seconds=None, project_id=None, methods=None):
     """Creates a signed URL given a queue name
diff --git a/zaqarclient/queues/v2/queues.py b/zaqarclient/queues/v2/queues.py
index 6099c7a8..72cebe1b 100644
--- a/zaqarclient/queues/v2/queues.py
+++ b/zaqarclient/queues/v2/queues.py
@@ -83,6 +83,11 @@ class Queue(queues.Queue):
 
         return self._metadata
 
+    def purge(self, resource_types=None):
+        req, trans = self.client._request_and_transport()
+        core.queue_purge(trans, req, self._name,
+                         resource_types=resource_types)
+
 
 def create_object(parent):
     return lambda args: Queue(parent, args["name"], auto_create=False)
diff --git a/zaqarclient/tests/queues/queues.py b/zaqarclient/tests/queues/queues.py
index 63b4eb3b..e992e6b1 100644
--- a/zaqarclient/tests/queues/queues.py
+++ b/zaqarclient/tests/queues/queues.py
@@ -558,6 +558,30 @@ class QueuesV2QueueUnitTest(QueuesV1_1QueueUnitTest):
             expect_metadata = {"name": 'test1'}
             self.assertEqual(expect_metadata, metadata)
 
+    def test_queue_purge(self):
+        with mock.patch.object(self.transport, 'send',
+                               autospec=True) as send_method:
+
+            resp = response.Response(None, None)
+            send_method.return_value = resp
+
+            self.queue.purge()
+
+            # NOTE(flwang): Nothing to assert here,
+            # just checking our way down to the transport
+            # doesn't crash.
+
+    def test_queue_purge_messages(self):
+        with mock.patch.object(self.transport, 'send',
+                               autospec=True) as send_method:
+
+            resp = response.Response(None, None)
+            send_method.return_value = resp
+
+            self.queue.purge(resource_types=['messages'])
+            self.assertEqual({"resource_types": [["messages"]]},
+                             json.loads(send_method.call_args[0][0].content))
+
 
 class QueuesV2QueueFunctionalTest(QueuesV1_1QueueFunctionalTest):