Skip to content

Commit c9a93f7

Browse files
author
Andrei Neagu
committed
refactor
1 parent b50d10f commit c9a93f7

File tree

2 files changed

+8
-8
lines changed
  • packages/service-library/src/servicelib/rabbitmq/rpc_interfaces/async_jobs
  • services/web/server/tests/unit/with_dbs/01/storage

2 files changed

+8
-8
lines changed

packages/service-library/src/servicelib/rabbitmq/rpc_interfaces/async_jobs/async_jobs.py

Lines changed: 6 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -31,9 +31,8 @@
3131
from ... import RabbitMQRPCClient
3232

3333
_DEFAULT_TIMEOUT_S: Final[NonNegativeInt] = 30
34-
35-
_RPC_METHOD_NAME_ADAPTER = TypeAdapter(RPCMethodName)
3634
_DEFAULT_POLL_INTERVAL_S: Final[float] = 0.1
35+
3736
_logger = logging.getLogger(__name__)
3837

3938

@@ -46,7 +45,7 @@ async def cancel(
4645
) -> None:
4746
await rabbitmq_rpc_client.request(
4847
rpc_namespace,
49-
_RPC_METHOD_NAME_ADAPTER.validate_python("cancel"),
48+
TypeAdapter(RPCMethodName).validate_python("cancel"),
5049
job_id=job_id,
5150
job_id_data=job_id_data,
5251
timeout_s=_DEFAULT_TIMEOUT_S,
@@ -62,7 +61,7 @@ async def status(
6261
) -> AsyncJobStatus:
6362
_result = await rabbitmq_rpc_client.request(
6463
rpc_namespace,
65-
_RPC_METHOD_NAME_ADAPTER.validate_python("status"),
64+
TypeAdapter(RPCMethodName).validate_python("status"),
6665
job_id=job_id,
6766
job_id_data=job_id_data,
6867
timeout_s=_DEFAULT_TIMEOUT_S,
@@ -80,7 +79,7 @@ async def result(
8079
) -> AsyncJobResult:
8180
_result = await rabbitmq_rpc_client.request(
8281
rpc_namespace,
83-
_RPC_METHOD_NAME_ADAPTER.validate_python("result"),
82+
TypeAdapter(RPCMethodName).validate_python("result"),
8483
job_id=job_id,
8584
job_id_data=job_id_data,
8685
timeout_s=_DEFAULT_TIMEOUT_S,
@@ -98,7 +97,7 @@ async def list_jobs(
9897
) -> list[AsyncJobGet]:
9998
_result: list[AsyncJobGet] = await rabbitmq_rpc_client.request(
10099
rpc_namespace,
101-
_RPC_METHOD_NAME_ADAPTER.validate_python("list_jobs"),
100+
TypeAdapter(RPCMethodName).validate_python("list_jobs"),
102101
filter_=filter_,
103102
job_id_data=job_id_data,
104103
timeout_s=_DEFAULT_TIMEOUT_S,
@@ -116,7 +115,7 @@ async def submit(
116115
) -> AsyncJobGet:
117116
_result = await rabbitmq_rpc_client.request(
118117
rpc_namespace,
119-
_RPC_METHOD_NAME_ADAPTER.validate_python(method_name),
118+
TypeAdapter(RPCMethodName).validate_python(method_name),
120119
job_id_data=job_id_data,
121120
**kwargs,
122121
timeout_s=_DEFAULT_TIMEOUT_S,

services/web/server/tests/unit/with_dbs/01/storage/test_storage.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -46,6 +46,7 @@
4646
from models_library.api_schemas_webserver.storage import (
4747
BatchDeletePathsBodyParams,
4848
DataExportPost,
49+
PathToExport,
4950
)
5051
from models_library.generics import Envelope
5152
from models_library.progress_bar import ProgressReport
@@ -671,7 +672,7 @@ async def test_get_async_job_links(
671672
)
672673

673674
_body = DataExportPost(
674-
paths=[f"{faker.uuid4()}/{faker.uuid4()}/{faker.file_name()}"]
675+
paths=[PathToExport(f"{faker.uuid4()}/{faker.uuid4()}/{faker.file_name()}")]
675676
)
676677
response = await client.post(
677678
f"/{API_VERSION}/storage/locations/0/export-data", data=_body.model_dump_json()

0 commit comments

Comments
 (0)