d5b24ef15c
The patch adds Shard support of the client. This will allow users to create and delete shards from Marconi. The API requires the Marconi instance to be and admin instance. However, there are no checks, restrictions, yet. This patch doesn't implement Shards listing. As soon as the MessageIterator lands, it'll be possible to reuse that code for the ShardsIterator. Partially-Implements blueprint: shard-management Partially-Implements blueprint: python-marconiclient-v1 Change-Id: I932c98d68a4a81b710dbea75a6bac123b7afaa08
190 lines
6.9 KiB
Python
190 lines
6.9 KiB
Python
# Copyright (c) 2013 Red Hat, Inc.
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
# you may not use this file except in compliance with the License.
|
|
# You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
|
|
# implied.
|
|
# See the License for the specific language governing permissions and
|
|
# limitations under the License.
|
|
|
|
import json
|
|
import mock
|
|
|
|
from marconiclient.queues.v1 import core
|
|
from marconiclient.tests import base
|
|
from marconiclient.tests.transport import dummy
|
|
import marconiclient.transport.errors as errors
|
|
from marconiclient.transport import request
|
|
from marconiclient.transport import response
|
|
|
|
|
|
class TestV1Core(base.TestBase):
|
|
|
|
def setUp(self):
|
|
super(TestV1Core, self).setUp()
|
|
self.transport = dummy.DummyTransport(self.conf)
|
|
|
|
def test_queue_create(self):
|
|
with mock.patch.object(self.transport, 'send',
|
|
autospec=True) as send_method:
|
|
send_method.return_value = response.Response(None, None)
|
|
|
|
req = request.Request()
|
|
core.queue_create(self.transport, req, 'test')
|
|
self.assertIn('queue_name', req.params)
|
|
|
|
def test_queue_delete(self):
|
|
with mock.patch.object(self.transport, 'send',
|
|
autospec=True) as send_method:
|
|
send_method.return_value = response.Response(None, None)
|
|
|
|
req = request.Request()
|
|
core.queue_delete(self.transport, req, 'test')
|
|
self.assertIn('queue_name', req.params)
|
|
|
|
def test_queue_exists(self):
|
|
with mock.patch.object(self.transport, 'send',
|
|
autospec=True) as send_method:
|
|
send_method.return_value = response.Response(None, None)
|
|
|
|
req = request.Request()
|
|
ret = core.queue_exists(self.transport, req, 'test')
|
|
self.assertIn('queue_name', req.params)
|
|
self.assertTrue(ret)
|
|
|
|
def test_queue_exists_not_found(self):
|
|
with mock.patch.object(self.transport, 'send',
|
|
autospec=True) as send_method:
|
|
|
|
send_method.side_effect = errors.ResourceNotFound
|
|
|
|
req = request.Request()
|
|
ret = core.queue_exists(self.transport, req, 'test')
|
|
self.assertIn('queue_name', req.params)
|
|
self.assertFalse(ret)
|
|
|
|
def test_get_queue_metadata(self):
|
|
with mock.patch.object(self.transport, 'send',
|
|
autospec=True) as send_method:
|
|
resp = response.Response(None, '{}')
|
|
send_method.return_value = resp
|
|
|
|
req = request.Request()
|
|
core.queue_get_metadata(self.transport, req, 'test')
|
|
|
|
def test_set_queue_metadata(self):
|
|
update_data = {'some': 'data'}
|
|
with mock.patch.object(self.transport, 'send',
|
|
autospec=True) as send_method:
|
|
send_method.return_value = response.Response(None, None)
|
|
|
|
req = request.Request()
|
|
core.queue_exists(self.transport, req, update_data, 'test')
|
|
self.assertIn('queue_name', req.params)
|
|
|
|
def test_message_post(self):
|
|
messages = [{'ttl': 30, 'body': 'Post It!'}]
|
|
|
|
with mock.patch.object(self.transport, 'send',
|
|
autospec=True) as send_method:
|
|
resp = response.Response(None, '{}')
|
|
send_method.return_value = resp
|
|
|
|
req = request.Request()
|
|
|
|
core.message_post(self.transport, req, 'test', messages)
|
|
self.assertIn('queue_name', req.params)
|
|
self.assertEqual(json.loads(req.content),
|
|
messages)
|
|
|
|
def test_message_list(self):
|
|
with mock.patch.object(self.transport, 'send',
|
|
autospec=True) as send_method:
|
|
resp = response.Response(None, '{}')
|
|
send_method.return_value = resp
|
|
|
|
req = request.Request()
|
|
|
|
core.message_list(self.transport, req, 'test')
|
|
self.assertIn('queue_name', req.params)
|
|
|
|
def test_message_list_kwargs(self):
|
|
with mock.patch.object(self.transport, 'send',
|
|
autospec=True) as send_method:
|
|
resp = response.Response(None, '{}')
|
|
send_method.return_value = resp
|
|
|
|
req = request.Request()
|
|
|
|
core.message_list(self.transport, req, 'test',
|
|
marker='supermarket',
|
|
echo=False, limit=10)
|
|
|
|
self.assertIn('queue_name', req.params)
|
|
self.assertIn('limit', req.params)
|
|
self.assertIn('echo', req.params)
|
|
self.assertIn('marker', req.params)
|
|
|
|
def test_message_get_many(self):
|
|
with mock.patch.object(self.transport, 'send',
|
|
autospec=True) as send_method:
|
|
resp = response.Response(None, '{}')
|
|
send_method.return_value = resp
|
|
|
|
req = request.Request()
|
|
|
|
ids = ['a', 'b']
|
|
core.message_get_many(self.transport, req,
|
|
'test', ids)
|
|
|
|
self.assertIn('queue_name', req.params)
|
|
self.assertIn('ids', req.params)
|
|
self.assertEqual(ids, req.params['ids'])
|
|
|
|
def test_message_get(self):
|
|
with mock.patch.object(self.transport, 'send',
|
|
autospec=True) as send_method:
|
|
resp = response.Response(None, '{}')
|
|
send_method.return_value = resp
|
|
|
|
req = request.Request()
|
|
core.message_get(self.transport, req,
|
|
'test', 'message_id')
|
|
|
|
def test_message_delete(self):
|
|
with mock.patch.object(self.transport, 'send',
|
|
autospec=True) as send_method:
|
|
resp = response.Response(None, None)
|
|
send_method.return_value = resp
|
|
|
|
req = request.Request()
|
|
core.message_delete(self.transport, req,
|
|
'test', 'message_id')
|
|
|
|
# ADMIN API
|
|
def test_shard_create(self):
|
|
with mock.patch.object(self.transport, 'send',
|
|
autospec=True) as send_method:
|
|
resp = response.Response(None, None)
|
|
send_method.return_value = resp
|
|
|
|
req = request.Request()
|
|
core.shard_create(self.transport, req,
|
|
'test_shard', {'uri': 'sqlite://',
|
|
'weight': 0})
|
|
|
|
def test_shard_delete(self):
|
|
with mock.patch.object(self.transport, 'send',
|
|
autospec=True) as send_method:
|
|
resp = response.Response(None, None)
|
|
send_method.return_value = resp
|
|
|
|
req = request.Request()
|
|
core.shard_delete(self.transport, req, 'test_shard')
|