def test_full_refund_of_partials(self, refund_create_mocked):
refund_create_mocked.return_value = stripe.Refund(id="R1")
self.charge.is_charged = True
with mock.patch("aa_stripe.signals.stripe_charge_refunded.send"
) as refund_signal_send:
to_refund = 50
# first refund
self.charge.refund(to_refund)
refund_create_mocked.assert_called_with(
charge=self.charge.stripe_charge_id,
amount=to_refund,
idempotency_key="{}-{}-{}-{}".format(
self.charge.object_id, self.charge.content_type_id, 0,
to_refund),
)
self.assertFalse(self.charge.is_refunded)
refund_signal_send.assert_called_with(sender=StripeCharge,
instance=self.charge)
# second refund
refund_create_mocked.reset_mock()
self.charge.refund(to_refund)
refund_create_mocked.assert_called_with(
charge=self.charge.stripe_charge_id,
amount=to_refund,
idempotency_key="{}-{}-{}-{}".format(
self.charge.object_id, self.charge.content_type_id, 50,
to_refund),
)
self.assertTrue(self.charge.is_refunded)
refund_signal_send.assert_called_with(sender=StripeCharge,
instance=self.charge)
self.assertEqual(self.charge.amount_refunded, 100)
def test_refund_over_charge_amount(self, refund_create_mocked):
refund_create_mocked.return_value = stripe.Refund(id="R1")
self.charge.is_charged = True
with self.assertRaises(StripeMethodNotAllowed) as ctx:
self.charge.refund(101)
self.assertEqual(ctx.exception.args[0], "Refunds exceed charge")
self.assertFalse(self.charge.is_refunded)
self.assertFalse(self.charge_refunded_signal_was_called)
def test_full_refund(self, refund_create_mocked):
refund_create_mocked.return_value = stripe.Refund(id="R1")
self.charge.is_charged = True
self.charge.refund()
self.assertTrue(self.charge.is_refunded)
self.assertEqual(self.charge.stripe_refund_id, "R1")
self.assertEqual(self.charge.amount_refunded, self.charge.amount)
self.assertTrue(self.charge_refunded_signal_was_called)
def test_already_refunded(self, refund_create_mocked):
refund_create_mocked.return_value = stripe.Refund(id="R1")
self.charge.is_charged = True
self.charge.is_refunded = True
with self.assertRaises(StripeMethodNotAllowed) as ctx:
self.charge.refund()
self.assertEqual(ctx.exception.args[0], "Already refunded.")
self.assertTrue(self.charge.is_refunded)
self.assertFalse(self.charge_refunded_signal_was_called)
def test_refund_on_not_charged(self, refund_create_mocked):
self.charge.refresh_from_db()
refund_create_mocked.return_value = stripe.Refund(id="R1")
self.assertTrue(
self.customer,
StripeCustomer.get_latest_active_customer_for_user(self.user))
with self.assertRaises(StripeMethodNotAllowed) as ctx:
self.charge.refund()
self.assertEqual(ctx.exception.args[0],
"Cannot refund not charged transaction.")
self.assertFalse(self.charge.is_refunded)
self.assertFalse(self.charge_refunded_signal_was_called)
def test_refund(self, refund_create_mocked):
data = {
"customer_id": "cus_AlSWz1ZQw7qG2z",
"currency": "usd",
"amount": 100,
"description": "ABC"
}
refund_create_mocked.return_value = stripe.Refund(id="R1")
customer = StripeCustomer.objects.create(
user=self.user,
stripe_customer_id=data["customer_id"],
stripe_js_response='"foo"')
self.assertTrue(
customer,
StripeCustomer.get_latest_active_customer_for_user(self.user))
charge = StripeCharge.objects.create(user=self.user,
amount=data["amount"],
customer=customer,
description=data["description"])
self.assertFalse(charge.is_refunded)
# refund - error: not charged
with self.assertRaises(StripeMethodNotAllowed):
charge.refund()
self.assertFalse(charge.is_refunded)
charge.is_charged = True
charge.stripe_charge_id = "abc"
charge.save()
# refund - passes
charge.refund()
refund_create_mocked.assert_called_with(charge=charge.stripe_charge_id)
self.assertTrue(charge.is_refunded)
self.assertEqual(charge.stripe_refund_id, "R1")
# refund - error: already refunded
with self.assertRaises(StripeMethodNotAllowed):
charge.refund()
self.assertTrue(charge.is_refunded)
class TestCharges(TestCase):
def _success_handler(self, sender, instance, **kwargs):
self.success_signal_was_called = True
def _exception_handler(self, sender, instance, **kwargs):
self.exception_signal_was_called = True
def _charge_refunded_handler(self, sender, instance, **kwargs):
self.charge_refunded_signal_was_called = True
def _reset_signals(self):
self.exception_signal_was_called = False
self.success_signal_was_called = False
self.charge_refunded_signal_was_called = False
def setUp(self):
self.user = UserModel.objects.create(email="*****@*****.**",
username="******",
password="******")
self._reset_signals()
stripe_charge_succeeded.connect(self._success_handler)
stripe_charge_card_exception.connect(self._exception_handler)
stripe_charge_refunded.connect(self._charge_refunded_handler)
self.data = {
"customer_id": "cus_AlSWz1ZQw7qG2z",
"currency": "usd",
"amount": 100,
"description": "ABC",
}
self.customer = StripeCustomer.objects.create(
user=self.user,
stripe_customer_id=self.data["customer_id"],
stripe_js_response='"foo"',
)
self.charge = StripeCharge.objects.create(
user=self.user,
amount=self.data["amount"],
customer=self.customer,
description=self.data["description"],
source=self.customer,
)
@mock.patch(
"aa_stripe.management.commands.charge_stripe.stripe.Charge.create")
def test_charge_unknown_stripe_error(self, charge_create_mocked):
stripe_error_json_body = {"error": {"type": "api_error"}}
charge_create_mocked.side_effect = StripeError(
json_body=stripe_error_json_body)
with self.assertRaises(SystemExit):
out = StringIO()
sys.stdout = out
call_command("charge_stripe")
self.charge.refresh_from_db()
self.assertFalse(self.success_signal_was_called)
self.assertFalse(self.exception_signal_was_called)
self.assertFalse(self.charge.is_charged)
self.assertFalse(self.charge.charge_attempt_failed)
self.assertDictEqual(self.charge.stripe_response,
stripe_error_json_body)
self.assertIn("Exception happened", out.getvalue())
@mock.patch(
"aa_stripe.management.commands.charge_stripe.stripe.Charge.create")
def test_charge_stripe_error(self, charge_create_mocked):
stripe_error_json_body = {
"error": {
"code": "resource_missing",
"doc_url":
"https://stripe.com/docs/error-codes/resource-missing",
"message": "No such customer: cus_ESrgXHlDA3E7mQ",
"param": "customer",
"type": "invalid_request_error",
}
}
charge_create_mocked.side_effect = StripeError(
json_body=stripe_error_json_body)
call_command("charge_stripe")
self.charge.refresh_from_db()
self.assertFalse(self.success_signal_was_called)
self.assertTrue(self.exception_signal_was_called)
self.assertFalse(self.charge.is_charged)
self.assertTrue(self.charge.charge_attempt_failed)
self.assertDictEqual(self.charge.stripe_response,
stripe_error_json_body)
@mock.patch(
"aa_stripe.management.commands.charge_stripe.stripe.Charge.create")
def test_charge_card_declined(self, charge_create_mocked):
card_error_json_body = {
"error": {
"charge": "ch_1F5C8nBszOVoiLmgPWC36cnI",
"code": "card_declined",
"decline_code": "generic_decline",
"doc_url": "https://stripe.com/docs/error-codes/card-declined",
"message": "Your card was declined.",
"type": "card_error",
}
}
charge_create_mocked.side_effect = CardError(
message="a", param="b", code="c", json_body=card_error_json_body)
call_command("charge_stripe")
self.charge.refresh_from_db()
self.assertFalse(self.success_signal_was_called)
self.assertTrue(self.exception_signal_was_called)
self.assertDictEqual(self.charge.stripe_response, card_error_json_body)
self.assertEqual(self.charge.stripe_charge_id,
"ch_1F5C8nBszOVoiLmgPWC36cnI")
@mock.patch(
"aa_stripe.management.commands.charge_stripe.stripe.Charge.create")
def test_charge_with_idempotency_key(self, charge_create_mocked):
charge_create_mocked.return_value = stripe.Charge(id="AA1")
idempotency_key = "idempotency_key123"
self.charge.charge(idempotency_key=idempotency_key)
self.assertTrue(self.charge.is_charged)
self.assertTrue(self.success_signal_was_called)
self.assertFalse(self.exception_signal_was_called)
self.assertEqual(self.charge.stripe_response["id"], "AA1")
charge_create_mocked.assert_called_with(
idempotency_key="{}-{}-{}".format(self.charge.object_id,
self.charge.content_type_id,
idempotency_key),
amount=self.charge.amount,
currency=self.data["currency"],
customer=self.data["customer_id"],
description=self.data["description"],
metadata={
"object_id": self.charge.object_id,
"content_type_id": self.charge.content_type_id,
},
)
@mock.patch(
"aa_stripe.management.commands.charge_stripe.stripe.Charge.create")
def test_already_charged(self, charge_create_mocked):
charge_create_mocked.return_value = stripe.Charge(id="AA1")
self.charge.charge()
self.assertTrue(self.charge.is_charged)
self.assertTrue(self.success_signal_was_called)
self.assertFalse(self.exception_signal_was_called)
self._reset_signals()
with self.assertRaises(StripeMethodNotAllowed) as ctx:
self.charge.charge()
self.assertEqual(ctx.exception.args[0], "Already charged.")
self.assertTrue(self.charge.is_charged)
self.assertFalse(self.success_signal_was_called)
self.assertFalse(self.exception_signal_was_called)
@mock.patch(
"aa_stripe.management.commands.charge_stripe.stripe.Charge.create")
def test_stripe_api_error(self, charge_create_mocked):
error_json = {
"error": {
"message": "An unknown error occurred",
"type": "api_error"
}
}
charge_create_mocked.side_effect = stripe.error.APIError(
message="An unknown error occurred",
json_body=error_json,
)
with self.assertRaises(StripeInternalError):
self.charge.charge()
assert not self.success_signal_was_called
assert not self.charge.is_charged
assert self.charge.charge_attempt_failed
assert self.charge.stripe_response == error_json
@mock.patch(
"aa_stripe.management.commands.charge_stripe.stripe.Refund.create")
def test_refund_on_not_charged(self, refund_create_mocked):
self.charge.refresh_from_db()
refund_create_mocked.return_value = stripe.Refund(id="R1")
self.assertTrue(
self.customer,
StripeCustomer.get_latest_active_customer_for_user(self.user))
with self.assertRaises(StripeMethodNotAllowed) as ctx:
self.charge.refund()
self.assertEqual(ctx.exception.args[0],
"Cannot refund not charged transaction.")
self.assertFalse(self.charge.is_refunded)
self.assertFalse(self.charge_refunded_signal_was_called)
@mock.patch(
"aa_stripe.management.commands.charge_stripe.stripe.Refund.create")
def test_already_refunded(self, refund_create_mocked):
refund_create_mocked.return_value = stripe.Refund(id="R1")
self.charge.is_charged = True
self.charge.is_refunded = True
with self.assertRaises(StripeMethodNotAllowed) as ctx:
self.charge.refund()
self.assertEqual(ctx.exception.args[0], "Already refunded.")
self.assertTrue(self.charge.is_refunded)
self.assertFalse(self.charge_refunded_signal_was_called)
@mock.patch(
"aa_stripe.management.commands.charge_stripe.stripe.Refund.create")
def test_full_refund(self, refund_create_mocked):
refund_create_mocked.return_value = stripe.Refund(id="R1")
self.charge.is_charged = True
self.charge.refund()
self.assertTrue(self.charge.is_refunded)
self.assertEqual(self.charge.stripe_refund_id, "R1")
self.assertEqual(self.charge.amount_refunded, self.charge.amount)
self.assertTrue(self.charge_refunded_signal_was_called)
@mock.patch(
"aa_stripe.management.commands.charge_stripe.stripe.Refund.create")
def test_partial_refund(self, refund_create_mocked):
refund_create_mocked.return_value = stripe.Refund(id="R1")
self.charge.is_charged = True
with mock.patch("aa_stripe.signals.stripe_charge_refunded.send"
) as refund_signal_send:
to_refund = 30
# first refund
self.charge.refund(to_refund)
refund_create_mocked.assert_called_with(
charge=self.charge.stripe_charge_id,
amount=to_refund,
idempotency_key="{}-{}-{}-{}".format(
self.charge.object_id, self.charge.content_type_id, 0,
to_refund),
)
self.assertFalse(self.charge.is_refunded)
refund_signal_send.assert_called_with(sender=StripeCharge,
instance=self.charge)
# second refund
self._reset_signals()
refund_create_mocked.reset_mock()
self.charge.refund(to_refund)
refund_create_mocked.assert_called_with(
charge=self.charge.stripe_charge_id,
amount=to_refund,
idempotency_key="{}-{}-{}-{}".format(
self.charge.object_id, self.charge.content_type_id, 30,
to_refund),
)
self.assertFalse(self.charge.is_refunded)
refund_signal_send.assert_called_with(sender=StripeCharge,
instance=self.charge)
self.assertEqual(self.charge.amount_refunded, 60)
@mock.patch(
"aa_stripe.management.commands.charge_stripe.stripe.Refund.create")
def test_full_refund_of_partials(self, refund_create_mocked):
refund_create_mocked.return_value = stripe.Refund(id="R1")
self.charge.is_charged = True
with mock.patch("aa_stripe.signals.stripe_charge_refunded.send"
) as refund_signal_send:
to_refund = 50
# first refund
self.charge.refund(to_refund)
refund_create_mocked.assert_called_with(
charge=self.charge.stripe_charge_id,
amount=to_refund,
idempotency_key="{}-{}-{}-{}".format(
self.charge.object_id, self.charge.content_type_id, 0,
to_refund),
)
self.assertFalse(self.charge.is_refunded)
refund_signal_send.assert_called_with(sender=StripeCharge,
instance=self.charge)
# second refund
refund_create_mocked.reset_mock()
self.charge.refund(to_refund)
refund_create_mocked.assert_called_with(
charge=self.charge.stripe_charge_id,
amount=to_refund,
idempotency_key="{}-{}-{}-{}".format(
self.charge.object_id, self.charge.content_type_id, 50,
to_refund),
)
self.assertTrue(self.charge.is_refunded)
refund_signal_send.assert_called_with(sender=StripeCharge,
instance=self.charge)
self.assertEqual(self.charge.amount_refunded, 100)
@mock.patch(
"aa_stripe.management.commands.charge_stripe.stripe.Refund.create")
def test_refund_over_charge_amount(self, refund_create_mocked):
refund_create_mocked.return_value = stripe.Refund(id="R1")
self.charge.is_charged = True
with self.assertRaises(StripeMethodNotAllowed) as ctx:
self.charge.refund(101)
self.assertEqual(ctx.exception.args[0], "Refunds exceed charge")
self.assertFalse(self.charge.is_refunded)
self.assertFalse(self.charge_refunded_signal_was_called)
@mock.patch(
"aa_stripe.management.commands.charge_stripe.stripe.Refund.create",
side_effect=[
stripe.error.InvalidRequestError("message", "param"),
stripe.Refund(id="R1"),
],
)
@mock.patch(
"aa_stripe.management.commands.charge_stripe.stripe.Charge.retrieve",
return_value=build_small_manual_refunded_charge(),
)
def test_refund_after_partial_manual_refund(self, refund_create_mocked,
charge_retrieve_mocked):
self.charge.is_charged = True
self.charge.stripe_charge_id = "AA1"
self.charge.refund(100)
self.assertEqual(self.charge.amount_refunded, 100)
@mock.patch(
"aa_stripe.management.commands.charge_stripe.stripe.Refund.create",
side_effect=stripe.error.InvalidRequestError(
"message", "param", code="charge_already_refunded"),
)
@mock.patch(
"aa_stripe.management.commands.charge_stripe.stripe.Charge.retrieve",
return_value=build_full_manual_refunded_charge(),
)
def test_already_manually_refunded(self, refund_create_mocked,
charge_retrieve_mocked):
self.charge.is_charged = True
self.charge.stripe_charge_id = "AA1"
self.charge.refund(100)
self.assertEqual(self.charge.amount_refunded, 100)
