forked from CGM_Public/pretix_original
132 lines
4.6 KiB
Python
132 lines
4.6 KiB
Python
import copy
|
|
import json
|
|
from datetime import datetime, timedelta
|
|
from unittest import mock
|
|
|
|
import pytest
|
|
from django.utils.timezone import now
|
|
from pytz import UTC
|
|
|
|
from pretix.base.models import (
|
|
Event, Item, Order, OrderPosition, Organizer, Quota, Team, User,
|
|
)
|
|
from pretix.plugins.banktransfer.models import BankImportJob, BankTransaction
|
|
|
|
|
|
@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(), plugins='pretix.plugins.banktransfer'
|
|
)
|
|
user = User.objects.create_user('dummy@dummy.dummy', 'dummy')
|
|
t = Team.objects.create(organizer=event.organizer, can_view_orders=True, can_change_orders=True)
|
|
t.members.add(user)
|
|
t.limit_events.add(event)
|
|
o1 = Order.objects.create(
|
|
code='1Z3AS', event=event,
|
|
status=Order.STATUS_PENDING,
|
|
datetime=now(), expires=now() + timedelta(days=10),
|
|
total=23, payment_provider='banktransfer'
|
|
)
|
|
o2 = Order.objects.create(
|
|
code='6789Z', event=event,
|
|
status=Order.STATUS_CANCELED,
|
|
datetime=now(), expires=now() + timedelta(days=10),
|
|
total=23, payment_provider='banktransfer'
|
|
)
|
|
quota = Quota.objects.create(name="Test", size=2, event=event)
|
|
item1 = Item.objects.create(event=event, name="Ticket", default_price=23)
|
|
quota.items.add(item1)
|
|
OrderPosition.objects.create(order=o1, item=item1, variation=None, price=23)
|
|
return event, user, o1, o2
|
|
|
|
|
|
RES_JOB = {
|
|
'event': 'dummy',
|
|
'id': 1,
|
|
'transactions': [
|
|
{'comment': '',
|
|
'message': '',
|
|
'payer': 'Foo',
|
|
'reference': '',
|
|
'checksum': '',
|
|
'amount': '0.00',
|
|
'date': 'unknown',
|
|
'state': 'error',
|
|
'order': None
|
|
}
|
|
],
|
|
'created': '2017-06-27T09:13:35.785251Z',
|
|
'state': 'pending'
|
|
}
|
|
|
|
|
|
@pytest.mark.django_db
|
|
def test_api_list(env, client):
|
|
testtime = datetime(2017, 12, 1, 10, 0, 0, tzinfo=UTC)
|
|
|
|
with mock.patch('django.utils.timezone.now') as mock_now:
|
|
mock_now.return_value = testtime
|
|
job = BankImportJob.objects.create(event=env[0], organizer=env[0].organizer)
|
|
BankTransaction.objects.create(event=env[0], import_job=job, payer='Foo',
|
|
state=BankTransaction.STATE_ERROR,
|
|
amount=0, date='unknown')
|
|
res = copy.copy(RES_JOB)
|
|
res['id'] = job.pk
|
|
res['created'] = testtime.isoformat().replace('+00:00', 'Z')
|
|
client.login(email='dummy@dummy.dummy', password='dummy')
|
|
r = json.loads(
|
|
client.get('/api/v1/organizers/{}/bankimportjobs/'.format(env[0].organizer.slug)).content.decode('utf-8')
|
|
)
|
|
assert r['results'] == [res]
|
|
|
|
|
|
@pytest.mark.django_db
|
|
def test_api_detail(env, client):
|
|
testtime = datetime(2017, 12, 1, 10, 0, 0, tzinfo=UTC)
|
|
|
|
with mock.patch('django.utils.timezone.now') as mock_now:
|
|
mock_now.return_value = testtime
|
|
job = BankImportJob.objects.create(event=env[0], organizer=env[0].organizer)
|
|
BankTransaction.objects.create(event=env[0], import_job=job, payer='Foo',
|
|
state=BankTransaction.STATE_ERROR,
|
|
amount=0, date='unknown')
|
|
res = copy.copy(RES_JOB)
|
|
res['id'] = job.pk
|
|
res['created'] = testtime.isoformat().replace('+00:00', 'Z')
|
|
client.login(email='dummy@dummy.dummy', password='dummy')
|
|
r = json.loads(
|
|
client.get(
|
|
'/api/v1/organizers/{}/bankimportjobs/{}/'.format(env[0].organizer.slug, job.pk)
|
|
).content.decode('utf-8')
|
|
)
|
|
assert r == res
|
|
|
|
|
|
@pytest.mark.django_db(transaction=True)
|
|
def test_api_create(env, client):
|
|
client.login(email='dummy@dummy.dummy', password='dummy')
|
|
r = client.post(
|
|
'/api/v1/organizers/{}/bankimportjobs/'.format(env[0].organizer.slug), json.dumps({
|
|
'event': 'dummy',
|
|
'transactions': [
|
|
{
|
|
'payer': 'Foo',
|
|
'reference': 'DUMMY-1Z3AS',
|
|
'amount': '23.00',
|
|
'date': 'yesterday' # test bogus date format
|
|
}
|
|
]
|
|
}), content_type="application/json"
|
|
)
|
|
assert r.status_code == 201
|
|
rdata = json.loads(r.content.decode('utf-8'))
|
|
# This is only because we don't run celery in tests, otherwise it wouldn't be completed yet.
|
|
assert rdata['state'] == 'completed'
|
|
assert len(rdata['transactions']) == 1
|
|
assert rdata['transactions'][0]['checksum']
|
|
env[2].refresh_from_db()
|
|
assert env[2].status == Order.STATUS_PAID
|