4f229832c0
The mock module has been added to Python 3.3 as unittest.mock.
The third party mock module doesn't seem to be maintained anymore.
This is follow up of 72c501454ea33b603918de35565a66dda8abd66a and
add hacking rule to avoid regression issue.
oslotest prepares mock for six in oslotest/__init__.py as follow:
six.add_move(six.MovedModule('mock', 'mock', 'unittest.mock')) and
oslo_messaging/tests/__init__.py imports oslotest before importing
test submodules to setup six.moves for mock, then "from six.moves
import mock" works well.
Note: unittest.mock also detects wrong usage of method
assert_called_once_with.
Change-Id: I3d09733789cfa2550cf47c24f2553357d36bcc0d
94 lines
3.2 KiB
Python
94 lines
3.2 KiB
Python
# 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
|
|
|
|
from oslo_messaging._drivers.zmq_driver.poller import green_poller
|
|
from oslo_messaging._drivers.zmq_driver.poller import threading_poller
|
|
from oslo_messaging._drivers.zmq_driver import zmq_async
|
|
from oslo_messaging.tests import utils as test_utils
|
|
|
|
zmq = zmq_async.import_zmq()
|
|
|
|
|
|
class TestImportZmq(test_utils.BaseTestCase):
|
|
|
|
@testtools.skipIf(zmq is None, "zmq not available")
|
|
def setUp(self):
|
|
super(TestImportZmq, self).setUp()
|
|
|
|
def test_when_eventlet_is_available_then_load_eventlet_green_zmq(self):
|
|
zmq_async.eventletutils.is_monkey_patched = lambda _: True
|
|
|
|
mock_try_import = mock.Mock()
|
|
zmq_async.importutils.try_import = mock_try_import
|
|
|
|
zmq_async.import_zmq()
|
|
|
|
mock_try_import.assert_called_with('eventlet.green.zmq', default=None)
|
|
|
|
def test_when_evetlet_is_unavailable_then_load_zmq(self):
|
|
zmq_async.eventletutils.is_monkey_patched = lambda _: False
|
|
|
|
mock_try_import = mock.Mock()
|
|
zmq_async.importutils.try_import = mock_try_import
|
|
|
|
zmq_async.import_zmq()
|
|
|
|
mock_try_import.assert_called_with('zmq', default=None)
|
|
|
|
|
|
class TestGetPoller(test_utils.BaseTestCase):
|
|
|
|
@testtools.skipIf(zmq is None, "zmq not available")
|
|
def setUp(self):
|
|
super(TestGetPoller, self).setUp()
|
|
|
|
def test_when_eventlet_is_available_then_return_GreenPoller(self):
|
|
zmq_async.eventletutils.is_monkey_patched = lambda _: True
|
|
|
|
poller = zmq_async.get_poller()
|
|
|
|
self.assertIsInstance(poller, green_poller.GreenPoller)
|
|
|
|
def test_when_eventlet_is_unavailable_then_return_ThreadingPoller(self):
|
|
zmq_async.eventletutils.is_monkey_patched = lambda _: False
|
|
|
|
poller = zmq_async.get_poller()
|
|
|
|
self.assertIsInstance(poller, threading_poller.ThreadingPoller)
|
|
|
|
|
|
class TestGetExecutor(test_utils.BaseTestCase):
|
|
|
|
@testtools.skipIf(zmq is None, "zmq not available")
|
|
def setUp(self):
|
|
super(TestGetExecutor, self).setUp()
|
|
|
|
def test_when_eventlet_module_is_available_then_return_GreenExecutor(self):
|
|
zmq_async.eventletutils.is_monkey_patched = lambda _: True
|
|
|
|
executor = zmq_async.get_executor('any method')
|
|
|
|
self.assertIsInstance(executor, green_poller.GreenExecutor)
|
|
self.assertEqual('any method', executor._method)
|
|
|
|
def test_when_eventlet_is_unavailable_then_return_ThreadingExecutor(self):
|
|
zmq_async.eventletutils.is_monkey_patched = lambda _: False
|
|
|
|
executor = zmq_async.get_executor('any method')
|
|
|
|
self.assertIsInstance(executor,
|
|
threading_poller.ThreadingExecutor)
|
|
self.assertEqual('any method', executor._method)
|