import datetime from decimal import Decimal import pytest from django.utils.timezone import now from pretix.base.models import Event, Item, Order, OrderPosition, Organizer from pretix.plugins.checkinlists.exporters import CSVCheckinList @pytest.fixture def event(): """Returns an event instance""" o = Organizer.objects.create(name='Dummy', slug='dummy') event = Event.objects.create( organizer=o, name='Dummy', slug='dummy', date_from=now(), plugins='pretix.plugins.checkinlists,tests.testdummy', ) event.settings.set('attendee_names_asked', True) event.settings.set('name_scheme', 'title_given_middle_family') event.settings.set('locales', ['en', 'de']) event.checkin_lists.create(name="Default", all_products=True) order_paid = Order.objects.create( code='FOO', event=event, email='dummy@dummy.test', status=Order.STATUS_PAID, datetime=now(), expires=now() + datetime.timedelta(days=10), total=33, locale='en' ) item_ticket = Item.objects.create(event=event, name="Ticket", default_price=23, admission=True) OrderPosition.objects.create( order=order_paid, item=item_ticket, variation=None, price=Decimal("23"), attendee_name_parts={"title": "Mr", "given_name": "Peter", "middle_name": "A", "family_name": "Jones"}, secret='hutjztuxhkbtwnesv2suqv26k6ttytxx' ) OrderPosition.objects.create( order=order_paid, item=item_ticket, variation=None, price=Decimal("13"), attendee_name_parts={"title": "Mrs", "given_name": "Andrea", "middle_name": "J", "family_name": "Zulu"}, secret='ggsngqtnmhx74jswjngw3fk8pfwz2a7k' ) return event def clean(d): return d.replace("\r", "").replace("\n", "") @pytest.mark.django_db def test_csv_simple(event): c = CSVCheckinList(event) _, _, content = c.render({ 'list': event.checkin_lists.first().pk, 'secrets': True, 'sort': 'name', 'questions': [] }) assert clean(content.decode()) == clean(""""Order code","Attendee name","Attendee name: Title","Attendee name: First name","Attendee name: Middle name","Attendee name: Family name","Product","Price","Checked in","Secret", "E-mail","Company","Voucher code" "FOO","Mr Peter A Jones","Mr","Peter","A","Jones","Ticket","23.00","","hutjztuxhkbtwnesv2suqv26k6ttytxx", "dummy@dummy.test","","" "FOO","Mrs Andrea J Zulu","Mrs","Andrea","J","Zulu","Ticket","13.00","","ggsngqtnmhx74jswjngw3fk8pfwz2a7k", "dummy@dummy.test","","" """) @pytest.mark.django_db def test_csv_order_by_name_parts(event): # noqa from django.conf import settings if not settings.JSON_FIELD_AVAILABLE: raise pytest.skip("Not supported on this database") c = CSVCheckinList(event) _, _, content = c.render({ 'list': event.checkin_lists.first().pk, 'secrets': True, 'sort': 'name:given_name', 'questions': [] }) assert clean(content.decode()) == clean(""""Order code","Attendee name","Attendee name: Title", "Attendee name: First name","Attendee name: Middle name","Attendee name: Family name","Product","Price", "Checked in","Secret","E-mail","Company","Voucher code" "FOO","Mrs Andrea J Zulu","Mrs","Andrea","J","Zulu","Ticket","13.00","","ggsngqtnmhx74jswjngw3fk8pfwz2a7k", "dummy@dummy.test","","" "FOO","Mr Peter A Jones","Mr","Peter","A","Jones","Ticket","23.00","","hutjztuxhkbtwnesv2suqv26k6ttytxx", "dummy@dummy.test","","" """) c = CSVCheckinList(event) _, _, content = c.render({ 'list': event.checkin_lists.first().pk, 'secrets': True, 'sort': 'name:family_name', 'questions': [] }) assert clean(content.decode()) == clean(""""Order code","Attendee name","Attendee name: Title", "Attendee name: First name","Attendee name: Middle name","Attendee name: Family name","Product","Price", "Checked in","Secret","E-mail","Company","Voucher code" "FOO","Mr Peter A Jones","Mr","Peter","A","Jones","Ticket","23.00","","hutjztuxhkbtwnesv2suqv26k6ttytxx", "dummy@dummy.test","","" "FOO","Mrs Andrea J Zulu","Mrs","Andrea","J","Zulu","Ticket","13.00","","ggsngqtnmhx74jswjngw3fk8pfwz2a7k", "dummy@dummy.test","","" """)