|
|
@@ -19,14 +19,17 @@ class ApiTestCase(test_base.CoriolisBaseTestCase):
|
|
|
@mock.patch('coriolis.cmd.api.CONF')
|
|
|
@mock.patch.object(service, 'get_worker_count_from_args')
|
|
|
@mock.patch.object(sys, 'argv')
|
|
|
+ @mock.patch(
|
|
|
+ 'oslo_reports.guru_meditation_report.TextGuruMeditation.setup_autorun')
|
|
|
def test_main(
|
|
|
self,
|
|
|
+ mock_gmr_setup,
|
|
|
mock_argv,
|
|
|
mock_get_worker_count_from_args,
|
|
|
mock_conf,
|
|
|
mock_setup_logging,
|
|
|
mock_WSGIService,
|
|
|
- mock_service
|
|
|
+ mock_service,
|
|
|
):
|
|
|
worker_count = mock.sentinel.worker_count
|
|
|
args = ['mock_arg_1', 'mock_arg_2']
|
|
|
@@ -45,6 +48,8 @@ class ApiTestCase(test_base.CoriolisBaseTestCase):
|
|
|
workers=mock_WSGIService.return_value.
|
|
|
get_workers_count.return_value)
|
|
|
mock_service.launch.return_value.wait.assert_called_once()
|
|
|
+ mock_gmr_setup.assert_called_once_with(
|
|
|
+ version="1.0.0", conf=mock_conf)
|
|
|
|
|
|
@mock.patch.object(service, 'service')
|
|
|
@mock.patch.object(service, 'WSGIService')
|
|
|
@@ -52,8 +57,11 @@ class ApiTestCase(test_base.CoriolisBaseTestCase):
|
|
|
@mock.patch('coriolis.cmd.api.CONF')
|
|
|
@mock.patch.object(service, 'get_worker_count_from_args')
|
|
|
@mock.patch.object(sys, 'argv')
|
|
|
+ @mock.patch(
|
|
|
+ 'oslo_reports.guru_meditation_report.TextGuruMeditation.setup_autorun')
|
|
|
def test_main_no_worker_count(
|
|
|
self,
|
|
|
+ mock_gmr_setup,
|
|
|
mock_argv,
|
|
|
mock_get_worker_count_from_args,
|
|
|
mock_conf,
|
|
|
@@ -78,3 +86,5 @@ class ApiTestCase(test_base.CoriolisBaseTestCase):
|
|
|
workers=mock_WSGIService.return_value.
|
|
|
get_workers_count.return_value)
|
|
|
mock_service.launch.return_value.wait.assert_called_once()
|
|
|
+ mock_gmr_setup.assert_called_once_with(
|
|
|
+ version="1.0.0", conf=mock_conf)
|