|
29 | 29 | UpdateDeploymentPayload, |
30 | 30 | ) |
31 | 31 | from ai.backend.common.dto.manager.v2.deployment.types import ( |
32 | | - BlueGreenConfigInfo, |
33 | 32 | ClusterConfigInfoDTO, |
34 | 33 | DeploymentMetadataInfoDTO, |
35 | 34 | DeploymentNetworkAccessInfoDTO, |
36 | | - DeploymentPolicyInfo, |
37 | 35 | DeploymentStrategyInfoDTO, |
38 | 36 | ExtraVFolderMountGQLDTO, |
39 | | - IntOrPercent, |
40 | 37 | ModelMountConfigInfoDTO, |
41 | 38 | ModelRuntimeConfigInfoDTO, |
42 | 39 | ReplicaStateInfo, |
43 | 40 | ResourceConfigInfoDTO, |
44 | | - RollingUpdateConfigInfo, |
45 | 41 | ) |
46 | 42 | from ai.backend.common.dto.manager.v2.deployment_options.response import ( |
47 | 43 | DeploymentHandlerOptionsInfo, |
@@ -169,7 +165,6 @@ def _make_deployment_node(**kwargs: object) -> DeploymentNode: |
169 | 165 | "options": _make_deployment_options(), |
170 | 166 | "scaling_state": ScalingState.STABLE, |
171 | 167 | "current_revision_id": None, |
172 | | - "policy": None, |
173 | 168 | } |
174 | 169 | defaults.update(kwargs) |
175 | 170 | return DeploymentNode(**defaults) |
@@ -316,48 +311,17 @@ def test_creation_with_required_fields(self) -> None: |
316 | 311 | ) |
317 | 312 | assert node.id == deployment_id |
318 | 313 | assert node.current_revision_id is None |
319 | | - assert node.policy is None |
320 | 314 |
|
321 | 315 | def test_current_revision_id_defaults_to_none(self) -> None: |
322 | 316 | node = _make_deployment_node() |
323 | 317 | assert node.current_revision_id is None |
324 | 318 |
|
325 | | - def test_policy_defaults_to_none(self) -> None: |
326 | | - node = _make_deployment_node() |
327 | | - assert node.policy is None |
328 | | - |
329 | 319 | def test_with_current_revision_id(self) -> None: |
330 | 320 | revision_id = uuid.uuid4() |
331 | 321 | node = _make_deployment_node(current_revision_id=revision_id) |
332 | 322 | assert node.current_revision_id is not None |
333 | 323 | assert node.current_revision_id == revision_id |
334 | 324 |
|
335 | | - def test_with_rolling_policy(self) -> None: |
336 | | - rolling = RollingUpdateConfigInfo( |
337 | | - max_surge=IntOrPercent(count=1), |
338 | | - max_unavailable=IntOrPercent(count=0), |
339 | | - ) |
340 | | - policy = DeploymentPolicyInfo( |
341 | | - strategy=DeploymentStrategy.ROLLING, |
342 | | - rolling_update=rolling, |
343 | | - blue_green=None, |
344 | | - ) |
345 | | - node = _make_deployment_node(policy=policy) |
346 | | - assert node.policy is not None |
347 | | - assert node.policy.strategy == DeploymentStrategy.ROLLING |
348 | | - assert node.policy.rolling_update is not None |
349 | | - |
350 | | - def test_with_blue_green_policy(self) -> None: |
351 | | - bg = BlueGreenConfigInfo(auto_promote=False, promote_delay_seconds=0) |
352 | | - policy = DeploymentPolicyInfo( |
353 | | - strategy=DeploymentStrategy.BLUE_GREEN, |
354 | | - rolling_update=None, |
355 | | - blue_green=bg, |
356 | | - ) |
357 | | - node = _make_deployment_node(policy=policy) |
358 | | - assert node.policy is not None |
359 | | - assert node.policy.strategy == DeploymentStrategy.BLUE_GREEN |
360 | | - |
361 | 325 | def test_metadata_accessible(self) -> None: |
362 | 326 | project_id = str(uuid.uuid4()) |
363 | 327 | metadata = _make_deployment_metadata(project_id=project_id, name="my-deploy") |
|
0 commit comments