/usr/lib/python2.7/dist-packages/oslo_messaging/tests/drivers/zmq/test_zmq_transport_url.py is in python-oslo.messaging 5.35.0-0ubuntu1.
This file is owned by root:root, with mode 0o644.
The actual contents of the file can be viewed below.
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 | # Copyright 2016 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.
from six.moves import mock
import testtools
import oslo_messaging
from oslo_messaging._drivers import common
from oslo_messaging._drivers.zmq_driver.matchmaker.zmq_matchmaker_base \
import MatchmakerDummy
from oslo_messaging._drivers.zmq_driver.matchmaker import zmq_matchmaker_redis
from oslo_messaging._drivers.zmq_driver import zmq_async
from oslo_messaging.tests import utils as test_utils
zmq = zmq_async.import_zmq()
redis = zmq_matchmaker_redis.redis
sentinel = zmq_matchmaker_redis.redis_sentinel
class TestZmqTransportUrl(test_utils.BaseTestCase):
@testtools.skipIf(zmq is None, "zmq not available")
def setUp(self):
super(TestZmqTransportUrl, self).setUp()
def setup_url(self, url):
transport = oslo_messaging.get_transport(self.conf, url)
self.addCleanup(transport.cleanup)
driver = transport._driver
return driver, url
def mock_redis(self):
if redis is None:
self.skipTest("redis not available")
else:
redis_patcher = mock.patch.object(redis, 'StrictRedis')
self.addCleanup(redis_patcher.stop)
return redis_patcher.start()
def mock_sentinel(self):
if sentinel is None:
self.skipTest("sentinel not available")
else:
sentinel_patcher = mock.patch.object(sentinel, 'Sentinel')
self.addCleanup(sentinel_patcher.stop)
return sentinel_patcher.start()
def test_empty_url(self):
self.mock_redis()
driver, url = self.setup_url("zmq:///")
self.assertIs(zmq_matchmaker_redis.MatchmakerRedis,
driver.matchmaker.__class__)
self.assertEqual('zmq', driver.matchmaker.url.transport)
def test_error_url(self):
self.assertRaises(common.RPCException, self.setup_url, "zmq+error:///")
def test_dummy_url(self):
driver, url = self.setup_url("zmq+dummy:///")
self.assertIs(MatchmakerDummy,
driver.matchmaker.__class__)
self.assertEqual('zmq+dummy', driver.matchmaker.url.transport)
def test_redis_url(self):
self.mock_redis()
driver, url = self.setup_url("zmq+redis:///")
self.assertIs(zmq_matchmaker_redis.MatchmakerRedis,
driver.matchmaker.__class__)
self.assertEqual('zmq+redis', driver.matchmaker.url.transport)
def test_sentinel_url(self):
self.mock_sentinel()
driver, url = self.setup_url("zmq+sentinel:///")
self.assertIs(zmq_matchmaker_redis.MatchmakerSentinel,
driver.matchmaker.__class__)
self.assertEqual('zmq+sentinel', driver.matchmaker.url.transport)
def test_host_with_credentials_url(self):
self.mock_redis()
driver, url = self.setup_url("zmq://:password@host:60000/")
self.assertIs(zmq_matchmaker_redis.MatchmakerRedis,
driver.matchmaker.__class__)
self.assertEqual('zmq', driver.matchmaker.url.transport)
self.assertEqual(
[{"host": "host", "port": 60000, "password": "password"}],
driver.matchmaker._redis_hosts
)
def test_redis_multiple_hosts_url(self):
self.mock_redis()
driver, url = self.setup_url(
"zmq+redis://host1:60001,host2:60002,host3:60003/"
)
self.assertIs(zmq_matchmaker_redis.MatchmakerRedis,
driver.matchmaker.__class__)
self.assertEqual('zmq+redis', driver.matchmaker.url.transport)
self.assertEqual(
[{"host": "host1", "port": 60001, "password": None},
{"host": "host2", "port": 60002, "password": None},
{"host": "host3", "port": 60003, "password": None}],
driver.matchmaker._redis_hosts
)
def test_sentinel_multiple_hosts_url(self):
self.mock_sentinel()
driver, url = self.setup_url(
"zmq+sentinel://host1:20001,host2:20002,host3:20003/"
)
self.assertIs(zmq_matchmaker_redis.MatchmakerSentinel,
driver.matchmaker.__class__)
self.assertEqual('zmq+sentinel', driver.matchmaker.url.transport)
self.assertEqual(
[("host1", 20001), ("host2", 20002), ("host3", 20003)],
driver.matchmaker._sentinel_hosts
)
|