adiciona valores bancários como opcionais

Author: peidrao

bb_wrapper/models/pagamentos.py

@@ -37,6 +37,18 @@ class TipoChavePIX(IntEnum):
     uuid = 4
 
 
+class TipoContaPIX(IntEnum):
+    """
+    1: Conta do tipo corrente
+    2: Conta do tipo pagamento
+    3: Conta do tipo poupança
+    """
+
+    conta_corrente = 1
+    conta_pagamento = 3
+    conta_poupanca = 3
+
+
 class FinalidadeTED(IntEnum):
     """
     1: Conta corrente outros bancos
@@ -100,10 +112,10 @@ class TransferenciaDadosBancariosPIX(BaseModel):
     data: str
     valor: float
     documento: str
-    tipoConta: str
-    agencia: str
-    conta: str
-    digitoVerificadorConta: str
+    tipoConta: TipoContaPIX
+    agencia: Optional[str]
+    conta: Optional[str]
+    digitoVerificadorConta: Optional[str]
     formaIdentificacao: Optional[str]
     descricaoPagamento: Optional[str]
     cpf: Optional[str]

bb_wrapper/wrapper/pagamento_lote.py

@@ -617,10 +617,10 @@ def criar_transferencia_por_dados_bancarios_pix(
         data_transferencia,
         valor_transferencia,
         tipo_conta_favorecido,
-        agencia_favorecido,
-        conta_favorecido,
-        digito_verificador_conta,
         numero_ispb,
+        agencia_favorecido=None,
+        conta_favorecido=None,
+        digito_verificador_conta=None,
         conta_pagamento=None,
         descricao="",
         tipo_pagamento=128,

tests/wrapper/test_pagamentos.py

@@ -1018,6 +1018,63 @@ def test_criar_transferencia_pix_dados_bancarios(self):
         self.assertEqual(2, self.total_requests())
         self.mock_responses.assert_call_count(request_url, 1)
 
+    def test_criar_transferencia_pix_dados_bancarios2(self):
+        """
+        Teste para verificar a URL da requisição e
+        dados de transferência PIX por dados bancarios
+        """
+
+        request_url = PagamentoLoteBBWrapper()._construct_url(
+            "lotes-transferencias-pix"
+        )
+
+        expected_json = {
+            "numeroRequisicao": "123",
+            "agenciaDebito": "345",
+            "contaCorrenteDebito": "678",
+            "digitoVerificadorContaCorrente": "X",
+            "tipoPagamento": 128,
+            "listaTransferencias": [
+                {
+                    "data": "19042023",
+                    "valor": "50",
+                    "descricaoPagamento": "Uma transferência via dados bancários",
+                    "formaIdentificacao": 5,
+                    "tipoConta": 1,
+                    "contaPagamento": 12345678,
+                    "numeroISPB": "360305",
+                    "cpf": "28779295827",
+                }
+            ],
+        }
+
+        self.mock_responses.add(
+            responses.POST,
+            request_url,
+            headers=self._build_authorization_header(1),
+            json=expected_json,
+        )
+
+        response = PagamentoLoteBBWrapper().criar_transferencia_por_dados_bancarios_pix(
+            n_requisicao="123",
+            agencia="345",
+            conta="678",
+            dv_conta="X",
+            data_transferencia="19042023",
+            valor_transferencia=12,
+            tipo_conta_favorecido=1,
+            conta_pagamento=12345678,
+            numero_ispb="360305",
+            descricao="Uma transferência via dados bancários",
+            documento="28779295827",
+        )
+
+        self.assertEqual(request_url, response.url)
+        self.assertEqual(self._get_headers(), response.headers)
+        self.assertEqual(expected_json, response.json())
+        self.assertEqual(2, self.total_requests())
+        self.mock_responses.assert_call_count(request_url, 1)
+
     def test_consultar_pix(self):
         """
         - Teste para consultar um pix de um determinado lote