Files
pretix_original/src/tests/control/test_vouchers.py
Raphael Michel 8123effa65 Add sub-events and relative date settings (#503)
* Data model

* little crud

* SubEventItemForm etc

* Drop SubEventItem.active, quota editor

* Fix failing tests

* First frontend stuff

* Addons form stuff

* Quota calculation

* net price display on EventIndex

* Add tests, solve some bugs

* Correct quota selection in more places, consolidate pricing logic

* Fix failing quota tests

* Fix TypeError

* Add tests for checkout

* Fixed a bug in QuotaForm

* Prevent immutable cart if a quota was removed from an item

* Add tests for pricing

* Handle waiting list

* Filter in check-in list

* Fixed import lost in rebase

* Fix waiting list widget

* Voucher management

* Voucher redemption

* Fix broken tests

* Add subevents to OrderChangeManager

* Create a subevent during event creation

* Fix bulk voucher creation

* Introduce subevent.active

* Copy from for subevents

* Show active in list

* ICal download for subevents

* Check start and end of presale

* Failing tests / show cart logic

* Test

* Rebase migrations

* REST API integration of sub-events

* Integrate quota calculation into the traditional quota form

* Make subevent argument to add_position optional

* Log-display foo

* pretixdroid and subevents

* Filter by subevent

* Add more tests

* Some mor tests

* Rebase fixes

* More tests

* Relative dates

* Restrict selection in relative datetime widgets

* Filter subevent list

* Re-label has_subevents

* Rebase fixes, subevents in calendar view

* Performance and caching issues

* Refactor calendar templates

* Permission tests

* Calendar fixes and month selection

* subevent selection

* Rename subevents to dates

* Add tests for calendar views
2017-07-11 13:56:00 +02:00

478 lines
20 KiB
Python

import datetime
import json
from django.utils.timezone import now
from tests.base import SoupTest, extract_form_fields
from pretix.base.models import (
Event, Item, ItemVariation, Organizer, Quota, Team, User, Voucher,
)
class VoucherFormTest(SoupTest):
def setUp(self):
super().setUp()
self.user = User.objects.create_user('dummy@dummy.dummy', 'dummy')
self.orga = Organizer.objects.create(name='CCC', slug='ccc')
self.event = Event.objects.create(
organizer=self.orga, name='30C3', slug='30c3',
date_from=datetime.datetime(2013, 12, 26, tzinfo=datetime.timezone.utc),
)
t = Team.objects.create(organizer=self.orga, can_view_vouchers=True, can_change_vouchers=True)
t.members.add(self.user)
t.limit_events.add(self.event)
self.client.login(email='dummy@dummy.dummy', password='dummy')
self.quota_shirts = Quota.objects.create(event=self.event, name='Shirts', size=2)
self.shirt = Item.objects.create(event=self.event, name='T-Shirt', default_price=12)
self.quota_shirts.items.add(self.shirt)
self.shirt_red = ItemVariation.objects.create(item=self.shirt, default_price=14, value='Red')
self.shirt_blue = ItemVariation.objects.create(item=self.shirt, value='Blue')
self.quota_shirts.variations.add(self.shirt_red)
self.quota_shirts.variations.add(self.shirt_blue)
self.quota_tickets = Quota.objects.create(event=self.event, name='Tickets', size=5)
self.ticket = Item.objects.create(event=self.event, name='Early-bird ticket',
default_price=23)
self.quota_tickets.items.add(self.ticket)
def _create_voucher(self, data, expected_failure=False):
count_before = self.event.vouchers.count()
doc = self.get_doc('/control/event/%s/%s/vouchers/add' % (self.orga.slug, self.event.slug))
form_data = extract_form_fields(doc.select('.container-fluid form')[0])
form_data.update(data)
doc = self.post_doc('/control/event/%s/%s/vouchers/add' % (self.orga.slug, self.event.slug), form_data)
if expected_failure:
assert doc.select(".alert-danger")
assert count_before == self.event.vouchers.count()
else:
assert doc.select(".alert-success")
assert count_before + 1 == self.event.vouchers.count()
def _create_bulk_vouchers(self, data, expected_failure=False):
count_before = self.event.vouchers.count()
doc = self.get_doc('/control/event/%s/%s/vouchers/bulk_add' % (self.orga.slug, self.event.slug))
form_data = extract_form_fields(doc.select('.container-fluid form')[0])
form_data.update(data)
doc = self.post_doc('/control/event/%s/%s/vouchers/bulk_add' % (self.orga.slug, self.event.slug), form_data)
if expected_failure:
assert doc.select(".alert-danger")
assert count_before == self.event.vouchers.count()
else:
assert doc.select(".alert-success")
assert count_before + len(form_data.get('codes').split("\n")) == self.event.vouchers.count()
def _change_voucher(self, v, data, expected_failure=False):
doc = self.get_doc('/control/event/%s/%s/vouchers/%s/' % (self.orga.slug, self.event.slug, v.pk))
form_data = extract_form_fields(doc.select('.container-fluid form')[0])
form_data.update(data)
doc = self.post_doc('/control/event/%s/%s/vouchers/%s/' % (self.orga.slug, self.event.slug, v.pk), form_data)
if expected_failure:
assert doc.select(".alert-danger")
else:
assert doc.select(".alert-success")
def test_list(self):
self.event.vouchers.create(item=self.ticket, code='ABCDEFG')
doc = self.client.get('/control/event/%s/%s/vouchers/' % (self.orga.slug, self.event.slug))
assert 'ABCDEFG' in doc.rendered_content
def test_csv(self):
self.event.vouchers.create(item=self.ticket, code='ABCDEFG')
doc = self.client.get('/control/event/%s/%s/vouchers/?download=yes' % (self.orga.slug, self.event.slug))
assert doc.content.strip() == '"Voucher code","Valid until","Product","Reserve quota","Bypass quota",' \
'"Price effect","Value","Tag","Redeemed","Maximum usages"\r\n"ABCDEFG","",' \
'"Early-bird ticket","No","No","No effect","","","0","1"'.encode('utf-8')
def test_filter_status_valid(self):
v = self.event.vouchers.create(item=self.ticket)
doc = self.client.get('/control/event/%s/%s/vouchers/?status=v' % (self.orga.slug, self.event.slug))
assert v.code in doc.rendered_content
v.redeemed = 1
v.save()
doc = self.client.get('/control/event/%s/%s/vouchers/?status=v' % (self.orga.slug, self.event.slug))
assert v.code not in doc.rendered_content
def test_filter_status_redeemed(self):
v = self.event.vouchers.create(item=self.ticket, redeemed=1)
doc = self.client.get('/control/event/%s/%s/vouchers/?status=r' % (self.orga.slug, self.event.slug))
assert v.code in doc.rendered_content
v.redeemed = 0
v.save()
doc = self.client.get('/control/event/%s/%s/vouchers/?status=r' % (self.orga.slug, self.event.slug))
assert v.code not in doc.rendered_content
def test_filter_status_expired(self):
v = self.event.vouchers.create(item=self.ticket, valid_until=now() + datetime.timedelta(days=1))
doc = self.client.get('/control/event/%s/%s/vouchers/?status=e' % (self.orga.slug, self.event.slug))
assert v.code not in doc.rendered_content
v.valid_until = now() - datetime.timedelta(days=1)
v.save()
doc = self.client.get('/control/event/%s/%s/vouchers/?status=e' % (self.orga.slug, self.event.slug))
assert v.code in doc.rendered_content
def test_filter_tag(self):
self.event.vouchers.create(item=self.ticket, code='ABCDEFG', comment='Foo', tag='bar')
doc = self.client.get('/control/event/%s/%s/vouchers/?tag=bar' % (self.orga.slug, self.event.slug))
assert 'ABCDEFG' in doc.rendered_content
doc = self.client.get('/control/event/%s/%s/vouchers/?tag=baz' % (self.orga.slug, self.event.slug))
assert 'ABCDEFG' not in doc.rendered_content
def test_search_code(self):
self.event.vouchers.create(item=self.ticket, code='ABCDEFG', comment='Foo')
doc = self.client.get('/control/event/%s/%s/vouchers/?search=ABCDEFG' % (self.orga.slug, self.event.slug))
assert 'ABCDEFG' in doc.rendered_content
doc = self.client.get('/control/event/%s/%s/vouchers/?search=Foo' % (self.orga.slug, self.event.slug))
assert 'ABCDEFG' in doc.rendered_content
doc = self.client.get('/control/event/%s/%s/vouchers/?search=12345' % (self.orga.slug, self.event.slug))
assert 'ABCDEFG' not in doc.rendered_content
def test_bulk_rng(self):
rng = self.client.get('/control/event/%s/%s/vouchers/rng?num=7' % (self.orga.slug, self.event.slug))
codes = json.loads(rng.content.decode('utf-8'))['codes']
assert len(codes) == 7
assert all([len(r) == 16 for r in codes])
def test_display_voucher_code(self):
count_before = self.event.vouchers.count()
doc = self.get_doc('/control/event/%s/%s/vouchers/add' % (self.orga.slug, self.event.slug))
form_data = extract_form_fields(doc.select('.container-fluid form')[0])
form_data.update({
'itemvar': '%d' % self.ticket.pk
})
doc = self.post_doc('/control/event/%s/%s/vouchers/add' % (self.orga.slug, self.event.slug), form_data)
v = Voucher.objects.latest('pk')
assert v.code in doc.select(".alert-success")[0].text
assert count_before + 1 == self.event.vouchers.count()
def test_create_non_blocking_item_voucher(self):
self._create_voucher({
'itemvar': '%d' % self.ticket.pk
})
v = Voucher.objects.latest('pk')
assert not v.block_quota
assert v.item.pk == self.ticket.pk
assert v.variation is None
assert v.quota is None
def test_create_non_blocking_variation_voucher(self):
self._create_voucher({
'itemvar': '%d-%d' % (self.shirt.pk, self.shirt_red.pk)
})
v = Voucher.objects.latest('pk')
assert not v.block_quota
assert v.item.pk == self.shirt.pk
assert v.variation.pk == self.shirt_red.pk
assert v.quota is None
def test_create_non_blocking_quota_voucher(self):
self._create_voucher({
'itemvar': 'q-%d' % self.quota_tickets.pk
})
v = Voucher.objects.latest('pk')
assert not v.block_quota
assert v.item is None
assert v.variation is None
assert v.quota.pk == self.quota_tickets.pk
def test_create_blocking_item_voucher_quota_free(self):
self._create_voucher({
'itemvar': '%d' % self.ticket.pk,
'block_quota': 'on'
})
v = Voucher.objects.latest('pk')
assert v.block_quota
def test_create_blocking_item_voucher_quota_full(self):
self._create_voucher({
'itemvar': '%d' % self.shirt.pk,
'block_quota': 'on'
}, expected_failure=True)
def test_create_blocking_item_voucher_quota_full_invalid(self):
self.quota_shirts.size = 0
self.quota_shirts.save()
self._create_voucher({
'itemvar': '%d-%d' % (self.shirt.pk, self.shirt_red.pk),
'block_quota': 'on',
'valid_until': (now() - datetime.timedelta(days=3)).strftime('%Y-%m-%d %H:%M:%S')
})
def test_create_blocking_variation_voucher_quota_free(self):
self._create_voucher({
'itemvar': '%d-%d' % (self.shirt.pk, self.shirt_red.pk),
'block_quota': 'on'
})
v = Voucher.objects.latest('pk')
assert v.block_quota
def test_create_blocking_variation_voucher_quota_full(self):
self.quota_shirts.size = 0
self.quota_shirts.save()
self._create_voucher({
'itemvar': '%d-%d' % (self.shirt.pk, self.shirt_red.pk),
'block_quota': 'on'
}, expected_failure=True)
def test_create_blocking_quota_voucher_quota_free(self):
self._create_voucher({
'itemvar': 'q-%d' % self.quota_tickets.pk,
'block_quota': 'on'
})
v = Voucher.objects.latest('pk')
assert v.block_quota
def test_create_blocking_quota_voucher_quota_full(self):
self.quota_tickets.size = 0
self.quota_tickets.save()
self._create_voucher({
'itemvar': 'q-%d' % self.quota_tickets.pk,
'block_quota': 'on'
}, expected_failure=True)
def test_change_non_blocking_voucher(self):
v = self.event.vouchers.create(item=self.ticket)
self._change_voucher(v, {
'itemvar': 'q-%d' % self.quota_tickets.pk
})
v.refresh_from_db()
assert v.item is None
assert v.variation is None
assert v.quota.pk == self.quota_tickets.pk
def test_change_blocking_voucher_unchanged_quota_full(self):
self.quota_tickets.size = 0
self.quota_tickets.save()
v = self.event.vouchers.create(item=self.ticket, block_quota=True)
self._change_voucher(v, {
})
v.refresh_from_db()
assert v.block_quota
def test_change_voucher_to_blocking_quota_full(self):
self.quota_tickets.size = 0
self.quota_tickets.save()
v = self.event.vouchers.create(item=self.ticket)
self._change_voucher(v, {
'block_quota': 'on'
}, expected_failure=True)
def test_change_voucher_to_blocking_quota_free(self):
v = self.event.vouchers.create(item=self.ticket)
self._change_voucher(v, {
'block_quota': 'on'
})
v.refresh_from_db()
assert v.block_quota
def test_change_voucher_validity_to_valid_quota_full(self):
self.quota_tickets.size = 0
self.quota_tickets.save()
v = self.event.vouchers.create(item=self.ticket, valid_until=now() - datetime.timedelta(days=3),
block_quota=True)
self._change_voucher(v, {
'valid_until': (now() + datetime.timedelta(days=3)).strftime('%Y-%m-%d %H:%M:%S')
}, expected_failure=True)
v.refresh_from_db()
assert v.valid_until < now()
def test_change_voucher_validity_to_valid_quota_free(self):
v = self.event.vouchers.create(item=self.ticket, valid_until=now() - datetime.timedelta(days=3),
block_quota=True)
self._change_voucher(v, {
'valid_until': (now() + datetime.timedelta(days=3)).strftime('%Y-%m-%d %H:%M:%S')
})
v.refresh_from_db()
assert v.valid_until > now()
def test_change_item_of_blocking_voucher_quota_free(self):
ticket2 = Item.objects.create(event=self.event, name='Late-bird ticket', default_price=23)
self.quota_tickets.items.add(ticket2)
v = self.event.vouchers.create(item=self.ticket, block_quota=True)
self._change_voucher(v, {
'itemvar': '%d' % ticket2.pk,
})
def test_change_item_of_blocking_voucher_quota_full(self):
self.quota_shirts.size = 0
self.quota_shirts.save()
hoodie = Item.objects.create(event=self.event, name='Hoodie', default_price=23)
self.quota_shirts.items.add(hoodie)
v = self.event.vouchers.create(item=self.ticket, block_quota=True)
self._change_voucher(v, {
'itemvar': '%d' % hoodie.pk,
}, expected_failure=True)
def test_change_variation_of_blocking_voucher_quota_free(self):
self.quota_shirts.variations.remove(self.shirt_blue)
self.quota_tickets.variations.add(self.shirt_blue)
v = self.event.vouchers.create(item=self.shirt, variation=self.shirt_red, block_quota=True)
self._change_voucher(v, {
'itemvar': '%d-%d' % (self.shirt.pk, self.shirt_blue.pk),
})
def test_change_variation_of_blocking_voucher_quota_full(self):
self.quota_shirts.variations.remove(self.shirt_blue)
self.quota_tickets.variations.add(self.shirt_blue)
self.quota_tickets.size = 0
self.quota_tickets.save()
v = self.event.vouchers.create(item=self.shirt, variation=self.shirt_red, block_quota=True)
self._change_voucher(v, {
'itemvar': '%d-%d' % (self.shirt.pk, self.shirt_blue.pk),
}, expected_failure=True)
def test_change_quota_of_blocking_voucher_quota_free(self):
v = self.event.vouchers.create(quota=self.quota_tickets, block_quota=True)
self._change_voucher(v, {
'itemvar': 'q-%d' % self.quota_shirts.pk,
})
def test_change_quota_of_blocking_voucher_quota_full(self):
self.quota_shirts.size = 0
self.quota_shirts.save()
v = self.event.vouchers.create(quota=self.quota_tickets, block_quota=True)
self._change_voucher(v, {
'itemvar': 'q-%d' % self.quota_shirts.pk,
}, expected_failure=True)
def test_change_item_of_blocking_voucher_without_quota_change(self):
self.quota_tickets.size = 0
self.quota_tickets.save()
ticket2 = Item.objects.create(event=self.event, name='Standard Ticket', default_price=23)
self.quota_tickets.items.add(ticket2)
v = self.event.vouchers.create(item=self.ticket, block_quota=True)
self._change_voucher(v, {
'itemvar': '%d' % ticket2.pk,
})
def test_change_variation_of_blocking_voucher_without_quota_change(self):
self.quota_shirts.size = 0
self.quota_shirts.save()
v = self.event.vouchers.create(item=self.shirt, variation=self.shirt_red, block_quota=True)
self._change_voucher(v, {
'itemvar': '%d-%d' % (self.shirt.pk, self.shirt_blue.pk),
})
def test_create_duplicate_code(self):
v = self.event.vouchers.create(quota=self.quota_tickets)
self._create_voucher({
'code': v.code,
}, expected_failure=True)
def test_change_code_to_duplicate(self):
v1 = self.event.vouchers.create(quota=self.quota_tickets)
v2 = self.event.vouchers.create(quota=self.quota_tickets)
self._change_voucher(v1, {
'code': v2.code
}, expected_failure=True)
def test_create_bulk(self):
self._create_bulk_vouchers({
'codes': 'ABCDE\nDEFGH',
'itemvar': '%d' % self.shirt.pk,
})
def test_create_blocking_bulk_quota_full(self):
self.quota_tickets.size = 0
self.quota_tickets.save()
self._create_bulk_vouchers({
'codes': 'ABCDE\nDEFGH',
'itemvar': '%d' % self.ticket.pk,
'block_quota': 'on'
}, expected_failure=True)
def test_create_blocking_bulk_quota_free(self):
self.quota_tickets.size = 5
self.quota_tickets.save()
self._create_bulk_vouchers({
'codes': 'ABCDE\nDEFGH',
'itemvar': '%d' % self.ticket.pk,
'block_quota': 'on'
})
def test_create_blocking_bulk_quota_partial(self):
self.quota_tickets.size = 1
self.quota_tickets.save()
self._create_bulk_vouchers({
'codes': 'ABCDE\nDEFGH',
'itemvar': '%d' % self.ticket.pk,
'block_quota': 'on'
}, expected_failure=True)
def test_create_bulk_with_duplicate_code(self):
v = self.event.vouchers.create(quota=self.quota_tickets)
self._create_bulk_vouchers({
'codes': 'ABCDE\n%s' % v.code,
'itemvar': '%d' % self.shirt.pk,
}, expected_failure=True)
def test_delete_voucher(self):
v = self.event.vouchers.create(quota=self.quota_tickets)
doc = self.get_doc('/control/event/%s/%s/vouchers/%s/delete' % (self.orga.slug, self.event.slug, v.pk),
follow=True)
assert not doc.select(".alert-danger")
doc = self.post_doc('/control/event/%s/%s/vouchers/%s/delete' % (self.orga.slug, self.event.slug, v.pk),
{}, follow=True)
assert doc.select(".alert-success")
assert not self.event.vouchers.filter(pk=v.id).exists()
def test_delete_voucher_redeemed(self):
v = self.event.vouchers.create(quota=self.quota_tickets, redeemed=1)
doc = self.get_doc('/control/event/%s/%s/vouchers/%s/delete' % (self.orga.slug, self.event.slug, v.pk),
follow=True)
assert doc.select(".alert-danger")
doc = self.post_doc('/control/event/%s/%s/vouchers/%s/delete' % (self.orga.slug, self.event.slug, v.pk),
{}, follow=True)
assert doc.select(".alert-danger")
def test_subevent_optional(self):
self.event.has_subevents = True
self.event.save()
self._create_voucher({
'itemvar': '%d' % self.ticket.pk,
})
def test_subevent_required_for_blocking(self):
self.event.has_subevents = True
self.event.save()
self._create_voucher({
'itemvar': '%d' % self.ticket.pk,
'block_quota': 'on'
}, expected_failure=True)
def test_subevent_blocking_quota_free(self):
self.event.has_subevents = True
self.event.save()
se1 = self.event.subevents.create(name="Foo", date_from=now())
se2 = self.event.subevents.create(name="Bar", date_from=now())
self.quota_tickets.subevent = se1
self.quota_tickets.save()
q2 = Quota.objects.create(event=self.event, name='Tickets', size=0, subevent=se2)
q2.items.add(self.ticket)
self._create_voucher({
'itemvar': '%d' % self.ticket.pk,
'block_quota': 'on',
'subevent': se1.pk
})
def test_subevent_blocking_quota_full(self):
self.event.has_subevents = True
self.event.save()
se1 = self.event.subevents.create(name="Foo", date_from=now())
se2 = self.event.subevents.create(name="Bar", date_from=now())
self.quota_tickets.subevent = se1
self.quota_tickets.size = 0
self.quota_tickets.save()
q2 = Quota.objects.create(event=self.event, name='Tickets', size=5, subevent=se2)
q2.items.add(self.ticket)
self._create_voucher({
'itemvar': '%d' % self.ticket.pk,
'block_quota': 'on',
'subevent': se1.pk
}, expected_failure=True)