forked from CGM_Public/pretix_original
259 lines
7.4 KiB
Python
259 lines
7.4 KiB
Python
import json
|
|
from datetime import timedelta
|
|
from decimal import Decimal
|
|
|
|
import pytest
|
|
from django.test import RequestFactory
|
|
from django.utils.timezone import now
|
|
from stripe.error import APIConnectionError, CardError, StripeError
|
|
|
|
from pretix.base.models import Event, Order, OrderRefund, Organizer
|
|
from pretix.base.payment import PaymentException
|
|
from pretix.plugins.stripe.payment import StripeCC
|
|
|
|
|
|
@pytest.fixture
|
|
def env():
|
|
o = Organizer.objects.create(name='Dummy', slug='dummy')
|
|
event = Event.objects.create(
|
|
organizer=o, name='Dummy', slug='dummy',
|
|
date_from=now(), live=True
|
|
)
|
|
o1 = Order.objects.create(
|
|
code='FOOBAR', event=event, email='dummy@dummy.test',
|
|
status=Order.STATUS_PENDING,
|
|
datetime=now(), expires=now() + timedelta(days=10),
|
|
total=Decimal('13.37')
|
|
)
|
|
return event, o1
|
|
|
|
|
|
@pytest.fixture(autouse=True)
|
|
def no_messages(monkeypatch):
|
|
# Patch out template rendering for performance improvements
|
|
monkeypatch.setattr("django.contrib.messages.api.add_message", lambda *args, **kwargs: None)
|
|
|
|
|
|
@pytest.fixture
|
|
def factory():
|
|
return RequestFactory()
|
|
|
|
|
|
class MockedRefunds():
|
|
pass
|
|
|
|
|
|
class MockedCharge():
|
|
def __init__(self):
|
|
self.status = ''
|
|
self.paid = False
|
|
self.id = 'ch_123345345'
|
|
self.refunds = MockedRefunds()
|
|
|
|
def refresh(self):
|
|
pass
|
|
|
|
|
|
@pytest.mark.django_db
|
|
def test_perform_success(env, factory, monkeypatch):
|
|
event, order = env
|
|
|
|
def charge_create(**kwargs):
|
|
assert kwargs['amount'] == 1337
|
|
assert kwargs['currency'] == 'eur'
|
|
assert kwargs['source'] == 'tok_189fTT2eZvKYlo2CvJKzEzeu'
|
|
c = MockedCharge()
|
|
c.status = 'succeeded'
|
|
c.paid = True
|
|
return c
|
|
|
|
monkeypatch.setattr("stripe.Charge.create", charge_create)
|
|
prov = StripeCC(event)
|
|
req = factory.post('/', {
|
|
'stripe_token': 'tok_189fTT2eZvKYlo2CvJKzEzeu',
|
|
'stripe_last4': '4242',
|
|
'stripe_brand': 'Visa'
|
|
})
|
|
req.session = {}
|
|
prov.checkout_prepare(req, {})
|
|
assert 'payment_stripe_token' in req.session
|
|
payment = order.payments.create(
|
|
provider='stripe_cc', amount=order.total
|
|
)
|
|
prov.execute_payment(req, payment)
|
|
order.refresh_from_db()
|
|
assert order.status == Order.STATUS_PAID
|
|
|
|
|
|
@pytest.mark.django_db
|
|
def test_perform_success_zero_decimal_currency(env, factory, monkeypatch):
|
|
event, order = env
|
|
event.currency = 'JPY'
|
|
event.save()
|
|
|
|
def charge_create(**kwargs):
|
|
assert kwargs['amount'] == 13
|
|
assert kwargs['currency'] == 'jpy'
|
|
assert kwargs['source'] == 'tok_189fTT2eZvKYlo2CvJKzEzeu'
|
|
c = MockedCharge()
|
|
c.status = 'succeeded'
|
|
c.paid = True
|
|
return c
|
|
|
|
monkeypatch.setattr("stripe.Charge.create", charge_create)
|
|
prov = StripeCC(event)
|
|
req = factory.post('/', {
|
|
'stripe_token': 'tok_189fTT2eZvKYlo2CvJKzEzeu',
|
|
'stripe_last4': '4242',
|
|
'stripe_brand': 'Visa'
|
|
})
|
|
req.session = {}
|
|
prov.checkout_prepare(req, {})
|
|
assert 'payment_stripe_token' in req.session
|
|
payment = order.payments.create(
|
|
provider='stripe_cc', amount=order.total
|
|
)
|
|
prov.execute_payment(req, payment)
|
|
order.refresh_from_db()
|
|
assert order.status == Order.STATUS_PAID
|
|
|
|
|
|
@pytest.mark.django_db
|
|
def test_perform_card_error(env, factory, monkeypatch):
|
|
event, order = env
|
|
|
|
def charge_create(**kwargs):
|
|
raise CardError(message='Foo', param='foo', code=100)
|
|
|
|
monkeypatch.setattr("stripe.Charge.create", charge_create)
|
|
prov = StripeCC(event)
|
|
req = factory.post('/', {
|
|
'stripe_token': 'tok_189fTT2eZvKYlo2CvJKzEzeu',
|
|
'stripe_last4': '4242',
|
|
'stripe_brand': 'Visa'
|
|
})
|
|
req.session = {}
|
|
prov.checkout_prepare(req, {})
|
|
assert 'payment_stripe_token' in req.session
|
|
with pytest.raises(PaymentException):
|
|
payment = order.payments.create(
|
|
provider='stripe_cc', amount=order.total
|
|
)
|
|
prov.execute_payment(req, payment)
|
|
order.refresh_from_db()
|
|
assert order.status == Order.STATUS_PENDING
|
|
|
|
|
|
@pytest.mark.django_db
|
|
def test_perform_stripe_error(env, factory, monkeypatch):
|
|
event, order = env
|
|
|
|
def charge_create(**kwargs):
|
|
raise StripeError(message='Foo')
|
|
|
|
monkeypatch.setattr("stripe.Charge.create", charge_create)
|
|
prov = StripeCC(event)
|
|
req = factory.post('/', {
|
|
'stripe_token': 'tok_189fTT2eZvKYlo2CvJKzEzeu',
|
|
'stripe_last4': '4242',
|
|
'stripe_brand': 'Visa'
|
|
})
|
|
req.session = {}
|
|
prov.checkout_prepare(req, {})
|
|
assert 'payment_stripe_token' in req.session
|
|
with pytest.raises(PaymentException):
|
|
payment = order.payments.create(
|
|
provider='stripe_cc', amount=order.total
|
|
)
|
|
prov.execute_payment(req, payment)
|
|
order.refresh_from_db()
|
|
assert order.status == Order.STATUS_PENDING
|
|
|
|
|
|
@pytest.mark.django_db
|
|
def test_perform_failed(env, factory, monkeypatch):
|
|
event, order = env
|
|
|
|
def charge_create(**kwargs):
|
|
c = MockedCharge()
|
|
c.status = 'failed'
|
|
c.paid = True
|
|
c.failure_message = 'Foo'
|
|
return c
|
|
|
|
monkeypatch.setattr("stripe.Charge.create", charge_create)
|
|
prov = StripeCC(event)
|
|
req = factory.post('/', {
|
|
'stripe_token': 'tok_189fTT2eZvKYlo2CvJKzEzeu',
|
|
'stripe_last4': '4242',
|
|
'stripe_brand': 'Visa'
|
|
})
|
|
req.session = {}
|
|
prov.checkout_prepare(req, {})
|
|
assert 'payment_stripe_token' in req.session
|
|
with pytest.raises(PaymentException):
|
|
payment = order.payments.create(
|
|
provider='stripe_cc', amount=order.total
|
|
)
|
|
prov.execute_payment(req, payment)
|
|
order.refresh_from_db()
|
|
assert order.status == Order.STATUS_PENDING
|
|
|
|
|
|
@pytest.mark.django_db
|
|
def test_refund_success(env, factory, monkeypatch):
|
|
event, order = env
|
|
|
|
def charge_retr(*args, **kwargs):
|
|
def refund_create(amount):
|
|
r = MockedCharge()
|
|
r.id = 'foo'
|
|
r.status = 'succeeded'
|
|
return r
|
|
|
|
c = MockedCharge()
|
|
c.refunds.create = refund_create
|
|
return c
|
|
|
|
monkeypatch.setattr("stripe.Charge.retrieve", charge_retr)
|
|
order.status = Order.STATUS_PAID
|
|
p = order.payments.create(provider='stripe_cc', amount=order.total, info=json.dumps({
|
|
'id': 'ch_123345345'
|
|
}))
|
|
order.save()
|
|
prov = StripeCC(event)
|
|
refund = order.refunds.create(
|
|
provider='stripe_cc', amount=order.total, payment=p,
|
|
)
|
|
prov.execute_refund(refund)
|
|
refund.refresh_from_db()
|
|
assert refund.state == OrderRefund.REFUND_STATE_DONE
|
|
|
|
|
|
@pytest.mark.django_db
|
|
def test_refund_unavailable(env, factory, monkeypatch):
|
|
event, order = env
|
|
|
|
def charge_retr(*args, **kwargs):
|
|
def refund_create(amount):
|
|
raise APIConnectionError(message='Foo')
|
|
|
|
c = MockedCharge()
|
|
c.refunds.create = refund_create
|
|
return c
|
|
|
|
monkeypatch.setattr("stripe.Charge.retrieve", charge_retr)
|
|
order.status = Order.STATUS_PAID
|
|
p = order.payments.create(provider='stripe_cc', amount=order.total, info=json.dumps({
|
|
'id': 'ch_123345345'
|
|
}))
|
|
order.save()
|
|
prov = StripeCC(event)
|
|
refund = order.refunds.create(
|
|
provider='stripe_cc', amount=order.total, payment=p
|
|
)
|
|
with pytest.raises(PaymentException):
|
|
prov.execute_refund(refund)
|
|
refund.refresh_from_db()
|
|
assert refund.state != OrderRefund.REFUND_STATE_DONE
|