def test_remove_bankaccount_by_customer(
self,
customer_retrieve_source_mock,
customer_retrieve_mock,
bank_account_retrieve_mock,
bank_account_delete_mock,
):
stripe_bank_account = BankAccount._api_create(
customer=self.customer, source=FAKE_BANK_ACCOUNT_SOURCE["id"])
BankAccount.sync_from_stripe_data(stripe_bank_account)
self.assertEqual(
1,
BankAccount.objects.filter(id=stripe_bank_account["id"]).count())
bank_account = self.customer.bank_account.all()[0]
bank_account.remove()
self.assertEqual(
0,
BankAccount.objects.filter(id=stripe_bank_account["id"]).count())
api_key = bank_account.default_api_key
stripe_account = bank_account._get_stripe_account_id(api_key)
bank_account_delete_mock.assert_called_once_with(
self.customer.id,
bank_account.id,
api_key=api_key,
stripe_account=stripe_account,
)
def test_remove_already_deleted_card(self, customer_retrieve_mock):
stripe_bank_account = BankAccount._api_create(
customer=self.customer, source=FAKE_BANK_ACCOUNT_SOURCE["id"])
BankAccount.sync_from_stripe_data(stripe_bank_account)
self.assertEqual(self.customer.bank_account.count(), 1)
bank_account_object = self.customer.bank_account.first()
BankAccount.objects.filter(id=stripe_bank_account["id"]).delete()
self.assertEqual(self.customer.bank_account.count(), 0)
bank_account_object.remove()
self.assertEqual(self.customer.bank_account.count(), 0)
def test_remove_bankaccount_by_account(
self,
account_retrieve_mock,
):
stripe_bank_account = BankAccount._api_create(
account=self.custom_account, source=FAKE_BANK_ACCOUNT_IV["id"]
)
BankAccount.sync_from_stripe_data(stripe_bank_account)
# remove BankAccount
count, _ = BankAccount.objects.filter(id=stripe_bank_account["id"]).delete()
self.assertEqual(1, count)
def test__str__(self, fake_stripe_data, has_account, has_customer,
monkeypatch):
def mock_customer_get(*args, **kwargs):
data = deepcopy(FAKE_CUSTOMER_IV)
data["default_source"] = None
data["sources"] = []
return data
def mock_account_get(*args, **kwargs):
return deepcopy(FAKE_CUSTOM_ACCOUNT)
# monkeypatch stripe.Account.retrieve and stripe.Customer.retrieve calls to return
# the desired json response.
monkeypatch.setattr(stripe.Account, "retrieve", mock_account_get)
monkeypatch.setattr(stripe.Customer, "retrieve", mock_customer_get)
bankaccount = BankAccount.sync_from_stripe_data(fake_stripe_data)
default = False
if has_account:
default = fake_stripe_data["default_for_currency"]
assert (
f"{fake_stripe_data['bank_name']} {fake_stripe_data['currency']} {'Default' if default else ''} {fake_stripe_data['routing_number']} {fake_stripe_data['last4']}"
== str(bankaccount))
if has_customer:
customer = Customer.objects.filter(
id=fake_stripe_data["customer"]).first()
default_source = customer.default_source
default_payment_method = customer.default_payment_method
if (default_payment_method
and fake_stripe_data["id"] == default_payment_method.id
) or (default_source
and fake_stripe_data["id"] == default_source.id):
# current bankaccount is the default payment method or source
default = True
assert (
f"{fake_stripe_data['bank_name']} {fake_stripe_data['routing_number']} ({bankaccount.human_readable_status}) {'Default' if default else ''} {fake_stripe_data['currency']}"
== str(bankaccount))
if not has_account and not has_customer:
# ensure account and customer do not exist
fake_stripe_data_2 = deepcopy(fake_stripe_data)
fake_stripe_data_2["account"] = None
fake_stripe_data_2["customer"] = None
bankaccount = BankAccount.sync_from_stripe_data(fake_stripe_data_2)
default = fake_stripe_data_2["default_for_currency"]
assert (
f"{fake_stripe_data_2['bank_name']} {fake_stripe_data_2['currency']} {'Default' if default else ''} {fake_stripe_data_2['routing_number']} {fake_stripe_data_2['last4']}"
== str(bankaccount))
def test_remove_already_deleted_bankaccount_by_account(
self,
account_retrieve_mock,
bank_account_delete_mock,
):
stripe_bank_account = BankAccount._api_create(
account=self.custom_account, source=FAKE_BANK_ACCOUNT_IV["id"])
bank_account = BankAccount.sync_from_stripe_data(stripe_bank_account)
self.assertEqual(
1,
BankAccount.objects.filter(id=stripe_bank_account["id"]).count())
api_key = bank_account.default_api_key
stripe_account = bank_account._get_stripe_account_id(api_key)
assert bank_account.customer is None
assert bank_account.account is not None
# remove BankAccount
bank_account.remove()
self.assertEqual(
0,
BankAccount.objects.filter(id=stripe_bank_account["id"]).count())
bank_account_delete_mock.assert_called_once_with(
self.custom_account.id,
bank_account.id,
api_key=api_key,
stripe_account=stripe_account,
)
# remove BankAccount again
count, _ = BankAccount.objects.filter(
id=stripe_bank_account["id"]).delete()
self.assertEqual(0, count)
def test_attach_objects_hook_without_customer(self):
FAKE_BANK_ACCOUNT_DICT = deepcopy(FAKE_BANK_ACCOUNT_SOURCE)
FAKE_BANK_ACCOUNT_DICT["customer"] = None
bank_account = BankAccount.sync_from_stripe_data(
FAKE_BANK_ACCOUNT_DICT)
self.assertEqual(bank_account.customer, None)
def test_remove(
self,
customer_retrieve_mock,
bank_account_retrieve_mock,
bank_account_delete_mock,
):
stripe_bank_account = BankAccount._api_create(
customer=self.customer, source=FAKE_BANK_ACCOUNT_SOURCE["id"])
BankAccount.sync_from_stripe_data(stripe_bank_account)
self.assertEqual(1, self.customer.bank_account.count())
bank_account = self.customer.bank_account.all()[0]
bank_account.remove()
self.assertEqual(0, self.customer.bank_account.count())
self.assertTrue(bank_account_delete_mock.called)
def test_api_retrieve_by_customer_equals_retrieval_by_account(
self, account_retrieve_external_account_mock,
customer_retrieve_mock):
# deepcopy the BankAccount object
FAKE_BANK_ACCOUNT_DICT = deepcopy(FAKE_BANK_ACCOUNT_IV)
bankaccount = BankAccount.sync_from_stripe_data(FAKE_BANK_ACCOUNT_DICT)
bankaccount_by_customer = bankaccount.api_retrieve()
# Add account
FAKE_BANK_ACCOUNT_DICT["account"] = FAKE_CUSTOM_ACCOUNT["id"]
FAKE_BANK_ACCOUNT_DICT["customer"] = None
bankaccount = BankAccount.sync_from_stripe_data(FAKE_BANK_ACCOUNT_DICT)
bankaccount_by_account = bankaccount.api_retrieve()
# assert the same bankaccount object gets retrieved
self.assertCountEqual(bankaccount_by_customer, bankaccount_by_account)
def test_create_bank_account_finds_customer_with_account_absent(self):
bank_account = BankAccount.sync_from_stripe_data(FAKE_BANK_ACCOUNT_SOURCE)
self.assertEqual(self.customer, bank_account.customer)
self.assertEqual(
bank_account.get_stripe_dashboard_url(),
self.customer.get_stripe_dashboard_url(),
)
self.assert_fks(
bank_account,
expected_blank_fks={
"djstripe.BankAccount.account",
"djstripe.Customer.default_payment_method",
"djstripe.Customer.default_source",
"djstripe.Customer.coupon",
},
)
def test_create_bank_account_finds_account_with_customer_absent(self):
FAKE_BANK_ACCOUNT_DICT = deepcopy(FAKE_BANK_ACCOUNT_SOURCE)
FAKE_BANK_ACCOUNT_DICT["account"] = self.standard_account.id
FAKE_BANK_ACCOUNT_DICT["customer"] = None
bank_account = BankAccount.sync_from_stripe_data(FAKE_BANK_ACCOUNT_DICT)
self.assertEqual(self.standard_account, bank_account.account)
self.assertEqual(
bank_account.get_stripe_dashboard_url(),
self.standard_account.get_stripe_dashboard_url(),
)
self.assert_fks(
bank_account,
expected_blank_fks={
"djstripe.BankAccount.customer",
"djstripe.Customer.default_payment_method",
"djstripe.Customer.default_source",
"djstripe.Customer.coupon",
},
)
def test_human_readable_status(self, fake_stripe_data, monkeypatch):
def mock_customer_get(*args, **kwargs):
data = deepcopy(FAKE_CUSTOMER_IV)
data["default_source"] = None
data["sources"] = []
return data
def mock_account_get(*args, **kwargs):
return deepcopy(FAKE_CUSTOM_ACCOUNT)
# monkeypatch stripe.Account.retrieve and stripe.Customer.retrieve calls to return
# the desired json response.
monkeypatch.setattr(stripe.Account, "retrieve", mock_account_get)
monkeypatch.setattr(stripe.Customer, "retrieve", mock_customer_get)
bankaccount = BankAccount.sync_from_stripe_data(fake_stripe_data)
if fake_stripe_data["status"] == "new":
assert bankaccount.human_readable_status == "Pending Verification"
else:
assert (bankaccount.human_readable_status ==
enums.BankAccountStatus.humanize(
fake_stripe_data["status"]))
def test_attach_objects_hook_without_account(self):
bank_account = BankAccount.sync_from_stripe_data(
FAKE_BANK_ACCOUNT_SOURCE)
self.assertEqual(bank_account.account, None)