import unittest from unittest.mock import patch from gds_helpers import to_xml, export, to_dict, xml_tostring from pycaps.shipment import ShipmentType from pycaps.ncshipment import NonContractShipmentType from purplship.domain.Types import ShipmentRequest from purplship.mappers.caps import CanadaPostProxy from tests.caps.fixture import proxy from tests.utils import strip MockedShipmentResponseXML = """ 347881315405043891 created 12345 """ class TestCanadaPostShipment(unittest.TestCase): def setUp(self): self.Shipment = ShipmentType() self.Shipment.build(to_xml(ShipmentRequestXML)) self.NCShipment = NonContractShipmentType() self.NCShipment.build(to_xml(NCShipmentRequestXML)) def test_create_shipment_request(self): payload = ShipmentRequest(**shipment_data) Shipment_ = proxy.mapper.create_shipment_request(payload) self.assertEqual(export(Shipment_), export(self.Shipment)) def test_create_ncshipment_request(self): payload = ShipmentRequest(**ncshipment_data) NCShipment_ = proxy.mapper.create_shipment_request(payload) # removed for testing purpose NCShipment_.delivery_spec.parcel_characteristics.document = None self.assertEqual(export(NCShipment_), export(self.NCShipment)) @patch("purplship.mappers.caps.caps_proxy.http", return_value="") def test_create_shipment(self, http_mock): proxy.create_shipment(self.Shipment) xmlStr = http_mock.call_args[1]["data"].decode("utf-8") self.assertEqual(strip(xmlStr), strip(ShipmentRequestXML)) @patch("purplship.mappers.caps.caps_proxy.http", return_value="") def test_create_ncshipment(self, http_mock): proxy.create_shipment(self.NCShipment) xmlStr = http_mock.call_args[1]["data"].decode("utf-8") self.assertEqual(strip(xmlStr), strip(NCShipmentRequestXML)) def test_parse_shipment_response(self): parsed_response = proxy.mapper.parse_shipment_response( to_xml(ShipmentResponseXML) ) self.assertEqual(to_dict(parsed_response), to_dict(ParsedShipmentResponse)) def test_parse_ncshipment_response(self): parsed_response = proxy.mapper.parse_shipment_response( to_xml(NCShipmentResponseXML) ) self.assertEqual(to_dict(parsed_response), to_dict(NCParsedShipmentResponse)) @patch( "purplship.mappers.caps.caps_proxy.http", return_value=MockedShipmentResponseXML ) def test_get_info_calls(self, http_mock): with patch.object( CanadaPostProxy, "_get_info", return_value="" ) as get_info_mock: proxy_ = CanadaPostProxy(proxy.client) proxy_.create_shipment(self.NCShipment) link = xml_tostring(get_info_mock.call_args[0][0]) self.assertEqual(strip(link), strip(ShipmentPriceLinkXML)) @patch("purplship.mappers.caps.caps_proxy.http", return_value="") def test_get_info(self, http_mock): proxy._get_info(to_xml(ShipmentPriceLinkXML)) args = http_mock.call_args[1] self.assertEqual(to_dict(args), GetInfoRequestArgs) if __name__ == "__main__": unittest.main() NCParsedShipmentResponse = [ { "carrier": "CanadaPost", "charges": [ {"amount": "18.10", "currency": "CAD", "name": "base-amount"}, {"amount": "0.00", "currency": "CAD", "name": "gst-amount"}, {"amount": "0.00", "currency": "CAD", "name": "pst-amount"}, {"amount": "2.53", "currency": "CAD", "name": "hst-amount"}, {"amount": "1.36", "currency": "CAD", "name": "FUELSC"}, {"amount": "0", "currency": "CAD", "name": "DC"}, ], "documents": [ "https://ct.soa-gw.canadapost.ca/rs/artifact/76108cb5192002d5/10238/0" ], "reference": {"type": "Shipment Id", "value": "406951321983787352"}, "services": ["DOM.EP", "DC"], "shipment_date": "2012-03-14", "total_charge": { "amount": "21.99", "currency": "CAD", "name": "Shipment charge", }, "tracking_numbers": ["12345678901234"], }, [], ] ParsedShipmentResponse = [ { "carrier": "CanadaPost", "charges": [ {"amount": "9.19", "currency": "CAD", "name": "base-amount"}, {"amount": "0.00", "currency": "CAD", "name": "gst-amount"}, {"amount": "0", "currency": "CAD", "name": "pst-amount"}, {"amount": "2.29", "currency": "CAD", "name": "hst-amount"}, {"amount": "0.00", "currency": "CAD", "name": "AUTDISC"}, {"amount": "0.90", "currency": "CAD", "name": "FUELSC"}, {"amount": "0", "currency": "CAD", "name": "DC"}, {"amount": "7.50", "currency": "CAD", "name": "UP"}, ], "documents": ["https://XX/rs/artifact/11111111/5555555/0"], "reference": {"type": "Shipment Id", "value": "347881315405043891"}, "services": ["DOM.EP", "DC", "UP"], "shipment_date": "2011-10-07", "total_charge": { "amount": "19.88", "currency": "CAD", "name": "Shipment charge", }, "tracking_numbers": ["12345"], }, [], ] GetInfoRequestArgs = { "url": "https://XX/rs/111111111/2222222222/shipment/347881315405043891/price", "headers": { "Accept": "application/vnd.cpc.shipment-v8+xml", "Authorization": "Basic dXNlcm5hbWU6cGFzc3dvcmQ=", "Accept-language": "en-CA", }, "method": "GET", } ShipmentPriceLinkXML = """ """ ShipmentRequestXML = """ 123456789 true K2B8J6 DOM.EP Bob CGI 1 (450) 823-8432 502 MAIN ST N MONTREAL QC CA H2B1A0 Jain CGI 23 jardin private Ottawa ON CA K1K4T3 20. 6 12 9 false john.doe@yahoo.com true false true 8.5x11 true false true 0040662505 Account """ ShipmentResponseXML = """ 347881315405043891 created 12345 DOM.EP 9.19 DC 0 UP 7.50 AUTDISC 0.00 FUELSC 0.90 17.59 0.00 0 2.29 19.88 false true 2 2011-10-07 10.000 """ NCShipmentRequestXML = """ J8R1A2 DOM.EP Canada Post Corporation 555-555-5555 2701 Riverside Drive Ottawa ON K1A0B1 John Doe Consumer 2701 Receiver Drive Ottawa ON CA K1A0B1 15. 1 1 1 true """ NCShipmentResponseXML = """ 406951321983787352 12345678901234 J4W4T0 BP BROSSARD 0192 0001111111 DOM.EP 15.000 18.10 19.46 0.00 0.00 2.53 DC 0 FUELSC 1.36 Canada Post Corporation www.canadapost.ca John Doe 076838 2012-03-13T08:27:20-05:00 VIS 21.99 CAD Sale false true 1 2012-03-14 """ shipment_data = { "shipper": { "company_name": "CGI", "address_lines": ["502 MAIN ST N"], "city": "MONTREAL", "postal_code": "H2B1A0", "country_code": "CA", "person_name": "Bob", "phone_number": "1 (450) 823-8432", "state_code": "QC", "account_number": "123456789", }, "recipient": { "company_name": "CGI", "address_lines": ["23 jardin private"], "city": "Ottawa", "postal_code": "K1K4T3", "country_code": "CA", "person_name": "Jain", "state_code": "ON", }, "shipment": { "items": [{"height": 9, "length": 6, "width": 12, "weight": 20.0}], "label": {"format": "8.5x11"}, "services": ["Expedited_Parcel"], "dimension_unit": "CM", "weight_unit": "KG", "options": [{"code": "Collect_on_delivery"}], "extra": { "cpc-pickup-indicator": True, "requested-shipping-point": "K2B8J6", "mailing-tube": False, "notification": { "email": "john.doe@yahoo.com", "on-shipment": True, "on-exception": False, "on-delivery": True, }, "preferences": { "show-packing-instructions": True, "show-postage-rate": False, "show-insured-value": True, }, "settlement-info": { "contract-id": "0040662505", "intended-method-of-payment": "Account", }, }, }, } ncshipment_data = { "shipper": { "company_name": "Canada Post Corporation", "address_lines": ["2701 Riverside Drive"], "city": "Ottawa", "postal_code": "K1A0B1", "phone_number": "555-555-5555", "account_number": "123456789", "state_code": "ON", }, "recipient": { "company_name": "Consumer", "address_lines": ["2701 Receiver Drive"], "city": "Ottawa", "postal_code": "K1A0B1", "country_code": "CA", "person_name": "John Doe", "state_code": "ON", }, "shipment": { "items": [{"height": 1, "length": 1, "width": 1, "weight": 15.0}], "dimension_unit": "CM", "weight_unit": "KG", "services": ["Expedited_Parcel"], "options": [{"code": "Collect_on_delivery"}], "extra": { "requested-shipping-point": "J8R1A2", "preferences": {"show-packing-instructions": True}, }, }, }