Update Payment object, so it contains lines, billing and shipping address and restrict_payment_methods_to_country

Author: geertjanvdenbosch

mollie/api/objects/payment.py

@@ -161,6 +161,24 @@ def subscription_id(self):
     def cancel_url(self):
         return self._get_property("cancelUrl")
 
+    @property
+    def lines(self):
+        # We do not use the OrderLine object here, because lines on a Payment
+        # are not a resource, but simply a property that exists on the Payment.
+        return self._get_property("lines")
+
+    @property
+    def restrict_payment_methods_to_country(self):
+        return self._get_property("restrictPaymentMethodsToCountry")
+
+    @property
+    def shipping_address(self):
+        return self._get_property("shippingAddress")
+
+    @property
+    def billing_address(self):
+        return self._get_property("billingAddress")
+
     # documented _links
 
     @property

tests/responses/payment_single_no_links.json

@@ -10,6 +10,46 @@
     "value": "10.00",
     "currency": "EUR"
   },
+    "restrictPaymentMethodsToCountry": "NL",
+  "billingAddress": {
+    "streetAndNumber": "Keizersgracht 313",
+    "city": "Amsterdam",
+    "region": "Noord-Holland",
+    "postalCode": "1234AB",
+    "country": "NL",
+    "title": "Dhr.",
+    "givenName": "Piet",
+    "familyName": "Mondriaan",
+    "email": "[email protected]",
+    "phone": "+31309202070"
+  },
+  "shippingAddress": {
+    "streetAndNumber": "Keizersgracht 313",
+    "streetAdditional": "4th floor",
+    "city": "Haarlem",
+    "region": "Noord-Holland",
+    "postalCode": "5678AB",
+    "country": "NL",
+    "title": "Mr.",
+    "givenName": "Chuck",
+    "familyName": "Norris",
+    "email": "[email protected]"
+  },
+  "lines": [
+    {
+      "type": "physical",
+      "description": "Product 1",
+      "quantity": 1,
+      "unitPrice": {
+        "value": "2.00",
+        "currency": "EUR"
+      },
+      "totalAmount": {
+        "value": "2.00",
+        "currency": "EUR"
+      }
+    }
+  ],
   "description": "Order #12345",
   "method": "ideal",
   "metadata": {

tests/test_payments.py

@@ -110,6 +110,45 @@ def test_get_single_payment(client, response):
     assert payment.capture_mode == "automatic"
     assert payment.capture_before == "2023-01-20T09:13:37+00+00"
     assert payment.capture_delay == "20 hours"
+    assert payment.restrict_payment_methods_to_country == "NL"
+    # lines property
+    assert payment.lines == [
+        {
+            "type": "physical",
+            "description": "Product 1",
+            "quantity": 1,
+            "unitPrice": {"value": "2.00", "currency": "EUR"},
+            "totalAmount": {"value": "2.00", "currency": "EUR"},
+        }
+    ]
+    # billing_address property
+    assert payment.billing_address == {
+        "streetAndNumber": "Keizersgracht 313",
+        "city": "Amsterdam",
+        "region": "Noord-Holland",
+        "postalCode": "1234AB",
+        "country": "NL",
+        "title": "Dhr.",
+        "givenName": "Piet",
+        "familyName": "Mondriaan",
+        "email": "[email protected]",
+        "phone": "+31309202070",
+    }
+
+    # shipping_address property
+    assert payment.shipping_address == {
+        "streetAndNumber": "Keizersgracht 313",
+        "streetAdditional": "4th floor",
+        "city": "Haarlem",
+        "region": "Noord-Holland",
+        "postalCode": "5678AB",
+        "country": "NL",
+        "title": "Mr.",
+        "givenName": "Chuck",
+        "familyName": "Norris",
+        "email": "[email protected]",
+    }
+
     # properties from _links
     assert payment.checkout_url == "https://www.mollie.com/payscreen/select-method/7UhSN1zuXS"
     assert payment.changepaymentstate_url is None