import 15.2.0 Octopus source

[ceph.git] / ceph / src / pybind / mgr / dashboard / tests / test_rest_tasks.py

diff --git a/ceph/src/pybind/mgr/dashboard/tests/test_rest_tasks.py b/ceph/src/pybind/mgr/dashboard/tests/test_rest_tasks.py
index 191ef812934fb2970a8959cf1398f91ecc403fdd..e9d7907f0524bdc246bd8392908fe761398652df 100644 (file)
--- a/ceph/src/pybind/mgr/dashboard/tests/test_rest_tasks.py
+++ b/ceph/src/pybind/mgr/dashboard/tests/test_rest_tasks.py
@@ -3,9 +3,15 @@
 
 import time
 
+try:
+    import mock
+except ImportError:
+    import unittest.mock as mock
+
 from . import ControllerTestCase
 from ..controllers import Controller, RESTController, Task
 from ..controllers.task import Task as TaskController
+from ..services import progress
 from ..tools import NotificationQueue, TaskManager
 
 
@@ -48,6 +54,9 @@ class TaskControllerTest(ControllerTestCase):
     @classmethod
     def setup_server(cls):
         # pylint: disable=protected-access
+        progress.get_progress_tasks = mock.MagicMock()
+        progress.get_progress_tasks.return_value = ([], [])
+
         NotificationQueue.start_queue()
         TaskManager.init()
         TaskTest._cp_config['tools.authenticate.on'] = False