# -*- coding: utf-8 -*- # Part of Odoo. See LICENSE file for full copyright and licensing details. import logging from unittest.mock import patch from odoo import Command from odoo.addons.delivery_easypost.models.easypost_request import EasypostRequest from odoo.addons.delivery_easypost.tests.common import EasypostTestCommon from odoo.exceptions import UserError from odoo.tests import tagged, Form _logger = logging.getLogger(__name__) @tagged('-standard', 'external') class TestDeliveryEasypost(EasypostTestCommon): def wiz_put_in_pack(self, picking): """ Helper to use the 'stock.put.in.pack' wizard in order to call the 'action_put_in_pack' method. """ wiz_action = picking.action_put_in_pack() self.assertEqual(wiz_action['res_model'], 'stock.put.in.pack', 'Wrong wizard returned') wiz = Form.from_action(self.env, wiz_action) wiz.package_type_id = picking.carrier_id.easypost_default_package_type_id wiz.save().action_put_in_pack() def test_easypost_one_package_shipping(self): """ Try to rate and ship an order from New York to Miami. It will not use a specific package and everything will be consider to be inside the same package. """ SaleOrder = self.env['sale.order'] sol_1_vals = {'product_id': self.server.id} sol_2_vals = {'product_id': self.miniServer.id} so_vals_fedex = {'partner_id': self.jackson.id, 'order_line': [(0, None, sol_1_vals), (0, None, sol_2_vals)]} sale_order_fedex = SaleOrder.create(so_vals_fedex) # I add delivery cost in Sales order delivery_wizard = Form(self.env['choose.delivery.carrier'].with_context({ 'default_order_id': sale_order_fedex.id, 'default_carrier_id': self.easypost_fedex_carrier.id })) choose_delivery_carrier = delivery_wizard.save() choose_delivery_carrier.update_price() self.assertGreater(choose_delivery_carrier.delivery_price, 0.00, "Could't get rate for this order from easypost fedex") choose_delivery_carrier.button_confirm() sale_order_fedex.action_confirm() self.assertEqual(len(sale_order_fedex.picking_ids), 1, "The Sales Order did not generate a picking for ep-fedex.") picking_fedex = sale_order_fedex.picking_ids[0] picking_fedex.action_assign() picking_fedex.move_line_ids.write({'quantity': 1}) self.assertGreater(picking_fedex.weight, 0.0, "Picking weight should be positive.(ep-fedex)") # Set a service in order to test rate request for a specific service. self.easypost_fedex_carrier.easypost_default_service_id = self.env['easypost.service'].search([('name', '=', 'STANDARD_OVERNIGHT')]).id if not self.easypost_fedex_carrier.easypost_default_service_id: _logger.warning('"STANDARD_OVERNIGHT" is not anymore a fedex service, easypost default service is not tested.') picking_fedex.move_ids.picked = True picking_fedex._action_done() self.assertGreater(picking_fedex.carrier_price, 0.0, "Easypost carrying price is probably incorrect(fedex)") self.assertIsNot(picking_fedex.carrier_tracking_ref, False, "Easypost did not return any tracking number (fedex)") # broken -> FedEx returned error: Ground Shipping is not authorized for this User def test_easypost_multiple_packages_shipping(self): """ Same than test with one package. This time it will use the put in pack functionality. It will send twice the default package type with 2 servers and 3 mini servers. """ SaleOrder = self.env['sale.order'] sol_1_vals = {'product_id': self.server.id} sol_2_vals = {'product_id': self.miniServer.id} so_vals_fedex = {'partner_id': self.jackson.id, 'order_line': [(0, None, sol_1_vals), (0, None, sol_2_vals)]} sale_order_fedex = SaleOrder.create(so_vals_fedex) # I add delivery cost in Sales order delivery_wizard = Form(self.env['choose.delivery.carrier'].with_context({ 'default_order_id': sale_order_fedex.id, 'default_carrier_id': self.easypost_fedex_carrier.id })) choose_delivery_carrier = delivery_wizard.save() choose_delivery_carrier.update_price() self.assertGreater(choose_delivery_carrier.delivery_price, 0.00, "Could't get rate for this order from easypost fedex") choose_delivery_carrier.button_confirm() sale_order_fedex.action_confirm() self.assertEqual(len(sale_order_fedex.picking_ids), 1, "The Sales Order did not generate a picking for ep-fedex.") picking_fedex = sale_order_fedex.picking_ids[0] self.assertEqual(picking_fedex.carrier_id.id, sale_order_fedex.carrier_id.id, "Carrier is not the same on Picking and on SO(easypost-fedex).") picking_fedex.action_assign() picking_fedex.move_ids[0].write({'quantity': 2, 'picked': True}) self.wiz_put_in_pack(picking_fedex) picking_fedex.move_ids[0].move_line_ids.result_package_id.package_type_id = self.fedex_default_package_type.id picking_fedex.move_ids[0].move_line_ids.result_package_id.shipping_weight = 10.0 picking_fedex.move_ids[1].write({'quantity': 3, 'picked': True}) self.wiz_put_in_pack(picking_fedex) picking_fedex.move_ids[1].move_line_ids.result_package_id.package_type_id = self.fedex_default_package_type.id picking_fedex.move_ids[1].move_line_ids.result_package_id.shipping_weight = 10.0 self.assertGreater(picking_fedex.weight, 0.0, "Picking weight should be positive.(ep-fedex)") picking_fedex._action_done() self.assertGreater(picking_fedex.carrier_price, 0.0, "Easypost carrying price is probably incorrect(fedex)") self.assertIsNot(picking_fedex.carrier_tracking_ref, False, "Easypost did not return any tracking number (fedex)") def test_easypost_one_package_international_shipping(self): """ Same than test_easypost_one_package_shipping with an international shipping. (it matters due to customs info). """ SaleOrder = self.env['sale.order'] sol_1_vals = {'product_id': self.server.id} sol_2_vals = {'product_id': self.miniServer.id} so_vals_fedex = {'partner_id': self.agrolait.id, 'order_line': [(0, None, sol_1_vals), (0, None, sol_2_vals)]} # Modify price due to US customs info. If you a value greater than # 2500$, it requires an specific AES code. self.server['list_price'] = 10.0 self.miniServer['list_price'] = 10.0 sale_order_fedex = SaleOrder.create(so_vals_fedex) # I add delivery cost in Sales order delivery_wizard = Form(self.env['choose.delivery.carrier'].with_context({ 'default_order_id': sale_order_fedex.id, 'default_carrier_id': self.easypost_fedex_carrier.id })) choose_delivery_carrier = delivery_wizard.save() choose_delivery_carrier.update_price() self.assertGreater(choose_delivery_carrier.delivery_price, 0.00, "Could't get rate for this order from easypost fedex") choose_delivery_carrier.button_confirm() sale_order_fedex.action_confirm() self.assertEqual(len(sale_order_fedex.picking_ids), 1, "The Sales Order did not generate a picking for ep-fedex.") picking_fedex = sale_order_fedex.picking_ids[0] self.assertEqual(picking_fedex.carrier_id.id, sale_order_fedex.carrier_id.id, "Carrier is not the same on Picking and on SO(easypost-fedex).") picking_fedex.action_assign() picking_fedex.move_line_ids.write({'quantity': 1}) picking_fedex.move_ids.picked = True self.assertGreater(picking_fedex.weight, 0.0, "Picking weight should be positive.(ep-fedex)") try: picking_fedex._action_done() except UserError as exc: if "carrier is not responding to our request" in exc.args[0]: _logger.warning('easypost test aborted, carrier is unresponsive.') return raise self.assertGreater(picking_fedex.carrier_price, 0.0, "Easypost carrying price is probably incorrect(fedex)") self.assertIsNot(picking_fedex.carrier_tracking_ref, False, "Easypost did not return any tracking number (fedex)") def test_easypost_extralight_package_shipping(self): """ Try to rate and ship an order from New York to Miami. Use a very light product. """ SaleOrder = self.env['sale.order'] sol_1_vals = {'product_id': self.microServer.id} so_vals_fedex = {'partner_id': self.jackson.id, 'order_line': [(0, None, sol_1_vals)]} sale_order_fedex = SaleOrder.create(so_vals_fedex) # I add delivery cost in Sales order delivery_wizard = Form(self.env['choose.delivery.carrier'].with_context({ 'default_order_id': sale_order_fedex.id, 'default_carrier_id': self.easypost_fedex_carrier.id })) choose_delivery_carrier = delivery_wizard.save() choose_delivery_carrier.update_price() self.assertGreater(choose_delivery_carrier.delivery_price, 0.00, "Could't get rate for this order from easypost fedex") choose_delivery_carrier.button_confirm() sale_order_fedex.action_confirm() def test_easypost_sends_correct_delivery_type_for_amazon(self): amazon_expected_delivery_type = self.easypost_fedex_carrier._get_delivery_type() self.assertEqual(amazon_expected_delivery_type, 'FedEx') @tagged('standard', '-external') class TestMockedDeliveryEasypost(TestDeliveryEasypost): def setUp(self): # this is needed because we use call the API for the carrier setup. with self.patch_easypost_requests(): super().setUp() def test_easypost_one_package_shipping(self): with self.patch_easypost_requests(): super().test_easypost_one_package_shipping() def test_easypost_multiple_packages_shipping(self): with self.patch_easypost_requests(): super().test_easypost_multiple_packages_shipping() def test_easypost_one_package_international_shipping(self): with self.patch_easypost_requests(): super().test_easypost_one_package_international_shipping() def test_easypost_extralight_package_shipping(self): with self.patch_easypost_requests(): super().test_easypost_extralight_package_shipping() def test_easypost_sends_correct_delivery_type_for_amazon(self): with self.patch_easypost_requests(): super().test_easypost_sends_correct_delivery_type_for_amazon() def test_easypost_retrieve_currency_from_pricelist(self): ''' Ensure the currency sent to easypost is correctly taken from the SO's pricelist instead of the package (which defaults to the company's). ''' def check_requested_currency(endpoint, data): # The currency can only be right for the delivery call. The rating call doesn't have a # picking to retrieve the currency through. if endpoint == 'orders' and 'OUT' in data.get('order[shipments][0][options][print_custom_1]', ''): self.assertEqual(data['order[from_address][country]'], 'BE') self.assertEqual(data['order[shipments][0][customs_info][customs_items][0][currency]'], 'USD') pricelist_usd = self.env['product.pricelist'].with_company(self.be_company).create({ 'name': 'Pricelist USD', 'currency_id': self.currency_usd.id, }) so = self.env['sale.order'].with_company(self.be_company).create({ 'partner_id': self.jackson.id, 'pricelist_id': pricelist_usd.id, 'order_line': [Command.create({ 'product_id': self.miniServer.id, 'product_uom_qty': 1.0, 'price_unit': 20, })] }) delivery_wizard = Form(self.env['choose.delivery.carrier'].with_context({ 'default_order_id': so.id, 'default_carrier_id': self.easypost_fedex_carrier.id })) with self.patch_easypost_requests(check_requested_currency): choose_delivery_carrier = delivery_wizard.save() choose_delivery_carrier.update_price() self.assertGreater(choose_delivery_carrier.delivery_price, 0.00, "Could't get rate for this order from easypost fedex") choose_delivery_carrier.button_confirm() so.action_confirm() picking = so.picking_ids picking._action_done() def test_easypost_dhl_express_multi_shipment_rate_error(self): """ DHL Express returns the rate of a multi-package order on the master (first) shipment only, along with a benign `rate_error` message "multi-shipment rate includes this shipment." on the order. Validating such a transfer must succeed instead of raising that message as an error. """ message = { 'carrier': 'DHLExpress', 'type': 'rate_error', 'message': 'DHLExpress multi-shipment rate includes this shipment.', } rates = [ {'id': 'rate_1', 'service': 'ExpressWorldwide', 'carrier': 'DHLExpress', 'rate': '10.00', 'currency': 'USD'}, {'id': 'rate_2', 'service': 'Express1200', 'carrier': 'DHLExpress', 'rate': '20.00', 'currency': 'USD'}, ] order_response = { 'id': 'order_test', 'messages': [message], 'rates': rates, 'shipments': [{'id': 'shp_1', 'rates': rates}, {'id': 'shp_2', 'rates': []}], } buy_response = { 'id': 'order_test', 'messages': [message], 'shipments': [ { 'id': 'shp_1', 'rates': rates, 'tracking_code': '123456789', 'tracker': {'public_url': 'https://track.easypost.test/123456789'}, 'postage_label': {'label_url': 'https://easypost.test/label.pdf'}, 'forms': [], }, {'id': 'shp_2', 'rates': [], 'tracking_code': None, 'tracker': None, 'postage_label': None, 'forms': []}, ], } def _dhl_request(endpoint, request_type='get', data=None): return buy_response if endpoint.endswith('/buy') else order_response with self.patch_easypost_requests(): sale_order_dhl = self.env['sale.order'].create({ 'partner_id': self.agrolait.id, 'order_line': [ Command.create({'product_id': self.server.id}), Command.create({'product_id': self.miniServer.id}), ], }) delivery_wizard = Form(self.env['choose.delivery.carrier'].with_context({ 'default_order_id': sale_order_dhl.id, 'default_carrier_id': self.easypost_dhl_carrier.id, })) choose_delivery_carrier = delivery_wizard.save() with patch.object(EasypostRequest, '_make_api_request', side_effect=_dhl_request): choose_delivery_carrier.update_price() choose_delivery_carrier.button_confirm() sale_order_dhl.action_confirm() picking_dhl = sale_order_dhl.picking_ids picking_dhl.action_assign() for move in picking_dhl.move_ids: move.write({'quantity': 1, 'picked': True}) self.wiz_put_in_pack(picking_dhl) move.move_line_ids.result_package_id.shipping_weight = 10.0 picking_dhl._action_done() self.assertEqual(picking_dhl.carrier_tracking_ref, '123456789')