diff --git a/shopinvader_delivery_carrier/services/abstract_sale.py b/shopinvader_delivery_carrier/services/abstract_sale.py index 2ba6c0779b..f16a78ae7d 100644 --- a/shopinvader_delivery_carrier/services/abstract_sale.py +++ b/shopinvader_delivery_carrier/services/abstract_sale.py @@ -9,14 +9,14 @@ class AbstractSaleService(AbstractComponent): def _convert_shipping(self, cart): res = super(AbstractSaleService, self)._convert_shipping(cart) - selected_carrier = {} if cart.carrier_id: - carrier = cart.carrier_id - selected_carrier = { - "id": carrier.id, - "name": carrier.name, - "description": carrier.name, - } + # we do not need an estimation of the price + # so we do not pass the cart to the _prepare_carrier method + # and we remove the field + selected_carrier = self._prepare_carrier(cart.carrier_id) + selected_carrier.pop("price") + else: + selected_carrier = {} res.update( { "amount": { @@ -51,13 +51,9 @@ def _convert_amount(self, sale): ) return result - def _prepare_carrier(self, carrier, cart): - return { - "id": carrier.id, - "name": carrier.name, - "description": carrier.name, - "price": carrier.rate_shipment(cart).get("price", 0.0), - } + def _prepare_carrier(self, carrier, cart=None): + service = self.component(usage="delivery_carrier") + return service._prepare_carrier(carrier, cart=cart) # TODO: this method seems not used def _get_available_carrier(self, cart): diff --git a/shopinvader_delivery_carrier/services/delivery_carrier.py b/shopinvader_delivery_carrier/services/delivery_carrier.py index d741f09b58..56572593ec 100644 --- a/shopinvader_delivery_carrier/services/delivery_carrier.py +++ b/shopinvader_delivery_carrier/services/delivery_carrier.py @@ -76,6 +76,11 @@ def _validator_return_search(self): "required": False, "nullable": True, }, + "code": { + "type": "string", + "required": False, + "nullable": True, + }, "description": { "type": "string", "required": False, diff --git a/shopinvader_delivery_carrier/tests/common.py b/shopinvader_delivery_carrier/tests/common.py index ea7525844c..4b05f7fb5b 100644 --- a/shopinvader_delivery_carrier/tests/common.py +++ b/shopinvader_delivery_carrier/tests/common.py @@ -11,6 +11,8 @@ def setUpClass(cls): super(CommonCarrierCase, cls).setUpClass() cls.free_carrier = cls.env.ref("delivery.free_delivery_carrier") cls.poste_carrier = cls.env.ref("delivery.delivery_carrier") + cls.free_carrier.default_code = "FREE" + cls.poste_carrier.default_code = "POSTE" cls.product_1 = cls.env.ref("product.product_product_4b") cls.precision = 2 diff --git a/shopinvader_delivery_carrier/tests/test_carrier.py b/shopinvader_delivery_carrier/tests/test_carrier.py index 76481d159b..00108b66b0 100644 --- a/shopinvader_delivery_carrier/tests/test_carrier.py +++ b/shopinvader_delivery_carrier/tests/test_carrier.py @@ -21,6 +21,16 @@ def test_deprecated_apply_delivery_method(self): def test_setting_free_carrier(self): cart = self._set_carrier(self.free_carrier) self.assertEqual(cart["shipping"]["amount"]["total"], 0) + self.assertEqual( + cart["shipping"]["selected_carrier"], + { + "description": self.free_carrier.description or None, + "id": self.free_carrier.id, + "name": self.free_carrier.name, + "code": self.free_carrier.code, + "type": None, + }, + ) def test_setting_poste_carrier(self): cart = self._set_carrier(self.poste_carrier) @@ -59,6 +69,18 @@ def test_setting_poste_carrier(self): cart_amount["tax_without_shipping"], tax_without_shipping ) + # Check Selected carrier + self.assertEqual( + cart["shipping"]["selected_carrier"], + { + "description": self.poste_carrier.description or None, + "id": self.poste_carrier.id, + "name": self.poste_carrier.name, + "code": self.poste_carrier.code, + "type": None, + }, + ) + def test_reset_carrier_on_add_item(self): self._apply_carrier_and_assert_set() cart = self.add_item(self.product_1.id, 2) diff --git a/shopinvader_delivery_carrier/tests/test_delivery_carrier.py b/shopinvader_delivery_carrier/tests/test_delivery_carrier.py index 6266c2ec2f..924803566f 100644 --- a/shopinvader_delivery_carrier/tests/test_delivery_carrier.py +++ b/shopinvader_delivery_carrier/tests/test_delivery_carrier.py @@ -23,6 +23,7 @@ def test_search_all(self): "description": self.free_carrier.name or None, "id": self.free_carrier.id, "name": self.free_carrier.name, + "code": self.free_carrier.code, "type": None, }, { @@ -30,6 +31,7 @@ def test_search_all(self): "description": self.poste_carrier.name or None, "id": self.poste_carrier.id, "name": self.poste_carrier.name, + "code": self.poste_carrier.code, "type": None, }, ], @@ -46,6 +48,7 @@ def test_search_current_cart(self): "description": self.free_carrier.name or None, "id": self.free_carrier.id, "name": self.free_carrier.name, + "code": self.free_carrier.code, "type": None, }, { @@ -53,6 +56,7 @@ def test_search_current_cart(self): "description": self.poste_carrier.name or None, "id": self.poste_carrier.id, "name": self.poste_carrier.name, + "code": self.poste_carrier.code, "type": None, }, ],