karrio-server-graph 2025.5rc11__py3-none-any.whl → 2025.5rc13__py3-none-any.whl

karrio/server/graph/schemas/base/__init__.py

@@ -332,6 +332,24 @@ class Mutation:
     ) -> mutations.UpdateRateSheetMutation:
         return mutations.UpdateRateSheetMutation.mutate(info, **input.to_dict())
 
+    @strawberry.mutation
+    def update_rate_sheet_zone_cell(
+        self, info: Info, input: inputs.UpdateRateSheetZoneCellMutationInput
+    ) -> mutations.UpdateRateSheetZoneCellMutation:
+        return mutations.UpdateRateSheetZoneCellMutation.mutate(info, **input.to_dict())
+
+    @strawberry.mutation
+    def batch_update_rate_sheet_cells(
+        self, info: Info, input: inputs.BatchUpdateRateSheetCellsMutationInput
+    ) -> mutations.BatchUpdateRateSheetCellsMutation:
+        return mutations.BatchUpdateRateSheetCellsMutation.mutate(info, **input.to_dict())
+
+    @strawberry.mutation
+    def delete_rate_sheet_service(
+        self, info: Info, input: inputs.DeleteRateSheetServiceMutationInput
+    ) -> mutations.DeleteRateSheetServiceMutation:
+        return mutations.DeleteRateSheetServiceMutation.mutate(info, **input.to_dict())
+
     @strawberry.mutation
     def delete_rate_sheet(
         self, info: Info, input: inputs.DeleteMutationInput

karrio/server/graph/schemas/base/inputs.py

@@ -245,7 +245,6 @@ class AddressInput:
     email: typing.Optional[str] = strawberry.UNSET
     phone_number: typing.Optional[str] = strawberry.UNSET
     state_code: typing.Optional[str] = strawberry.UNSET
-    suburb: typing.Optional[str] = strawberry.UNSET
     residential: typing.Optional[bool] = strawberry.UNSET
     street_number: typing.Optional[str] = strawberry.UNSET
     address_line1: typing.Optional[str] = strawberry.UNSET
@@ -517,6 +516,35 @@ class UpdateRateSheetMutationInput(utils.BaseInput):
     metadata: typing.Optional[utils.JSON] = strawberry.UNSET
 
 
+@strawberry.input
+class UpdateRateSheetZoneCellMutationInput(utils.BaseInput):
+    id: str  # Rate sheet ID
+    service_id: str  # Service level ID
+    zone_id: str  # Zone ID
+    field: str  # Field name to update
+    value: utils.JSON  # New value
+
+
+@strawberry.input
+class CellUpdate(utils.BaseInput):
+    service_id: str
+    zone_id: str
+    field: str
+    value: utils.JSON
+
+
+@strawberry.input
+class BatchUpdateRateSheetCellsMutationInput(utils.BaseInput):
+    id: str  # Rate sheet ID
+    updates: typing.List[CellUpdate]
+
+
+@strawberry.input
+class DeleteRateSheetServiceMutationInput(utils.BaseInput):
+    rate_sheet_id: str
+    service_id: str
+
+
 @strawberry.input
 class RateSheetFilter(utils.Paginated):
     keyword: typing.Optional[str] = strawberry.UNSET

karrio/server/graph/schemas/base/mutations.py

@@ -538,8 +538,8 @@ class CreateRateSheetMutation(utils.BaseMutation):
                 carrier_name=rate_sheet.carrier_name,
             ).filter(id__in=carriers)
             for _ in _carriers:
-                _.settings.rate_sheet = rate_sheet
-                _.settings.save(update_fields=["rate_sheet"])
+                _.rate_sheet = rate_sheet
+                _.save(update_fields=["rate_sheet"])
 
         return CreateRateSheetMutation(rate_sheet=rate_sheet)
 
@@ -558,9 +558,10 @@ class UpdateRateSheetMutation(utils.BaseMutation):
         instance = providers.RateSheet.access_by(info.context.request).get(
             id=input["id"]
         )
+        data = input.copy()
         serializer = serializers.RateSheetModelSerializer(
             instance,
-            data=input,
+            data=data,
             context=info.context.request,
             partial=True,
         )
@@ -568,11 +569,136 @@ class UpdateRateSheetMutation(utils.BaseMutation):
         serializer.is_valid(raise_exception=True)
         rate_sheet = serializer.save()
 
+        # Handle services updates like in CreateRateSheetMutation
+        if "services" in data:
+            save_many_to_many_data(
+                "services",
+                serializers.ServiceLevelModelSerializer,
+                rate_sheet,
+                payload=data,
+                context=info.context.request,
+            )
+
         return UpdateRateSheetMutation(
             rate_sheet=providers.RateSheet.objects.get(id=input["id"])
         )
 
 
