mainsequence 8.1.1__tar.gz → 8.1.2__tar.gz

{mainsequence-8.1.1/mainsequence.egg-info → mainsequence-8.1.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mainsequence
-Version: 8.1.1
+Version: 8.1.2
 Summary: Main Sequence SDK 
 Author-email: Main Sequence GmbH <dev@main-sequence.io>
 License-Expression: LicenseRef-MainSequence-Proprietary

{mainsequence-8.1.1 → mainsequence-8.1.2}/mainsequence/client/models_helpers.py

@@ -1397,6 +1397,29 @@ class DeploymentRunPipeline(BaseModel):
     steps: list[DeploymentRunStep] = Field(default_factory=list)
 
 
+class DeploymentRunBillingComponents(BasePydanticModel):
+    image_build: Decimal | None = Field(max_digits=18, decimal_places=6)
+    image_registry_storage: Decimal | None = Field(max_digits=18, decimal_places=6)
+    image_registry_service: Decimal | None = Field(max_digits=18, decimal_places=6)
+
+
+class DeploymentRunBilling(BasePydanticModel):
+    scope: Literal["image_lifecycle"]
+    total_cost: Decimal | None = Field(max_digits=18, decimal_places=6)
+    currency: str = Field(min_length=1)
+    pricing_state: Literal[
+        "priced",
+        "partial",
+        "pending",
+        "unavailable",
+        "failed",
+    ]
+    components: DeploymentRunBillingComponents
+    priced_rows: int = Field(ge=0)
+    unpriced_rows: int = Field(ge=0)
+    reused_image_count: int = Field(ge=0)
+
+
 class DeploymentRunLogEntry(BaseModel):
     sequence: int
     timestamp: datetime.datetime | None = None
@@ -1473,6 +1496,7 @@ class DeploymentRun(CurrentCodeRepositoryBranchCollectionMixin, BaseObjectOrm, B
     builder_runtime: str = ""
     logs: DeploymentRunLogReference
     error: DeploymentRunError | None = None
+    billing: DeploymentRunBilling
 
     def get_logs(
         self,

{mainsequence-8.1.1 → mainsequence-8.1.2/mainsequence.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mainsequence
-Version: 8.1.1
+Version: 8.1.2
 Summary: Main Sequence SDK 
 Author-email: Main Sequence GmbH <dev@main-sequence.io>
 License-Expression: LicenseRef-MainSequence-Proprietary

{mainsequence-8.1.1 → mainsequence-8.1.2}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "mainsequence"
-version = "8.1.1"
+version = "8.1.2"
 description = "Main Sequence SDK "
 readme = "README.md"
 requires-python = ">=3.13"

{mainsequence-8.1.1 → mainsequence-8.1.2}/tests/test_filter_normalization.py

@@ -3,6 +3,7 @@ import datetime
 import json
 import pathlib
 import uuid
+from decimal import Decimal
 from types import SimpleNamespace
 
 import pytest
@@ -3145,6 +3146,83 @@ def _resource_release_pipeline_payload(*, running_step: str | None = None):
     }
 
 
+def _deployment_run_billing_payload(
+    *,
+    pricing_state: str = "priced",
+    total_cost: str | None = "0.000000",
+) -> dict:
+    return {
+        "scope": "image_lifecycle",
+        "total_cost": total_cost,
+        "currency": "USD",
+        "pricing_state": pricing_state,
+        "components": {
+            "image_build": total_cost,
+            "image_registry_storage": "0.000000" if total_cost is not None else None,
+            "image_registry_service": "0.000000" if total_cost is not None else None,
+        },
+        "priced_rows": 3 if pricing_state == "priced" else 0,
+        "unpriced_rows": 0 if pricing_state == "priced" else 3,
+        "reused_image_count": 0,
+    }
+
+
+@pytest.mark.parametrize(
+    "pricing_state",
+    ["priced", "partial", "pending", "unavailable", "failed"],
+)
+def test_deployment_run_billing_parses_every_backend_pricing_state(pricing_state):
+    total_cost = "1.250000" if pricing_state in {"priced", "partial"} else None
+
+    billing = models_helpers_mod.DeploymentRunBilling.model_validate(
+        _deployment_run_billing_payload(
+            pricing_state=pricing_state,
+            total_cost=total_cost,
+        )
+    )
+
+    assert billing.pricing_state == pricing_state
+    assert billing.total_cost == (Decimal(total_cost) if total_cost is not None else None)
+    assert billing.components.image_build == billing.total_cost
+
+
+def test_deployment_run_billing_contract_is_complete_and_closed():
+    payload = _deployment_run_billing_payload()
+    for field_name in (
+        "scope",
+        "total_cost",
+        "currency",
+        "pricing_state",
+        "components",
+        "priced_rows",
+        "unpriced_rows",
+        "reused_image_count",
+    ):
+        incomplete = dict(payload)
+        incomplete.pop(field_name)
+        with pytest.raises(ValidationError, match=field_name):
+            models_helpers_mod.DeploymentRunBilling.model_validate(incomplete)
+
+    with pytest.raises(ValidationError, match="extra_forbidden"):
+        models_helpers_mod.DeploymentRunBilling.model_validate(
+            payload | {"future_billing_field": "value"}
+        )
+
+    with pytest.raises(ValidationError, match="extra_forbidden"):
+        models_helpers_mod.DeploymentRunBilling.model_validate(
+            payload
+            | {
+                "components": payload["components"]
+                | {"future_component": "0.000000"}
+            }
+        )
+
+    with pytest.raises(ValidationError, match="greater_than_equal"):
+        models_helpers_mod.DeploymentRunBilling.model_validate(
+            payload | {"unpriced_rows": -1}
+        )
+
+
 def test_resource_release_deploy_current_version_posts_detail_action(monkeypatch):
     captured = {}
     release_uid = "2f4c4c3d-5669-4da5-9d86-b84633c1e6ed"
@@ -3189,6 +3267,7 @@ def test_resource_release_deploy_current_version_posts_detail_action(monkeypatch
                     "retention_expires_at": None,
                 },
                 "error": None,
+                "billing": _deployment_run_billing_payload(),
             }
 
     def _fake_make_request(*, s, loaders, r_type, url, payload, time_out=None):
@@ -3213,6 +3292,7 @@ def test_resource_release_deploy_current_version_posts_detail_action(monkeypatch
     assert run.builder_runtime == ""
     assert run.logs.state == "available"
     assert run.error is None
+    assert run.billing.total_cost == Decimal("0.000000")
     assert captured == {
         "r_type": "POST",
         "url": (
@@ -3256,6 +3336,7 @@ def test_deployment_run_collection_and_detail_use_unified_resource_release_contr
             "retention_expires_at": None,
         },
         "error": None,
+        "billing": _deployment_run_billing_payload(),
     }
 
     collection_payload = {
@@ -3311,6 +3392,8 @@ def test_deployment_run_collection_and_detail_use_unified_resource_release_contr
     assert detail.builder_runtime == ""
     assert detail.logs.state == "available"
     assert detail.error is None
+    assert runs[0].billing.pricing_state == "priced"
+    assert detail.billing.components.image_registry_storage == Decimal("0.000000")
     assert captured[0]["payload"] == {
         "params": {
             "code_repository_branch_uid": CODE_REPOSITORY_BRANCH_UID,
@@ -3350,6 +3433,10 @@ def test_unified_deployment_run_models_and_filters(monkeypatch):
                 "retention_expires_at": None,
             },
             "error": None,
+            "billing": _deployment_run_billing_payload(
+                pricing_state="pending",
+                total_cost=None,
+            ),
         }
     )
 
@@ -3367,6 +3454,8 @@ def test_unified_deployment_run_models_and_filters(monkeypatch):
     assert run.pipeline.steps[4].state == "pending"
     assert run.builder_image == "us-docker.pkg.dev/platform/static-builder:latest"
     assert run.builder_runtime == "nodejs22"
+    assert run.billing.pricing_state == "pending"
+    assert run.billing.total_cost is None
     assert normalized == {
         "code_repository_branch_uid": code_repository_branch_uid,
         "target_type__in": ["resource_release", "static_site"],