|
22 | 22 | from nova import context |
23 | 23 | from nova import rpc |
24 | 24 | from nova import test |
| 25 | +from nova import utils |
25 | 26 |
|
26 | 27 | CONF = nova.conf.CONF |
27 | 28 |
|
@@ -245,31 +246,9 @@ def test_get_server(self, mock_get, mock_ser, mock_TRANSPORT): |
245 | 246 |
|
246 | 247 | mock_ser.assert_called_once_with('foo') |
247 | 248 | access_policy = dispatcher.DefaultRPCAccessPolicy |
| 249 | + exc = 'threading' if utils.concurrency_mode_threading() else 'eventlet' |
248 | 250 | mock_get.assert_called_once_with(mock_TRANSPORT, tgt, ends, |
249 | | - executor='eventlet', serializer=ser, |
250 | | - access_policy=access_policy) |
251 | | - self.assertEqual('server', server) |
252 | | - |
253 | | - @mock.patch( |
254 | | - 'nova.utils.concurrency_mode_threading', |
255 | | - new=mock.Mock(return_value=True)) |
256 | | - @mock.patch.object(rpc, 'TRANSPORT') |
257 | | - @mock.patch.object(rpc, 'profiler', None) |
258 | | - @mock.patch.object(rpc, 'RequestContextSerializer') |
259 | | - @mock.patch.object(messaging, 'get_rpc_server') |
260 | | - def test_get_server_threading(self, mock_get, mock_ser, mock_TRANSPORT): |
261 | | - ser = mock.Mock() |
262 | | - tgt = mock.Mock() |
263 | | - ends = mock.Mock() |
264 | | - mock_ser.return_value = ser |
265 | | - mock_get.return_value = 'server' |
266 | | - |
267 | | - server = rpc.get_server(tgt, ends, serializer='foo') |
268 | | - |
269 | | - mock_ser.assert_called_once_with('foo') |
270 | | - access_policy = dispatcher.DefaultRPCAccessPolicy |
271 | | - mock_get.assert_called_once_with(mock_TRANSPORT, tgt, ends, |
272 | | - executor='threading', serializer=ser, |
| 251 | + executor=exc, serializer=ser, |
273 | 252 | access_policy=access_policy) |
274 | 253 | self.assertEqual('server', server) |
275 | 254 |
|
|
0 commit comments