+@strawberry.type
+class UpdateRateSheetZoneCellMutation(utils.BaseMutation):
+    rate_sheet: typing.Optional[types.RateSheetType] = None
+
+    @staticmethod
+    @transaction.atomic
+    @utils.authentication_required
+    @utils.authorization_required(["manage_carriers"])
+    def mutate(
+        info: Info, **input: inputs.UpdateRateSheetZoneCellMutationInput
+    ) -> "UpdateRateSheetZoneCellMutation":
+        rate_sheet = providers.RateSheet.access_by(info.context.request).get(
+            id=input["id"]
+        )
+        service = rate_sheet.services.get(id=input["service_id"])
+
+        try:
+            service.update_zone_cell(
+                zone_id=input["zone_id"], field=input["field"], value=input["value"]
+            )
+        except ValueError as e:
+            logger.error(f"Invalid zone id: {e}")
+            raise exceptions.ValidationError({"zone_id": "invalid zone id"})
+
+        return UpdateRateSheetZoneCellMutation(rate_sheet=rate_sheet)
+
+
+@strawberry.type
+class BatchUpdateRateSheetCellsMutation(utils.BaseMutation):
+    rate_sheet: typing.Optional[types.RateSheetType] = None
+
+    @staticmethod
+    @transaction.atomic
+    @utils.authentication_required
+    @utils.authorization_required(["manage_carriers"])
+    def mutate(
+        info: Info, **input: inputs.BatchUpdateRateSheetCellsMutationInput
+    ) -> "BatchUpdateRateSheetCellsMutation":
+        rate_sheet = providers.RateSheet.access_by(info.context.request).get(
+            id=input["id"]
+        )
+
+        # Group updates by service_id for efficient processing
+        service_updates = {}
+        for update in input["updates"]:
+            service_id = update["service_id"]
+            if service_id not in service_updates:
+                service_updates[service_id] = []
+            service_updates[service_id].append(
+                {
+                    "zone_id": update["zone_id"],
+                    "field": update["field"],
+                    "value": update["value"],
+                }
+            )
+
+        # Use optimized structure if available, otherwise fall back to legacy
+        if rate_sheet.zones is not None and rate_sheet.service_rates is not None:
+            # Use optimized batch update on rate sheet
+            all_updates = []
+            for service_id, updates in service_updates.items():
+                for update in updates:
+                    all_updates.append(
+                        {
+                            "service_id": service_id,
+                            "zone_id": update["zone_id"],
+                            "field": update["field"],
+                            "value": update["value"],
+                        }
+                    )
+            try:
+                rate_sheet.batch_update_service_rates(all_updates)
+            except Exception as e:
+                logger.error(f"Invalid zone id: {e}")
+                raise exceptions.ValidationError(
+                    {"rate_sheet": "failed to update rate sheet"}
+                )
+        else:
+            # Fall back to legacy per-service updates
+            for service_id, updates in service_updates.items():
+                try:
+                    service = rate_sheet.services.get(id=service_id)
+                    service.batch_update_cells(updates)
+                except ValueError as e:
+                    logger.error(f"Invalid zone id: {e}")
+                    raise exceptions.ValidationError(
+                        {"service_id": "failed to update service"}
+                    )
+
+        return BatchUpdateRateSheetCellsMutation(rate_sheet=rate_sheet)
+
+
+@strawberry.type
+class DeleteRateSheetServiceMutation(utils.BaseMutation):
+    rate_sheet: typing.Optional[types.RateSheetType] = None
+
+    @staticmethod
+    @transaction.atomic
+    @utils.authentication_required
+    @utils.authorization_required(["manage_carriers"])
+    def mutate(
+        info: Info, **input: inputs.DeleteRateSheetServiceMutationInput
+    ) -> "DeleteRateSheetServiceMutation":
+        rate_sheet = providers.RateSheet.access_by(info.context.request).get(
+            id=input["rate_sheet_id"]
+        )
+        service = rate_sheet.services.get(id=input["service_id"])
+
+        # Remove service from rate sheet and delete it
+        rate_sheet.services.remove(service)
+        service.delete()
+
+        return DeleteRateSheetServiceMutation(rate_sheet=rate_sheet)
+
+
 @strawberry.type
 class PartialShipmentMutation(utils.BaseMutation):
     shipment: typing.Optional[types.ShipmentType] = None
@@ -841,10 +967,14 @@ class CreateMetafieldMutation(utils.BaseMutation):
     ) -> "CreateMetafieldMutation":
         data = input.copy()
 
