Fix tests

Author: Bladieblah

mollie/api/objects/balance.py

@@ -2,7 +2,7 @@
 
 from .balance_report import BalanceReport
 from .base import ObjectBase
-from .list import ObjectList
+from .list import PaginationList
 
 
 class Balance(ObjectBase):
@@ -59,7 +59,7 @@ def get_report(self, **params: Any) -> BalanceReport:
 
         return BalanceReports(self.client, self).get_report(params=params)
 
-    def get_transactions(self, **params: Any) -> ObjectList:
+    def get_transactions(self, **params: Any) -> PaginationList:
         from ..resources import BalanceTransactions
 
         return BalanceTransactions(self.client, self).list(params=params)

mollie/api/objects/list.py

@@ -1,8 +1,10 @@
 from abc import ABC, abstractmethod
-from typing import TYPE_CHECKING, Type
+from typing import TYPE_CHECKING, Any, Optional, Type
+
 from .base import ObjectBase
 
 if TYPE_CHECKING:
+    from mollie.api.client import Client
     from mollie.api.resources.base import ResourceBase
 
 
@@ -87,11 +89,10 @@ def new(self, result):
         ...
 
 
-
 class PaginationList(ListBase):
     _parent: "ResourceBase"
 
-    def __init__(self, result, parent: "ResourceBase", client=None):
+    def __init__(self, result: Any, parent: "ResourceBase", client: "Client"):
         # If an empty dataset was injected, we mock the structure that the remainder of the clas expects.
         # TODO: it would be better if the ObjectList was initiated with a list of results, rather than with
         # the full datastructure as it is now, so we can remove all this mucking around with fake data,
@@ -130,7 +131,7 @@ def new(self, result):
 class ObjectList(ListBase):
     _object_type: Type[ObjectBase]
 
-    def __init__(self, result, object_type: Type[ObjectBase], client=None):
+    def __init__(self, result: Any, object_type: Type[ObjectBase], client: Optional["Client"] = None):
         # If an empty dataset was injected, we mock the structure that the remainder of the clas expects.
         # TODO: it would be better if the ObjectList was initiated with a list of results, rather than with
         # the full datastructure as it is now, so we can remove all this mucking around with fake data,
@@ -148,10 +149,10 @@ def get_next(self):
 
     def get_previous(self):
         return None
-    
+
     @property
     def object_type(self):
         return self._object_type
-    
+
     def new(self, result):
         return ObjectList(result, self._object_type, self.client)

mollie/api/resources/chargebacks.py

@@ -1,7 +1,7 @@
 from typing import TYPE_CHECKING, Any
 
 from ..objects.chargeback import Chargeback
-from ..objects.list import ObjectList
+from ..objects.list import PaginationList
 from .base import ResourceBase, ResourceGetMixin, ResourceListMixin
 
 if TYPE_CHECKING:
@@ -72,7 +72,7 @@ def __init__(self, client: "Client", profile: "Profile") -> None:
         self._profile = profile
         super().__init__(client)
 
-    def list(self, **params: Any) -> ObjectList:
+    def list(self, **params: Any) -> PaginationList:
         # Set the profileId in the query params
         params.update({"profileId": self._profile.id})
         return Chargebacks(self.client).list(**params)

mollie/api/resources/onboarding.py

@@ -12,6 +12,7 @@
 
 class Onboarding(ResourceGetMixin):
     """Resource handler for the `/onboarding` endpoint."""
+
     object_type = OnboardingObject
 
     def get(self, resource_id: str, **params: Any) -> OnboardingObject:

mollie/api/resources/payments.py

@@ -1,7 +1,7 @@
 from typing import TYPE_CHECKING, Any, Dict, Optional
 
 from ..objects.customer import Customer
-from ..objects.list import ObjectList
+from ..objects.list import PaginationList
 from ..objects.order import Order
 from ..objects.payment import Payment
 from ..objects.profile import Profile
@@ -72,7 +72,7 @@ def __init__(self, client: "Client", order: Order) -> None:
     def get_resource_path(self) -> str:
         return f"orders/{self._order.id}/payments"
 
-    def list(self) -> ObjectList:
+    def list(self) -> PaginationList:
         """
         List the payments that might have been embedded in the related order.
 
@@ -86,7 +86,7 @@ def list(self) -> ObjectList:
             },
             "count": len(payments),
         }
-        return ObjectList(data, Payment, self.client)
+        return PaginationList(data, self, self.client)
 
 
 class CustomerPayments(PaymentsBase, ResourceCreateMixin, ResourceListMixin):
@@ -143,7 +143,7 @@ def __init__(self, client: "Client", profile: Profile) -> None:
         self._profile = profile
         super().__init__(client)
 
-    def list(self, **params: Any) -> ObjectList:
+    def list(self, **params: Any) -> PaginationList:
         # Set the profileId in the query params
         params.update({"profileId": self._profile.id})
         return Payments(self.client).list(**params)

mollie/api/resources/refunds.py

@@ -1,6 +1,6 @@
 from typing import TYPE_CHECKING, Any, Dict, Optional
 
-from ..objects.list import ObjectList
+from ..objects.list import PaginationList
 from ..objects.order import Order
 from ..objects.payment import Payment
 from ..objects.profile import Profile
@@ -97,7 +97,7 @@ def __init__(self, client: "Client", profile: Profile) -> None:
         self._profile = profile
         super().__init__(client)
 
-    def list(self, **params: Any) -> ObjectList:
+    def list(self, **params: Any) -> PaginationList:
         # Set the profileId in the query params
         params.update({"profileId": self._profile.id})
         return Refunds(self.client).list(**params)