166cfbfb1b
Fixed pep8 errors: * E127 continuation line over-indented for visual indent * E226 missing whitespace around arithmetic operator * E231 missing whitespace after ',' * E241 multiple spaces after ',' * E261 at least two spaces before inline comment * E301 expected 1 blank line, found 0 * E302 expected 2 blank lines, found 1 * E303 too many blank lines (2) * E501 line too long (80 > 79 characters) Fixed pep8 warnings: * W292 no newline at end of file * W391 blank line at end of file Change-Id: I1736f8a42c3a335a17a0b9b6e64782487ed3a495
120 lines
3.9 KiB
Python
120 lines
3.9 KiB
Python
# Copyright 2015 Mirantis, 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 logging
|
|
import pickle
|
|
import time
|
|
|
|
import contextlib
|
|
import fixtures
|
|
import testtools
|
|
|
|
|
|
import oslo_messaging
|
|
from oslo_messaging._drivers import impl_zmq
|
|
from oslo_messaging._drivers.zmq_driver import zmq_async
|
|
from oslo_messaging._drivers.zmq_driver import zmq_socket
|
|
from oslo_messaging._drivers.zmq_driver.client import zmq_request
|
|
from oslo_messaging._drivers.zmq_driver.client.publishers \
|
|
import zmq_pub_publisher
|
|
from oslo_messaging.tests import utils as test_utils
|
|
from oslo_messaging.tests.drivers.zmq import zmq_common
|
|
|
|
LOG = logging.getLogger(__name__)
|
|
|
|
zmq = zmq_async.import_zmq()
|
|
|
|
|
|
class TestPubSub(zmq_common.ZmqBaseTestCase):
|
|
|
|
LISTENERS_COUNT = 3
|
|
|
|
def setUp(self):
|
|
super(TestPubSub, self).setUp()
|
|
|
|
kwargs = {'use_pub_sub': True}
|
|
self.config(**kwargs)
|
|
|
|
self.publisher = zmq_pub_publisher.PubPublisherProxy(
|
|
self.conf, self.driver.matchmaker)
|
|
|
|
self.listeners = []
|
|
for i in range(self.LISTENERS_COUNT):
|
|
self.listeners.append(zmq_common.TestServerListener(self.driver))
|
|
|
|
def _send_request(self, target):
|
|
# Needed only in test env to get listener a chance to connect
|
|
# before request fires
|
|
time.sleep(1)
|
|
with contextlib.closing(zmq_request.FanoutRequest(
|
|
target, context={}, message={'method': 'hello-world'},
|
|
timeout=0, retry=None)) as request:
|
|
self.publisher.send_request([request.create_envelope(),
|
|
pickle.dumps(request)])
|
|
|
|
def _check_listener(self, listener):
|
|
listener._received.wait(timeout=5)
|
|
self.assertTrue(listener._received.isSet())
|
|
method = listener.message.message[u'method']
|
|
self.assertEqual(u'hello-world', method)
|
|
|
|
def _check_listener_negative(self, listener):
|
|
listener._received.wait(timeout=1)
|
|
self.assertFalse(listener._received.isSet())
|
|
|
|
def test_single_listener(self):
|
|
target = oslo_messaging.Target(topic='testtopic', fanout=True)
|
|
self.listener.listen(target)
|
|
|
|
self._send_request(target)
|
|
|
|
self._check_listener(self.listener)
|
|
|
|
def test_all_listeners(self):
|
|
target = oslo_messaging.Target(topic='testtopic', fanout=True)
|
|
|
|
for listener in self.listeners:
|
|
listener.listen(target)
|
|
|
|
self._send_request(target)
|
|
|
|
for listener in self.listeners:
|
|
self._check_listener(listener)
|
|
|
|
def test_filtered(self):
|
|
target = oslo_messaging.Target(topic='testtopic', fanout=True)
|
|
target_wrong = oslo_messaging.Target(topic='wrong', fanout=True)
|
|
|
|
self.listeners[0].listen(target)
|
|
self.listeners[1].listen(target)
|
|
self.listeners[2].listen(target_wrong)
|
|
|
|
self._send_request(target)
|
|
|
|
self._check_listener(self.listeners[0])
|
|
self._check_listener(self.listeners[1])
|
|
self._check_listener_negative(self.listeners[2])
|
|
|
|
def test_topic_part_matching(self):
|
|
target = oslo_messaging.Target(topic='testtopic', server='server')
|
|
target_part = oslo_messaging.Target(topic='testtopic', fanout=True)
|
|
|
|
self.listeners[0].listen(target)
|
|
self.listeners[1].listen(target)
|
|
|
|
self._send_request(target_part)
|
|
|
|
self._check_listener(self.listeners[0])
|
|
self._check_listener(self.listeners[1])
|