Skip to content

Commit 87112f6

Browse files
fix fields
1 parent 736064d commit 87112f6

File tree

3 files changed

+22
-24
lines changed

3 files changed

+22
-24
lines changed

packages/models-library/src/models_library/aiodocker_api.py

Lines changed: 9 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -11,12 +11,13 @@
1111

1212

1313
class AioDockerContainerSpec(ContainerSpec):
14-
Env: dict[str, str | None] | None = Field(
14+
env: dict[str, str | None] | None = Field( # type: ignore[assignment]
1515
default=None,
16-
description="aiodocker expects here a dictionary and re-convert it back internally`.\n",
16+
alias="Env",
17+
description="aiodocker expects here a dictionary and re-convert it back internally",
1718
)
1819

19-
@field_validator("Env", mode="before")
20+
@field_validator("env", mode="before")
2021
@classmethod
2122
def convert_list_to_dict(cls, v):
2223
if v is not None and isinstance(v, list):
@@ -33,25 +34,22 @@ def convert_list_to_dict(cls, v):
3334
class AioDockerResources1(Resources1):
3435
# NOTE: The Docker REST API documentation is wrong!!!
3536
# Do not set that back to singular Reservation.
36-
Reservation: ResourceObject | None = Field(
37+
reservation: ResourceObject | None = Field(
3738
None, description="Define resources reservation.", alias="Reservations"
3839
)
3940

4041
model_config = ConfigDict(populate_by_name=True)
4142

4243

4344
class AioDockerTaskSpec(TaskSpec):
44-
ContainerSpec: AioDockerContainerSpec | None = Field(
45-
None,
45+
container_spec: AioDockerContainerSpec | None = Field(
46+
default=None, alias="ContainerSpec"
4647
)
4748

48-
Resources: AioDockerResources1 | None = Field(
49-
None,
50-
description="Resource requirements which apply to each individual container created\nas part of the service.\n",
51-
)
49+
resources: AioDockerResources1 | None = Field(default=None, alias="Resources")
5250

5351

5452
class AioDockerServiceSpec(ServiceSpec):
55-
TaskTemplate: AioDockerTaskSpec | None = None
53+
task_template: AioDockerTaskSpec | None = Field(default=None, alias="TaskTemplate")
5654

5755
model_config = ConfigDict(populate_by_name=True, alias_generator=camel_to_snake)

services/director-v2/tests/unit/with_dbs/test_api_route_clusters.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -745,7 +745,7 @@ async def test_ping_cluster(
745745
)
746746
response = await async_client.post(
747747
"/v2/clusters:ping",
748-
json=model_dump_with_secrets(valid_cluster, show_secrets=True, alias=True),
748+
json=model_dump_with_secrets(valid_cluster, show_secrets=True, by_alias=True),
749749
)
750750
response.raise_for_status()
751751
assert response.status_code == status.HTTP_204_NO_CONTENT

services/director-v2/tests/unit/with_dbs/test_modules_dynamic_sidecar_docker_service_specs.py

Lines changed: 12 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -445,9 +445,9 @@ async def test_get_dynamic_proxy_spec(
445445
expected_dynamic_sidecar_spec_model = AioDockerServiceSpec.model_validate(
446446
expected_dynamic_sidecar_spec
447447
)
448-
assert expected_dynamic_sidecar_spec_model.TaskTemplate
449-
assert expected_dynamic_sidecar_spec_model.TaskTemplate.ContainerSpec
450-
assert expected_dynamic_sidecar_spec_model.TaskTemplate.ContainerSpec.Env
448+
assert expected_dynamic_sidecar_spec_model.task_template
449+
assert expected_dynamic_sidecar_spec_model.task_template.container_spec
450+
assert expected_dynamic_sidecar_spec_model.task_template.container_spec.env
451451

452452
for count in range(1, 11): # loop to check it does not repeat copies
453453
print(f"{count:*^50}")
@@ -478,33 +478,33 @@ async def test_get_dynamic_proxy_spec(
478478
# NOTE: some flakiness here
479479
# state_exclude is a set and does not preserve order
480480
# when dumping to json it gets converted to a list
481-
assert dynamic_sidecar_spec.TaskTemplate
482-
assert dynamic_sidecar_spec.TaskTemplate.ContainerSpec
483-
assert dynamic_sidecar_spec.TaskTemplate.ContainerSpec.Env
484-
assert dynamic_sidecar_spec.TaskTemplate.ContainerSpec.Env[
481+
assert dynamic_sidecar_spec.task_template
482+
assert dynamic_sidecar_spec.task_template.container_spec
483+
assert dynamic_sidecar_spec.task_template.container_spec.env
484+
assert dynamic_sidecar_spec.task_template.container_spec.env[
485485
"DY_SIDECAR_STATE_EXCLUDE"
486486
]
487487

488-
dynamic_sidecar_spec.TaskTemplate.ContainerSpec.Env[
488+
dynamic_sidecar_spec.task_template.container_spec.env[
489489
"DY_SIDECAR_STATE_EXCLUDE"
490490
] = json.dumps(
491491
sorted(
492492
json.loads(
493-
dynamic_sidecar_spec.TaskTemplate.ContainerSpec.Env[
493+
dynamic_sidecar_spec.task_template.container_spec.env[
494494
"DY_SIDECAR_STATE_EXCLUDE"
495495
]
496496
)
497497
)
498498
)
499-
assert expected_dynamic_sidecar_spec_model.TaskTemplate.ContainerSpec.Env[
499+
assert expected_dynamic_sidecar_spec_model.task_template.container_spec.env[
500500
"DY_SIDECAR_STATE_EXCLUDE"
501501
]
502-
expected_dynamic_sidecar_spec_model.TaskTemplate.ContainerSpec.Env[
502+
expected_dynamic_sidecar_spec_model.task_template.container_spec.env[
503503
"DY_SIDECAR_STATE_EXCLUDE"
504504
] = json.dumps(
505505
sorted(
506506
json.loads(
507-
expected_dynamic_sidecar_spec_model.TaskTemplate.ContainerSpec.Env[
507+
expected_dynamic_sidecar_spec_model.task_template.container_spec.env[
508508
"DY_SIDECAR_STATE_EXCLUDE"
509509
]
510510
)

0 commit comments

Comments
 (0)