-        metafield = serializers.MetafieldModelSerializer.map(
-            data=data,
-            context=info.context.request,
-        ).save().instance
+        metafield = (
+            serializers.MetafieldModelSerializer.map(
+                data=data,
+                context=info.context.request,
+            )
+            .save()
+            .instance
+        )
 
         return CreateMetafieldMutation(metafield=metafield)
 
@@ -862,10 +992,14 @@ class UpdateMetafieldMutation(utils.BaseMutation):
         data = input.copy()
         instance = core.Metafield.access_by(info.context.request).get(id=data.get("id"))
 
-        metafield = serializers.MetafieldModelSerializer.map(
-            instance,
-            data=data,
-            context=info.context.request,
-        ).save().instance
+        metafield = (
+            serializers.MetafieldModelSerializer.map(
+                instance,
+                data=data,
+                context=info.context.request,
+            )
+            .save()
+            .instance
+        )
 
         return UpdateMetafieldMutation(metafield=metafield)

karrio/server/graph/schemas/base/types.py

@@ -639,7 +639,6 @@ class AddressType:
     email: typing.Optional[str]
     phone_number: typing.Optional[str]
     state_code: typing.Optional[str]
-    suburb: typing.Optional[str]
     residential: typing.Optional[bool]
     street_number: typing.Optional[str]
     address_line1: typing.Optional[str]
@@ -1118,6 +1117,7 @@ class ShipmentType:
 @strawberry.type
 class ServiceZoneType:
     object_type: str
+    id: typing.Optional[str] = None
     label: typing.Optional[str] = None
     rate: typing.Optional[float] = None
 
@@ -1176,7 +1176,8 @@ class ServiceLevelType:
 
     @strawberry.field
     def zones(self: providers.ServiceLevel) -> typing.List[ServiceZoneType]:
-        return [ServiceZoneType.parse(zone) for zone in self.zones or []]
+        # Use computed_zones for backward compatibility with optimized structure
+        return [ServiceZoneType.parse(zone) for zone in self.computed_zones]
 
     @strawberry.field
     def metadata(self: providers.RateSheet) -> typing.Optional[utils.JSON]:

karrio/server/graph/serializers.py

@@ -12,6 +12,7 @@ import karrio.server.manager.models as manager
 import karrio.server.graph.models as graph
 import karrio.server.core.models as core
 import karrio.server.user.models as auth
+import karrio.server.core.gateway as gateway
 
 
 class UserModelSerializer(serializers.ModelSerializer):
@@ -268,10 +269,10 @@ class ServiceLevelModelSerializer(serializers.ModelSerializer):
         )
         self.instance.save(update_fields=["zones"])
 
-    def update(self, instance, validated_data):
+    def update(self, instance, validated_data, context=None, **kwargs):
         """Handle partial updates of service level data including zones."""
         zones_data = validated_data.pop("zones", None)
-        instance = super().update(instance, validated_data)
+        instance = super().update(instance, validated_data, context=context)
 
         if zones_data is not None:
             # Handle zone updates if provided
@@ -362,10 +363,10 @@ class RateSheetModelSerializer(serializers.ModelSerializer):
             ).filter(id__in=list(_ids))
 
             for carrier in _carriers:
