ajustes

imobanco · Feb 1, 2024 · d1ea125 · d1ea125
1 parent ded5afb
commit d1ea125
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 3 deletions.
diff --git a/bb_wrapper/models/pagamentos.py b/bb_wrapper/models/pagamentos.py
@@ -103,6 +103,10 @@ def _set_data(cls, values):
         documento = values.pop("documento", None)
         if documento is not None:
             PixService().verify_document(documento, values)
+            if values.get("cpf") is None:
+                del values["cpf"]
+            if values.get("cnpj") is None:
+                del values["cnpj"]
 
         values.pop("chave")
         return values
@@ -146,6 +150,10 @@ def _set_data(cls, values):
         documento = values.pop("documento", None)
         if documento is not None:
             PixService().verify_document(documento, values)
+            if values.get("cpf") is None:
+                del values["cpf"]
+            if values.get("cnpj") is None:
+                del values["cnpj"]
 
         return values
 

diff --git a/tests/wrapper/test_pagamentos.py b/tests/wrapper/test_pagamentos.py
@@ -111,7 +111,6 @@ def test_criar_dados_chave_aleatoria_transferencia_pix(self):
             "listaTransferencias": [
                 {
                     "email": None,
-                    "cnpj": None,
                     "cpf": 11122233300,
                     "dddTelefone": None,
                     "telefone": None,
@@ -197,7 +196,6 @@ def test_criar_dados_telefone_transferencia_pix(self):
             "listaTransferencias": [
                 {
                     "email": None,
-                    "cnpj": None,
                     "cpf": 11122233300,
                     "dddTelefone": 11,
                     "telefone": 985732102,
@@ -240,7 +238,6 @@ def test_criar_dados_email_transferencia_pix(self):
             "listaTransferencias": [
                 {
                     "email": "[email protected]",
-                    "cnpj": None,
                     "cpf": 11122233300,
                     "dddTelefone": None,
                     "telefone": None,