313f175471
Zaqarclient v2 currently has missing features that involve the management of subscriptions, but in order to these to work it is necessary to add other features such as 'queues', 'messages', 'claims', 'pools' and 'flavors'. I added 'queues' and 'messages' as a priorital step before start working with the other features and with subscriptions as well. Change-Id: Iea082cbf9e1d0e4a745bc10e41e1206af6132fad
110 lines
4.1 KiB
Python
110 lines
4.1 KiB
Python
# Copyright (c) 2013 Rackspace, 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 tests.unit.queues.v1 import test_message as msg
|
|
from zaqarclient.queues.v1 import iterator as iterate
|
|
from zaqarclient.queues.v1 import message
|
|
from zaqarclient.transport import http
|
|
from zaqarclient.transport import response
|
|
|
|
|
|
class TestMessageIterator(msg.TestMessageIterator):
|
|
def test_no_next_iteration(self):
|
|
messages = {'links': [],
|
|
'messages': [{
|
|
'href': '/v2/queues/mine/messages/123123423',
|
|
'ttl': 800,
|
|
'age': 790,
|
|
'body': {'event': 'ActivateAccount',
|
|
'mode': 'active'}
|
|
}]
|
|
}
|
|
|
|
iterator = iterate._Iterator(self.queue.client,
|
|
messages,
|
|
'messages',
|
|
message.create_object(self.queue))
|
|
iterated = [msg for msg in iterator]
|
|
self.assertEqual(len(iterated), 1)
|
|
|
|
def test_stream(self):
|
|
messages = {'links': [],
|
|
'messages': [{
|
|
'href': '/v2/queues/mine/messages/123123423',
|
|
'ttl': 800,
|
|
'age': 790,
|
|
'body': {'event': 'ActivateAccount',
|
|
'mode': 'active'}
|
|
}]
|
|
}
|
|
|
|
with mock.patch.object(self.transport, 'send',
|
|
autospec=True) as send_method:
|
|
|
|
resp = response.Response(None, json.dumps(messages))
|
|
send_method.return_value = resp
|
|
|
|
# NOTE(flaper87): The first iteration will return 1 message
|
|
# and then call `_next_page` which will use the rel-next link
|
|
# to get a new set of messages.
|
|
link = {'rel': 'next',
|
|
'href': "/v2/queues/mine/messages?marker=6244-244224-783"}
|
|
messages['links'].append(link)
|
|
|
|
iterator = iterate._Iterator(self.queue.client,
|
|
messages,
|
|
'messages',
|
|
message.create_object(self.queue))
|
|
iterated = [msg for msg in iterator.stream()]
|
|
self.assertEqual(len(iterated), 2)
|
|
|
|
def test_iterator_respect_paging(self):
|
|
messages = {'links': [],
|
|
'messages': [{
|
|
'href': '/v2/queues/mine/messages/123123423',
|
|
'ttl': 800,
|
|
'age': 790,
|
|
'body': {'event': 'ActivateAccount',
|
|
'mode': 'active'}
|
|
}]
|
|
}
|
|
|
|
with mock.patch.object(self.transport, 'send',
|
|
autospec=True) as send_method:
|
|
|
|
resp = response.Response(None, json.dumps(messages))
|
|
send_method.return_value = resp
|
|
|
|
link = {'rel': 'next',
|
|
'href': "/v2/queues/mine/messages?marker=6244-244224-783"}
|
|
messages['links'].append(link)
|
|
|
|
iterator = iterate._Iterator(self.queue.client,
|
|
messages,
|
|
'messages',
|
|
message.create_object(self.queue))
|
|
iterated = [msg for msg in iterator]
|
|
self.assertEqual(len(iterated), 1)
|
|
|
|
|
|
class QueuesV2MessageHttpUnitTest(msg.QueuesV1MessageHttpUnitTest):
|
|
|
|
transport_cls = http.HttpTransport
|
|
url = 'http://127.0.0.1:8888/v2'
|
|
version = 2
|