Skip to content

Commit 37192b6

Browse files
author
Andrei Neagu
committed
fixed test
1 parent b1d46bc commit 37192b6

File tree

1 file changed

+7
-3
lines changed

1 file changed

+7
-3
lines changed

services/web/server/tests/unit/with_dbs/01/test_long_running_tasks.py

Lines changed: 7 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -6,6 +6,7 @@
66
# pylint: disable=no-self-argument
77

88
from typing import Any
9+
from unittest.mock import Mock
910

1011
import pytest
1112
from aiohttp.test_utils import TestClient
@@ -77,11 +78,14 @@ async def test_listing_tasks_with_list_inprocess_tasks_error(
7778

7879
class _DummyTaskManager:
7980
def list_tasks(self, *args, **kwargs):
80-
raise Exception() # pylint: disable=broad-exception-raised
81+
raise Exception # pylint: disable=broad-exception-raised # noqa: TRY002
82+
83+
mock = Mock()
84+
mock.tasks_manager = _DummyTaskManager()
8185

8286
mocker.patch(
83-
"servicelib.aiohttp.long_running_tasks._routes.get_tasks_manager",
84-
return_value=_DummyTaskManager(),
87+
"servicelib.aiohttp.long_running_tasks._routes.get_long_running_manager",
88+
return_value=mock,
8589
)
8690

8791
_async_jobs_listing_path = client.app.router["get_async_jobs"].url_for()

0 commit comments

Comments
 (0)