def test_init_custom_fields_bank_account_individual(self):
instance = MagicMock(BANK_ACCOUNT_IDENTIFIER="foo")
cpf = gen.cpf()
Token.init_custom_fields(instance, foo="bar", taxpayer_id=cpf)
self.assertEqual(instance.token_type, instance.BANK_ACCOUNT_TYPE)
self.assertEqual(instance.taxpayer_id, cpf)
def test_init_custom_fields_card(self):
"""
We set the CARD_IDENTIFIER to 'foo'
and pass foo='bar' on args!
"""
instance = MagicMock(CARD_IDENTIFIER="foo", )
Token.init_custom_fields(instance, foo="bar")
self.assertEqual(instance.token_type, instance.CARD_TYPE)
def test_init_custom_fields_with_bank_account(self):
instance = MagicMock(TYPES={"bank_account"},
BANK_ACCOUNT_TYPE="bank_account")
Token.init_custom_fields(instance,
type="bank_account",
bank_account=IndividualBankAccountFactory())
self.assertEqual(instance.token_type, "bank_account")
self.assertIsInstance(instance.bank_account, BankAccount)
def test_init_custom_fields_bank_account_business(self):
"""
We set the BANK_ACCOUNT_IDENTIFIER to 'foo'
and pass foo='bar' on args!
We must pass 'ein' or 'taxpayer_id' too
"""
instance = MagicMock(BANK_ACCOUNT_IDENTIFIER="foo")
cnpj = gen.cnpj()
Token.init_custom_fields(instance, foo="bar", ein=cnpj)
self.assertEqual(instance.token_type, instance.BANK_ACCOUNT_TYPE)
self.assertEqual(instance.ein, cnpj)
def test_init_custom_fields_with_allow_empty(self):
"""
Nesse cenário o Token é usado como se fosse
um cartão recebendo seus dados juntamente do id.
Dado um cartão c1 (previamente já criado noa Zoop)
Quando for criado um Token(id=c1.id, allow_empty=True) t1
Então o t1.token_type deve ser None
"""
c1 = MagicMock(id="1")
instance = MagicMock(TYPES={"card"},
CARD_TYPE="card",
_allow_empty=True)
Token.init_custom_fields(instance, id=c1.id)
self.assertEqual(instance.token_type, None)
def test_init_custom_fields_with_card(self):
instance = MagicMock(TYPES={"card"}, CARD_TYPE="card")
Token.init_custom_fields(instance, type="card", card=CardFactory())
self.assertEqual(instance.token_type, "card")
self.assertIsInstance(instance.card, Card)