Feat(wallets): add code to wallet (#361)

* add code to wallet

* fix tests

Author: annvelents

lago_python_client/models/charge.py

@@ -35,6 +35,7 @@ class Charge(BaseModel):
     filters: Optional[ChargeFilters]
     tax_codes: Optional[List[str]]
     applied_pricing_unit: Optional[AppliedPricingUnit]
+    accepts_target_wallet: Optional[bool]
 
 
 class Charges(BaseModel):
@@ -55,6 +56,7 @@ class ChargeResponse(BaseResponseModel):
     properties: Optional[Dict[str, Any]]
     filters: Optional[ChargeFilters]
     taxes: Optional[TaxesResponse]
+    accepts_target_wallet: Optional[bool]
 
 
 class ChargesResponse(BaseResponseModel):

lago_python_client/models/wallet.py

@@ -56,6 +56,7 @@ class Wallet(BaseModel):
     external_customer_id: Optional[str]
     rate_amount: Optional[str]
     name: Optional[str]
+    code: Optional[str]
     priority: Optional[int]
     paid_credits: Optional[str]
     granted_credits: Optional[str]
@@ -79,6 +80,7 @@ class WalletResponse(BaseResponseModel):
     status: str
     currency: str
     name: Optional[str]
+    code: Optional[str]
     priority: Optional[int]
     rate_amount: str
     credits_balance: str

tests/fixtures/plan.json

@@ -50,7 +50,8 @@
             "applied_to_organization": false,
             "created_at": "2022-04-29T08:59:51Z"
           }
-        ]
+        ],
+        "accepts_target_wallet": true
       }
     ],
     "minimum_commitment": {

tests/fixtures/wallet.json

@@ -6,6 +6,7 @@
     "status": "active",
     "currency": "EUR",
     "name": "name",
+    "code": "wallet_code",
     "priority": 30,
     "rate_amount": "1.5",
     "credits_balance": "10",

tests/test_wallet_client.py

@@ -31,6 +31,7 @@ def wallet_object():
     )
     return Wallet(
         name="name",
+        code="wallet_code",
         priority=30,
         external_customer_id="12345",
         rate_amount="1",
@@ -73,6 +74,7 @@ def test_valid_create_wallet_request(httpx_mock: HTTPXMock):
                 "external_customer_id": "12345",
                 "rate_amount": "1",
                 "name": "name",
+                "code": "wallet_code",
                 "priority": 30,
                 "paid_credits": "10",
                 "granted_credits": "10",