Adiciona get_by_document em BankAccountWrapper (#20)

* adiciona o GetByDocument Mixin

* adiciona testes

* fmt

imopay_wrapper/wrapper/address.py

@@ -1,8 +1,16 @@
-from .base import BaseImopayWrapper, CreateMixin, UpdateMixin, RetrieveMixin
+from .base import (
+    BaseImopayWrapper,
+    CreateMixin,
+    UpdateMixin,
+    RetrieveMixin,
+    GetByDocumentMixin,
+)
 from ..models.address import Address
 
 
-class AddressWrapper(BaseImopayWrapper, CreateMixin, UpdateMixin, RetrieveMixin):
+class AddressWrapper(
+    BaseImopayWrapper, CreateMixin, UpdateMixin, RetrieveMixin, GetByDocumentMixin
+):
     """
     Wrapper para os métodos de address
     """
@@ -26,8 +34,3 @@ def update(self, identifier, data: dict):
             action=self.action, subaction="update_by_name_and_uf", identifier=identifier
         )
         return self._patch(url, instance.to_dict())
-
-    def get_by_document(self, document):
-        data = {"cpf_cnpj": document}
-        url = self._construct_url(action=self.action, subaction="get_by_document")
-        return self._post(url, data)

imopay_wrapper/wrapper/bank_account.py

@@ -1,8 +1,16 @@
-from .base import BaseImopayWrapper, CreateMixin, RetrieveMixin, DestroyMixin
+from .base import (
+    BaseImopayWrapper,
+    CreateMixin,
+    RetrieveMixin,
+    DestroyMixin,
+    GetByDocumentMixin,
+)
 from ..models.bank_account import BankAccount
 
 
-class BankAccountWrapper(BaseImopayWrapper, CreateMixin, RetrieveMixin, DestroyMixin):
+class BankAccountWrapper(
+    BaseImopayWrapper, CreateMixin, RetrieveMixin, DestroyMixin, GetByDocumentMixin
+):
     """
     Wrapper para os métodos de contas bancárias
     """

imopay_wrapper/wrapper/base.py

@@ -258,3 +258,10 @@ class DestroyMixin:
     def destroy(self, identifier: str):
         url = self._construct_url(action=self.action, identifier=identifier)
         return self._delete(url)
+
+
+class GetByDocumentMixin:
+    def get_by_document(self, document: str):
+        data = {"cpf_cnpj": document}
+        url = self._construct_url(action=self.action, subaction="get_by_document")
+        return self._post(url, data)

tests/wrapper/test_address.py

@@ -2,13 +2,32 @@
 
 from ..utils import LocalImopayTestCase
 from imopay_wrapper import ImopayWrapper
+from imopay_wrapper.wrapper.base import (
+    CreateMixin,
+    RetrieveMixin,
+    GetByDocumentMixin,
+)
 from imopay_wrapper.models.address import Address
 
 
 class AddressWrapperTestCase(LocalImopayTestCase):
     def setUp(self):
         self.client = ImopayWrapper().address
 
+    def test_mixins(self):
+        """
+        Dado:
+            - um client ImopayWrapper().address
+        Quando:
+            - N/A
+        Então:
+            - client deve ser uma instância de
+                CreateMixin, RetrieveMixin, DestroyMixin, GetByDocumentMixin
+        """
+        mixins = (CreateMixin, RetrieveMixin, GetByDocumentMixin)
+        for mixin in mixins:
+            self.assertIsInstance(self.client, mixin)
+
     def test_model(self):
         self.assertEqual(self.client.model, Address)
 

tests/wrapper/test_bank_account.py

@@ -1,6 +1,11 @@
 from ..utils import LocalImopayTestCase
 from imopay_wrapper import ImopayWrapper
-from imopay_wrapper.wrapper.base import CreateMixin, RetrieveMixin, DestroyMixin
+from imopay_wrapper.wrapper.base import (
+    CreateMixin,
+    RetrieveMixin,
+    DestroyMixin,
+    GetByDocumentMixin,
+)
 from imopay_wrapper.models.bank_account import BankAccount
 
 
@@ -15,9 +20,12 @@ def test_mixins(self):
         Quando:
             - N/A
         Então:
-            - client deve ser uma instância de (CreateMixin, RetrieveMixin, DestroyMixin)  # noqa
+            - client deve ser uma instância de
+                CreateMixin, RetrieveMixin, DestroyMixin, GetByDocumentMixin
         """
-        self.assertIsInstance(self.client, (CreateMixin, RetrieveMixin, DestroyMixin))
+        mixins = (CreateMixin, RetrieveMixin, DestroyMixin, GetByDocumentMixin)
+        for mixin in mixins:
+            self.assertIsInstance(self.client, mixin)
 
     def test_model(self):
         """