Skip to content

Commit bb32ddc

Browse files
committed
clean up storage rpc tests
1 parent 9e05a7a commit bb32ddc

File tree

1 file changed

+8
-15
lines changed

1 file changed

+8
-15
lines changed

services/storage/tests/unit/test_data_export.py

Lines changed: 8 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -15,9 +15,16 @@
1515
from models_library.api_schemas_rpc_async_jobs.async_jobs import (
1616
AsyncJobGet,
1717
AsyncJobId,
18+
AsyncJobNameData,
1819
AsyncJobResult,
1920
AsyncJobStatus,
2021
)
22+
from models_library.api_schemas_rpc_async_jobs.exceptions import (
23+
JobAbortedError,
24+
JobError,
25+
JobNotDoneError,
26+
JobSchedulerError,
27+
)
2128
from models_library.api_schemas_storage import STORAGE_RPC_NAMESPACE
2229
from models_library.api_schemas_storage.data_export_async_jobs import (
2330
DataExportTaskStartInput,
@@ -34,18 +41,10 @@
3441
from servicelib.rabbitmq.rpc_interfaces.async_jobs import async_jobs
3542
from servicelib.rabbitmq.rpc_interfaces.storage.data_export import start_data_export
3643
from settings_library.rabbit import RabbitSettings
37-
from simcore_service_storage.api.rpc._async_jobs import (
38-
AsyncJobNameData,
39-
JobAbortedError,
40-
JobError,
41-
JobNotDoneError,
42-
JobSchedulerError,
43-
TaskStatus,
44-
)
4544
from simcore_service_storage.api.rpc._data_export import AccessRightError
4645
from simcore_service_storage.core.settings import ApplicationSettings
4746
from simcore_service_storage.modules.celery.client import TaskUUID
48-
from simcore_service_storage.modules.celery.models import TaskState
47+
from simcore_service_storage.modules.celery.models import TaskState, TaskStatus
4948
from simcore_service_storage.simcore_s3_dsm import SimcoreS3DataManager
5049

5150
pytest_plugins = [
@@ -61,12 +60,6 @@
6160
_faker = Faker()
6261

6362

64-
@pytest.fixture
65-
async def mock_rabbit_setup(mocker: MockerFixture):
66-
# fixture to avoid mocking the rabbit
67-
pass
68-
69-
7063
@dataclass
7164
class _MockCeleryClient:
7265
send_task_object: UUID | Exception | None = None

0 commit comments

Comments
 (0)