-                carrier.settings.rate_sheet = (
+                carrier.rate_sheet = (
                     self.instance if carrier.id in carriers else None
                 )
-                carrier.settings.save(update_fields=["rate_sheet"])
+                carrier.save(update_fields=["rate_sheet"])
 
     def update(self, instance, validated_data, **kwargs):
         """Handle updates of rate sheet data including services and carriers."""

karrio/server/graph/tests/test_partial_shipments.py

@@ -0,0 +1,603 @@
+import json
+from unittest.mock import ANY, patch
+from django.urls import reverse
+from rest_framework import status
+from karrio.core.models import RateDetails, ChargeDetails
+from karrio.server.graph.tests.base import GraphTestCase, Result
+import karrio.server.manager.models as manager
+
+
+class TestPartialShipmentMutation(GraphTestCase):
+    def setUp(self) -> None:
+        super().setUp()
+        # Create a draft shipment using the REST API
+        self.shipment = self._create_draft_shipment()
+
+    def _create_draft_shipment(self):
+        """Create a draft shipment using the REST API without service (no rates fetched)"""
+        url = reverse("karrio.server.manager:shipment-list")
+
+        with patch("karrio.server.core.gateway.utils.identity") as mock:
+            mock.return_value = RETURNED_RATES_VALUE
+            response = self.client.post(url, DRAFT_SHIPMENT_DATA)
+            self.assertEqual(response.status_code, status.HTTP_201_CREATED)
+            response_data = json.loads(response.content)
+            return response_data
+
+    def test_partial_update_options_with_none_values(self):
+        """Test updating options with explicit None values to remove existing options"""
+        shipment_id = self.shipment["id"]
+
+        response = self.query(
+            """
+            mutation partial_shipment_update($data: PartialShipmentMutationInput!) {
+              partial_shipment_update(input: $data) {
+                shipment {
+                  id
+                  options
+                }
+                errors {
+                  field
+                  messages
+                }
+              }
+            }
+            """,
+            operation_name="partial_shipment_update",
+            variables={
+                "data": {
+                    "id": shipment_id,
+                    "options": {
+                        "insurance": None,
+                        "signature_confirmation": None,
+                        "currency": "USD",
+                        "new_option": "test_value",
+                    },
+                }
+            },
+        )
+
+        self.assertResponseNoErrors(response)
+        self.assertDictEqual(response.data, OPTIONS_UPDATE_RESPONSE)
+
+    def test_partial_update_shipper_address_fields(self):
+        """Test updating random fields within shipper address"""
+        shipment_id = self.shipment["id"]
+
+        response = self.query(
+            """
+            mutation partial_shipment_update($data: PartialShipmentMutationInput!) {
+              partial_shipment_update(input: $data) {
+                shipment {
+                  id
+                  shipper {
+                    id
+                    company_name
+                    person_name
+                    phone_number
+                    email
+                    city
+                    postal_code
+                  }
+                }
+                errors {
+                  field
+                  messages
+                }
+              }
+            }
+            """,
+            operation_name="partial_shipment_update",
+            variables={
+                "data": {
+                    "id": shipment_id,
+                    "shipper": {
+                        "company_name": "Updated Corp Inc.",
+                        "email": "updated@example.com",
+                        "phone_number": "555-123-4567",
+                    },
+                }
+            },
+        )
+
+        self.assertResponseNoErrors(response)
+        self.assertDictEqual(response.data, SHIPPER_UPDATE_RESPONSE)
+
+    def test_partial_update_recipient_address_fields(self):
+        """Test updating random fields within recipient address"""
+        shipment_id = self.shipment["id"]
+
+        response = self.query(
+            """
+            mutation partial_shipment_update($data: PartialShipmentMutationInput!) {
+              partial_shipment_update(input: $data) {
+                shipment {
+                  id
+                  recipient {
+                    id
+                    company_name
+                    person_name
+                    address_line1
+                    city
+                    state_code
+                    residential
+                  }
+                }
+                errors {
+                  field
+                  messages
+                }
+              }
+            }
+            """,
+            operation_name="partial_shipment_update",
+            variables={
+                "data": {
+                    "id": shipment_id,
+                    "recipient": {
+                        "person_name": "John Updated",
+                        "address_line1": "456 Updated St",
+                        "residential": True,
+                    },
+                }
+            },
+        )
+
+        self.assertResponseNoErrors(response)
+        self.assertDictEqual(response.data, RECIPIENT_UPDATE_RESPONSE)
+
+    def test_partial_update_payment_and_metadata(self):
+        """Test updating payment information and metadata"""
+        shipment_id = self.shipment["id"]
+
+        response = self.query(
+            """
+            mutation partial_shipment_update($data: PartialShipmentMutationInput!) {
+              partial_shipment_update(input: $data) {
+                shipment {
+                  id
+                  payment {
+                    paid_by
+                    currency
+                    account_number
+                  }
+                  metadata
+                  reference
+                }
+                errors {
+                  field
+                  messages
+                }
+              }
+            }
+            """,
+            operation_name="partial_shipment_update",
+            variables={
+                "data": {
+                    "id": shipment_id,
+                    "payment": {"paid_by": "recipient", "account_number": "123456789"},
+                    "metadata": {"customer_id": "CUST123", "order_number": "ORD456"},
+                    "reference": "REF789",
+                }
+            },
+        )
+
+        self.assertResponseNoErrors(response)
+        self.assertDictEqual(response.data, PAYMENT_METADATA_UPDATE_RESPONSE)
+
+    def test_partial_update_parcel_information(self):
+        """Test updating parcel information"""
+        shipment_id = self.shipment["id"]
+
+        response = self.query(
+            """
+            mutation partial_shipment_update($data: PartialShipmentMutationInput!) {
+              partial_shipment_update(input: $data) {
+                shipment {
+                  id
+                  parcels {
+                    id
+                    weight
+                    weight_unit
+                    package_preset
+                    description
+                    reference_number
+                  }
+                }
+                errors {
+                  field
+                  messages
+                }
+              }
+            }
+            """,
+            operation_name="partial_shipment_update",
+            variables={
+                "data": {
+                    "id": shipment_id,
+                    "parcels": [
+                        {
+                            "weight": 2.5,
+                            "weight_unit": "LB",
+                            "description": "Updated parcel description",
+                            "package_preset": "canadapost_corrugated_medium_box",
+                        }
+                    ],
+                }
+            },
+        )
+
+        self.assertResponseNoErrors(response)
+        self.assertDictEqual(response.data, PARCEL_UPDATE_RESPONSE)
+
+    def test_partial_update_label_type_and_options_merge(self):
+        """Test updating label type and ensuring options are properly merged"""
+        shipment_id = self.shipment["id"]
+
+        # First, set some initial options
+        self.query(
+            """
+            mutation partial_shipment_update($data: PartialShipmentMutationInput!) {
+              partial_shipment_update(input: $data) {
+                shipment {
+                  id
+                  options
+                }
+              }
+            }
+            """,
+            operation_name="partial_shipment_update",
+            variables={
+                "data": {
+                    "id": shipment_id,
+                    "options": {
+                        "insurance": 100,
+                        "signature_confirmation": True,
+                        "currency": "CAD",
+                    },
+                }
+            },
+        )
+
+        # Now update label type and add more options
+        response = self.query(
+            """
+            mutation partial_shipment_update($data: PartialShipmentMutationInput!) {
+              partial_shipment_update(input: $data) {
+                shipment {
+                  id
+                  label_type
+                  options
+                }
+                errors {
+                  field
+                  messages
+                }
+              }
+            }
+            """,
+            operation_name="partial_shipment_update",
+            variables={
+                "data": {
+                    "id": shipment_id,
+                    "label_type": "ZPL",
+                    "options": {"delivery_confirmation": True, "priority": "high"},
+                }
+            },
+        )
+
+        self.assertResponseNoErrors(response)
+        self.assertDictEqual(response.data, LABEL_OPTIONS_MERGE_RESPONSE)
+
+    def test_full_shipment_update_simulation(self):
+        """Test updating multiple fields at once to simulate frontend save operation"""
+        shipment_id = self.shipment["id"]
+
+        # Set the shipment ID dynamically in the test data
+        full_update_data = FULL_UPDATE_DATA.copy()
+        full_update_data["data"]["id"] = shipment_id
+
+        response = self.query(
+            """
+            mutation partial_shipment_update($data: PartialShipmentMutationInput!) {
+              partial_shipment_update(input: $data) {
+                shipment {
+                  id
+                  status
+                  shipper {
+                    company_name
+                    person_name
+                    email
+                  }
+                  recipient {
+                    company_name
+                    person_name
+                    email
+                  }
+                  parcels {
+                    weight
+                    weight_unit
+                    description
+                  }
+                  payment {
+                    paid_by
+                    currency
+                  }
+                  options
+                  metadata
+                  reference
+                  label_type
+                }
+                errors {
+                  field
+                  messages
+                }
+              }
+            }
+            """,
+            operation_name="partial_shipment_update",
+            variables=full_update_data,
+        )
+
+        self.assertResponseNoErrors(response)
+        self.assertDictEqual(response.data, FULL_UPDATE_SIMULATION_RESPONSE)
+
+
+# Test data
+DRAFT_SHIPMENT_DATA = {
+    "recipient": {
+        "address_line1": "125 Church St",
+        "person_name": "John Poop",
+        "company_name": "A corp.",
+        "phone_number": "514 000 0000",
+        "city": "Moncton",
+        "country_code": "CA",
+        "postal_code": "E1C4Z8",
+        "residential": False,
+        "state_code": "NB",
+    },
+    "shipper": {
+        "address_line1": "5840 Oak St",
+        "person_name": "Jane Doe",
+        "company_name": "B corp.",
+        "phone_number": "514 000 9999",
+        "city": "Vancouver",
+        "country_code": "CA",
+        "postal_code": "V6M2V9",
+        "residential": False,
+        "state_code": "BC",
+    },
+    "parcels": [
+        {
+            "weight": 1,
+            "weight_unit": "KG",
+            "package_preset": "canadapost_corrugated_small_box",
+        }
+    ],
+    "payment": {"currency": "CAD", "paid_by": "sender"},
+    # Note: No service specified, so this creates a draft without rates
+}
+
+RETURNED_RATES_VALUE = [
+    [
+        RateDetails(
+            carrier_id="canadapost",
+            carrier_name="canadapost",
+            currency="CAD",
+            transit_days=2,
+            service="canadapost_priority",
+            total_charge=106.71,
+            extra_charges=[
+                ChargeDetails(amount=13.92, currency="CAD", name="Duty and taxes"),
+                ChargeDetails(amount=2.7, currency="CAD", name="Fuel surcharge"),
+                ChargeDetails(amount=-11.74, currency="CAD", name="SMB Savings"),
+                ChargeDetails(amount=-9.04, currency="CAD", name="Discount"),
+                ChargeDetails(amount=101.83, currency="CAD", name="Base surcharge"),
+            ],
+        )
+    ],
+    [],
+]
+
+FULL_UPDATE_DATA = {
+    "data": {
+        "id": None,  # Will be set dynamically in the test
+        "shipper": {
+            "company_name": "Full Update Shipper Corp",
+            "person_name": "Updated Shipper",
+            "email": "shipper@fullupdate.com",
+        },
+        "recipient": {
+            "company_name": "Full Update Recipient Corp",
+            "person_name": "Updated Recipient",
+            "email": "recipient@fullupdate.com",
+        },
+        "parcels": [
+            {"weight": 3.0, "weight_unit": "KG", "description": "Full update parcel"}
+        ],
+        "payment": {"paid_by": "third_party", "currency": "USD"},
+        "options": {"insurance": 250, "priority": "express"},
+        "metadata": {"full_update": True, "test_case": "full_update_simulation"},
+        "reference": "FULL_UPDATE_REF",
+        "label_type": "PNG",
+    }
+}
+
+# Expected responses for assertDictEqual
+OPTIONS_UPDATE_RESPONSE = {
+    "data": {
+        "partial_shipment_update": {
+            "shipment": {
+                "id": ANY,
+                "options": {
+                    "shipping_date": ANY,
+                    "shipment_date": ANY,
+                    "currency": "USD",
+                    "new_option": "test_value",
+                },
+            },
+            "errors": None,
+        }
+    }
+}
+
+SHIPPER_UPDATE_RESPONSE = {
+    "data": {
+        "partial_shipment_update": {
+            "shipment": {
+                "id": ANY,
+                "shipper": {
+                    "id": ANY,
+                    "company_name": "Updated Corp Inc.",
+                    "person_name": "Jane Doe",
+                    "phone_number": "555-123-4567",
+                    "email": "updated@example.com",
+                    "city": "Vancouver",
+                    "postal_code": "V6M2V9",
+                },
+            },
+            "errors": None,
+        }
+    }
+}
+
+RECIPIENT_UPDATE_RESPONSE = {
+    "data": {
+        "partial_shipment_update": {
+            "shipment": {
+                "id": ANY,
+                "recipient": {
+                    "id": ANY,
+                    "company_name": "A corp.",
+                    "person_name": "John Updated",
+                    "address_line1": "456 Updated St",
+                    "city": "Moncton",
+                    "state_code": "NB",
+                    "residential": True,
+                },
+            },
+            "errors": None,
+        }
+    }
+}
+
+PAYMENT_METADATA_UPDATE_RESPONSE = {
+    "data": {
+        "partial_shipment_update": {
+            "shipment": {
+                "id": ANY,
+                "payment": {
+                    "paid_by": "recipient",
+                    "currency": None,
+                    "account_number": "123456789",
+                },
+                "metadata": {
+                    "customer_id": "CUST123",
+                    "order_number": "ORD456",
+                },
+                "reference": "REF789",
+            },
+            "errors": None,
+        }
+    }
+}
+
+PARCEL_UPDATE_RESPONSE = {
+    "data": {
+        "partial_shipment_update": {
+            "shipment": {
+                "id": ANY,
+                "parcels": [
+                    {
+                        "id": ANY,
+                        "weight": 1.0,
+                        "weight_unit": "KG",
+                        "package_preset": "canadapost_corrugated_small_box",
+                        "description": None,
+                        "reference_number": ANY,
+                    },
+                    {
+                        "id": ANY,
+                        "weight": 2.5,
+                        "weight_unit": "LB",
+                        "package_preset": "canadapost_corrugated_medium_box",
+                        "description": "Updated parcel description",
+                        "reference_number": ANY,
+                    },
+                ],
+            },
+            "errors": None,
+        }
+    }
+}
+
+LABEL_OPTIONS_MERGE_RESPONSE = {
+    "data": {
+        "partial_shipment_update": {
+            "shipment": {
+                "id": ANY,
+                "label_type": "ZPL",
+                "options": {
+                    "shipping_date": ANY,
+                    "shipment_date": ANY,
+                    "currency": "CAD",
+                    "insurance": 100,
+                    "signature_confirmation": True,
+                    "delivery_confirmation": True,
+                    "priority": "high",
+                },
+            },
+            "errors": None,
+        }
+    }
+}
+
+FULL_UPDATE_SIMULATION_RESPONSE = {
+    "data": {
+        "partial_shipment_update": {
+            "shipment": {
+                "id": ANY,
+                "status": "draft",
+                "shipper": {
+                    "company_name": "Full Update Shipper Corp",
+                    "person_name": "Updated Shipper",
+                    "email": "shipper@fullupdate.com",
+                },
+                "recipient": {
+                    "company_name": "Full Update Recipient Corp",
+                    "person_name": "Updated Recipient",
+                    "email": "recipient@fullupdate.com",
+                },
+                "parcels": [
+                    {
+                        "weight": 1.0,
+                        "weight_unit": "KG",
+                        "description": None,
+                    },
+                    {
+                        "weight": 3.0,
+                        "weight_unit": "KG",
+                        "description": "Full update parcel",
+                    },
+                ],
+                "payment": {
+                    "paid_by": "third_party",
+                    "currency": "USD",
+                },
+                "options": {
+                    "shipping_date": ANY,
+                    "shipment_date": ANY,
+                    "insurance": 250,
+                    "priority": "express",
+                },
+                "metadata": {
+                    "full_update": True,
+                    "test_case": "full_update_simulation",
+                },
+                "reference": "FULL_UPDATE_REF",
+                "label_type": "PNG",
+            },
+            "errors": None,
+        }
+    }
+}

karrio/server/graph/tests/test_rate_sheets.py

@@ -305,8 +305,14 @@ UPDATE_RATE_SHEET_RESPONSE = {
                 "services": [
                     {
                         "id": ANY,
-                        "service_name": "UPS Standard",
-                        "zones": [{"label": "Zone 1", "rate": 10.0}],
+                        "service_name": "Updated Service",
+                        "zones": [
+                            {
+                                "country_codes": ["US", "CA"],
+                                "label": "Updated Zone",
+                                "rate": 20.0,
+                            }
+                        ],
                     }
                 ],
             }

{karrio_server_graph-2025.5rc11.dist-info → karrio_server_graph-2025.5rc13.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: karrio_server_graph
-Version: 2025.5rc11
+Version: 2025.5rc13
 Summary: Multi-carrier shipping API Graph module
 Author-email: karrio <hello@karrio.io>
 License-Expression: Apache-2.0

{karrio_server_graph-2025.5rc11.dist-info → karrio_server_graph-2025.5rc13.dist-info}/RECORD

@@ -4,7 +4,7 @@ karrio/server/graph/apps.py,sha256=grL5ySDnW2amzj0nTXFm-WCoTWViVYvKqGlsi4k5epI,9
 karrio/server/graph/forms.py,sha256=5iGC3hj6dsAsupYz4QubyygTeTKDRR56Q5mYq0vbGhI,1962
 karrio/server/graph/models.py,sha256=CEnE4AsVyjBufyK6ebWmUH3s8DwA0HvZg0fUoZb5Pn4,1321
 karrio/server/graph/schema.py,sha256=2dXM8nD1usOc1S6QSalajoFmgwYuXxsrwj20AJ5HtT4,1151
-karrio/server/graph/serializers.py,sha256=vwP_U59aLnsnYVjwn3jrpz7SxOFzT_vplzSK-2FcsrU,13692
+karrio/server/graph/serializers.py,sha256=kAVpyoN74gkA20ZnmDycvr_yrwFOb_A9di2-vrkdL7U,13760
 karrio/server/graph/urls.py,sha256=HKo0gkx5TgoWDV0ap2QCtueNTmaAqvX6qVDe3c2UT1E,183
 karrio/server/graph/utils.py,sha256=eeIKeTfGP-8U6MkdMdTyobnk6wbNEE75Bm0cFu85RfU,9031
 karrio/server/graph/views.py,sha256=qWfa-wteB-Mb7glAYz6SVlZSwHPw_ImMm7XVmewdmTQ,2889
@@ -15,10 +15,10 @@ karrio/server/graph/migrations/0001_initial.py,sha256=oeaS5JSkQP6w9ulYar3mdn695y
 karrio/server/graph/migrations/0002_auto_20210512_1353.py,sha256=TnUwR9EP0qp3gJ38f9w0PYawK2VheDtqXEgyRhYZS2M,538
 karrio/server/graph/migrations/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 karrio/server/graph/schemas/__init__.py,sha256=Kn-I1j3HP3jwZccpz6FL9r1k6b3UEAMVh2kFPCKNS0w,80
-karrio/server/graph/schemas/base/__init__.py,sha256=aarmPw7csVZCeRxRaU6FNZ6tbN8SOla-nbYqlXFn3lw,14272
-karrio/server/graph/schemas/base/inputs.py,sha256=Lptv7JfyXP6EKMSQ838oZ4Qyr9GckbrJzXiLg1CCJMc,20762
-karrio/server/graph/schemas/base/mutations.py,sha256=a1_OxbhRV89PrOZ-T0dMEfiGJAzEWSYosGDrTkStUO0,28731
-karrio/server/graph/schemas/base/types.py,sha256=KPN154yeFiIyFU0pfkhPjcP-h5Nr2hrzdQkDX--ufeU,46371
+karrio/server/graph/schemas/base/__init__.py,sha256=J6pf4SzfAtQ4yaNapEuHuoyOA0-UGhueqDD3n0JzTb0,15118
+karrio/server/graph/schemas/base/inputs.py,sha256=T1T7cNVGudgVGOkPnus6TfK_i-lT_cjOBlpaPTHummk,21363
+karrio/server/graph/schemas/base/mutations.py,sha256=1S-aUpoKwwDslLtbReJb_5kx9tDr3-EfAhQQ9D7-6K0,33583
+karrio/server/graph/schemas/base/types.py,sha256=lKC-mdLluLNaPn9DXQDxh7Vl8JT9b_s_CIqZkiO0NWo,46458
 karrio/server/graph/templates/graphql/graphiql.html,sha256=MQjQbBqoRE0QLsOUck8SaXo6B2oJO8dT6YZzUqbDan0,3786
 karrio/server/graph/templates/karrio/email_change_email.html,sha256=YHqTy9VGV_s7kio57Tg3v7TCIN3QlnPHi2ioTOcHJLE,467
 karrio/server/graph/templates/karrio/email_change_email.txt,sha256=NXXuzLR63hn2F1fVAjzmuguptuuTvujwqI7YLSrQoio,431
@@ -27,11 +27,12 @@ karrio/server/graph/tests/__init__.py,sha256=dPzsYY5hoO5gmY6fhL8tiz7Bfst8RB3JzsB
 karrio/server/graph/tests/base.py,sha256=m3k1CE-d9JHDsqfyXX0Fwksmkel33S7sh9-zkypp8wc,4004
 karrio/server/graph/tests/test_carrier_connections.py,sha256=PoXxJB53jBIz8j1GYCzTiaTW14Q0D1McdOkDYgcOqNM,6140
 karrio/server/graph/tests/test_metafield.py,sha256=K7Oon0CLEm_MUMbmcu0t2iAZvFN8Wl7Kp4QAWeUXo_Y,12783
-karrio/server/graph/tests/test_rate_sheets.py,sha256=iNdN0GR26wv8lZd7slUbHTs1HbasKlMk8gk-A-XRZ5c,10079
+karrio/server/graph/tests/test_partial_shipments.py,sha256=dPIdIq4wiyovOaIIzbIX69eZnBqCA4ZvBSiGKYADM2g,19031
+karrio/server/graph/tests/test_rate_sheets.py,sha256=cUzPV8dXQFPFh1r7W8bY6Lou3fjh8f9VGpyZrfbMXec,10300
 karrio/server/graph/tests/test_templates.py,sha256=WVU6vcfr6tEk917uSn1dECU8bkQtgD7FNuE-GJuFido,21626
 karrio/server/graph/tests/test_user_info.py,sha256=K91BL7SgxLWflCZriSVI8Vt5M5RIqmSCHKrgN2w8GmM,1928
 karrio/server/settings/graph.py,sha256=cz2yQHbp3xCfyFKuUkPEFfkI2fFVggExIY49wGz7mt0,106
-karrio_server_graph-2025.5rc11.dist-info/METADATA,sha256=9PSly89Cr9aY3qz-tZZFodyUPg_tvtF54Sc-yDcYfY4,744
-karrio_server_graph-2025.5rc11.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-karrio_server_graph-2025.5rc11.dist-info/top_level.txt,sha256=D1D7x8R3cTfjF_15mfiO7wCQ5QMtuM4x8GaPr7z5i78,12
-karrio_server_graph-2025.5rc11.dist-info/RECORD,,
+karrio_server_graph-2025.5rc13.dist-info/METADATA,sha256=qUDbtpb-Ih-hHRixZXLIiLxsPIcnsCJaqIf14lzGcJk,744
+karrio_server_graph-2025.5rc13.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+karrio_server_graph-2025.5rc13.dist-info/top_level.txt,sha256=D1D7x8R3cTfjF_15mfiO7wCQ5QMtuM4x8GaPr7z5i78,12
+karrio_server_graph-2025.5rc13.dist-info/RECORD,,