import datetime from datetime import timedelta from decimal import Decimal from bs4 import BeautifulSoup from django.conf import settings from django.test import TestCase from django.utils.timezone import now from pretix.base.models import ( CartPosition, Event, Item, ItemCategory, Order, OrderPosition, Organizer, Question, Quota, Voucher, ) class CheckoutTestCase(TestCase): def setUp(self): super().setUp() 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), plugins='pretix.plugins.stripe,pretix.plugins.banktransfer', live=True ) self.category = ItemCategory.objects.create(event=self.event, name="Everything", position=0) 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', category=self.category, default_price=23, admission=True) self.quota_tickets.items.add(self.ticket) self.event.settings.set('attendee_names_asked', False) self.event.settings.set('payment_banktransfer__enabled', True) self.client.get('/%s/%s/' % (self.orga.slug, self.event.slug)) self.session_key = self.client.cookies.get(settings.SESSION_COOKIE_NAME).value self._set_session('email', 'admin@localhost') def test_empty_cart(self): response = self.client.get('/%s/%s/checkout/start' % (self.orga.slug, self.event.slug), follow=True) self.assertRedirects(response, '/%s/%s/' % (self.orga.slug, self.event.slug), target_status_code=200) def test_questions(self): q1 = Question.objects.create( event=self.event, question='Age', type=Question.TYPE_NUMBER, required=True ) q2 = Question.objects.create( event=self.event, question='How have you heard from us?', type=Question.TYPE_STRING, required=False ) self.ticket.questions.add(q1) self.ticket.questions.add(q2) cr1 = CartPosition.objects.create( event=self.event, cart_id=self.session_key, item=self.ticket, price=23, expires=now() + timedelta(minutes=10) ) cr2 = CartPosition.objects.create( event=self.event, cart_id=self.session_key, item=self.ticket, price=20, expires=now() + timedelta(minutes=10) ) response = self.client.get('/%s/%s/checkout/questions/' % (self.orga.slug, self.event.slug), follow=True) doc = BeautifulSoup(response.rendered_content) self.assertEqual(len(doc.select('input[name=%s-question_%s]' % (cr1.id, q1.id))), 1) self.assertEqual(len(doc.select('input[name=%s-question_%s]' % (cr2.id, q1.id))), 1) self.assertEqual(len(doc.select('input[name=%s-question_%s]' % (cr1.id, q2.id))), 1) self.assertEqual(len(doc.select('input[name=%s-question_%s]' % (cr2.id, q2.id))), 1) # Not all required fields filled out, expect failure response = self.client.post('/%s/%s/checkout/questions/' % (self.orga.slug, self.event.slug), { '%s-question_%s' % (cr1.id, q1.id): '42', '%s-question_%s' % (cr2.id, q1.id): '', '%s-question_%s' % (cr1.id, q2.id): 'Internet', '%s-question_%s' % (cr2.id, q2.id): '', 'email': 'admin@localhost' }, follow=True) doc = BeautifulSoup(response.rendered_content) self.assertGreaterEqual(len(doc.select('.has-error')), 1) # Corrected request response = self.client.post('/%s/%s/checkout/questions/' % (self.orga.slug, self.event.slug), { '%s-question_%s' % (cr1.id, q1.id): '42', '%s-question_%s' % (cr2.id, q1.id): '23', '%s-question_%s' % (cr1.id, q2.id): 'Internet', '%s-question_%s' % (cr2.id, q2.id): '', 'email': 'admin@localhost' }, follow=True) self.assertRedirects(response, '/%s/%s/checkout/payment/' % (self.orga.slug, self.event.slug), target_status_code=200) cr1 = CartPosition.objects.get(id=cr1.id) cr2 = CartPosition.objects.get(id=cr2.id) self.assertEqual(cr1.answers.filter(question=q1).count(), 1) self.assertEqual(cr2.answers.filter(question=q1).count(), 1) self.assertEqual(cr1.answers.filter(question=q2).count(), 1) self.assertFalse(cr2.answers.filter(question=q2).exists()) def test_attendee_name_required(self): self.event.settings.set('attendee_names_asked', True) self.event.settings.set('attendee_names_required', True) cr1 = CartPosition.objects.create( event=self.event, cart_id=self.session_key, item=self.ticket, price=23, expires=now() + timedelta(minutes=10) ) response = self.client.get('/%s/%s/checkout/questions/' % (self.orga.slug, self.event.slug), follow=True) doc = BeautifulSoup(response.rendered_content) self.assertEqual(len(doc.select('input[name=%s-attendee_name]' % cr1.id)), 1) # Not all required fields filled out, expect failure response = self.client.post('/%s/%s/checkout/questions/' % (self.orga.slug, self.event.slug), { '%s-attendee_name' % cr1.id: '', 'email': 'admin@localhost' }, follow=True) doc = BeautifulSoup(response.rendered_content) self.assertGreaterEqual(len(doc.select('.has-error')), 1) # Corrected request response = self.client.post('/%s/%s/checkout/questions/' % (self.orga.slug, self.event.slug), { '%s-attendee_name' % cr1.id: 'Peter', 'email': 'admin@localhost' }, follow=True) self.assertRedirects(response, '/%s/%s/checkout/payment/' % (self.orga.slug, self.event.slug), target_status_code=200) cr1 = CartPosition.objects.get(id=cr1.id) self.assertEqual(cr1.attendee_name, 'Peter') def test_attendee_name_optional(self): self.event.settings.set('attendee_names_asked', True) self.event.settings.set('attendee_names_required', False) cr1 = CartPosition.objects.create( event=self.event, cart_id=self.session_key, item=self.ticket, price=23, expires=now() + timedelta(minutes=10) ) response = self.client.get('/%s/%s/checkout/questions/' % (self.orga.slug, self.event.slug), follow=True) doc = BeautifulSoup(response.rendered_content) self.assertEqual(len(doc.select('input[name=%s-attendee_name]' % cr1.id)), 1) # Not all fields filled out, expect success response = self.client.post('/%s/%s/checkout/questions/' % (self.orga.slug, self.event.slug), { '%s-attendee_name' % cr1.id: '', 'email': 'admin@localhost' }, follow=True) self.assertRedirects(response, '/%s/%s/checkout/payment/' % (self.orga.slug, self.event.slug), target_status_code=200) cr1 = CartPosition.objects.get(id=cr1.id) self.assertIsNone(cr1.attendee_name) def test_payment(self): # TODO: Test for correct payment method fees self.event.settings.set('payment_stripe__enabled', True) self.event.settings.set('payment_banktransfer__enabled', True) CartPosition.objects.create( event=self.event, cart_id=self.session_key, item=self.ticket, price=23, expires=now() + timedelta(minutes=10) ) response = self.client.get('/%s/%s/checkout/payment/' % (self.orga.slug, self.event.slug), follow=True) doc = BeautifulSoup(response.rendered_content) self.assertEqual(len(doc.select('input[name=payment]')), 2) response = self.client.post('/%s/%s/checkout/payment/' % (self.orga.slug, self.event.slug), { 'payment': 'banktransfer' }, follow=True) self.assertRedirects(response, '/%s/%s/checkout/confirm/' % (self.orga.slug, self.event.slug), target_status_code=200) def test_premature_confirm(self): response = self.client.get('/%s/%s/checkout/confirm/' % (self.orga.slug, self.event.slug), follow=True) self.assertRedirects(response, '/%s/%s/' % (self.orga.slug, self.event.slug), target_status_code=200) self.event.settings.set('payment_stripe__enabled', True) self.event.settings.set('payment_banktransfer__enabled', True) cr1 = CartPosition.objects.create( event=self.event, cart_id=self.session_key, item=self.ticket, price=23, expires=now() + timedelta(minutes=10) ) response = self.client.get('/%s/%s/checkout/confirm/' % (self.orga.slug, self.event.slug), follow=True) self.assertRedirects(response, '/%s/%s/checkout/payment/' % (self.orga.slug, self.event.slug), target_status_code=200) self._set_session('payment', 'banktransfer') self.event.settings.set('attendee_names_asked', True) self.event.settings.set('attendee_names_required', True) response = self.client.get('/%s/%s/checkout/confirm/' % (self.orga.slug, self.event.slug), follow=True) self.assertRedirects(response, '/%s/%s/checkout/questions/' % (self.orga.slug, self.event.slug), target_status_code=200) cr1.attendee_name = 'Peter' cr1.save() q1 = Question.objects.create( event=self.event, question='Age', type=Question.TYPE_NUMBER, required=True ) self.ticket.questions.add(q1) response = self.client.get('/%s/%s/checkout/confirm/' % (self.orga.slug, self.event.slug), follow=True) self.assertRedirects(response, '/%s/%s/checkout/questions/' % (self.orga.slug, self.event.slug), target_status_code=200) q1.required = False q1.save() response = self.client.get('/%s/%s/checkout/confirm/' % (self.orga.slug, self.event.slug), follow=True) self.assertEqual(response.status_code, 200) self._set_session('email', 'invalid') response = self.client.get('/%s/%s/checkout/confirm/' % (self.orga.slug, self.event.slug), follow=True) self.assertRedirects(response, '/%s/%s/checkout/questions/' % (self.orga.slug, self.event.slug), target_status_code=200) def _set_session(self, key, value): session = self.client.session session[key] = value session.save() def test_free_price(self): self.ticket.free_price = True self.ticket.save() cr1 = CartPosition.objects.create( event=self.event, cart_id=self.session_key, item=self.ticket, price=42, expires=now() + timedelta(minutes=10) ) self._set_session('payment', 'banktransfer') response = self.client.post('/%s/%s/checkout/confirm/' % (self.orga.slug, self.event.slug), follow=True) doc = BeautifulSoup(response.rendered_content) self.assertEqual(len(doc.select(".thank-you")), 1) self.assertFalse(CartPosition.objects.filter(id=cr1.id).exists()) self.assertEqual(Order.objects.count(), 1) self.assertEqual(OrderPosition.objects.count(), 1) self.assertEqual(OrderPosition.objects.first().price, 42) def test_confirm_in_time(self): cr1 = CartPosition.objects.create( event=self.event, cart_id=self.session_key, item=self.ticket, price=23, expires=now() + timedelta(minutes=10) ) self._set_session('payment', 'banktransfer') response = self.client.post('/%s/%s/checkout/confirm/' % (self.orga.slug, self.event.slug), follow=True) doc = BeautifulSoup(response.rendered_content) self.assertEqual(len(doc.select(".thank-you")), 1) self.assertFalse(CartPosition.objects.filter(id=cr1.id).exists()) self.assertEqual(Order.objects.count(), 1) self.assertEqual(OrderPosition.objects.count(), 1) def test_confirm_expired_available(self): cr1 = CartPosition.objects.create( event=self.event, cart_id=self.session_key, item=self.ticket, price=23, expires=now() - timedelta(minutes=10) ) self._set_session('payment', 'banktransfer') response = self.client.post('/%s/%s/checkout/confirm/' % (self.orga.slug, self.event.slug), follow=True) doc = BeautifulSoup(response.rendered_content) self.assertEqual(len(doc.select(".thank-you")), 1) self.assertFalse(CartPosition.objects.filter(id=cr1.id).exists()) self.assertEqual(Order.objects.count(), 1) self.assertEqual(OrderPosition.objects.count(), 1) def test_confirm_price_changed(self): self.ticket.default_price = 24 self.ticket.save() cr1 = CartPosition.objects.create( event=self.event, cart_id=self.session_key, item=self.ticket, price=23, expires=now() - timedelta(minutes=10) ) self._set_session('payment', 'banktransfer') response = self.client.post('/%s/%s/checkout/confirm/' % (self.orga.slug, self.event.slug), follow=True) doc = BeautifulSoup(response.rendered_content) self.assertEqual(len(doc.select(".alert-danger")), 1) cr1 = CartPosition.objects.get(id=cr1.id) self.assertEqual(cr1.price, 24) def test_confirm_free_price_increased(self): self.ticket.default_price = 24 self.ticket.free_price = True self.ticket.save() cr1 = CartPosition.objects.create( event=self.event, cart_id=self.session_key, item=self.ticket, price=23, expires=now() - timedelta(minutes=10) ) self._set_session('payment', 'banktransfer') response = self.client.post('/%s/%s/checkout/confirm/' % (self.orga.slug, self.event.slug), follow=True) doc = BeautifulSoup(response.rendered_content) self.assertEqual(len(doc.select(".alert-danger")), 1) cr1 = CartPosition.objects.get(id=cr1.id) self.assertEqual(cr1.price, 24) def test_voucher(self): v = Voucher.objects.create(item=self.ticket, price=Decimal('12.00'), event=self.event, valid_until=now() + timedelta(days=2)) cr1 = CartPosition.objects.create( event=self.event, cart_id=self.session_key, item=self.ticket, price=12, expires=now() + timedelta(minutes=10), voucher=v ) self._set_session('payment', 'banktransfer') response = self.client.post('/%s/%s/checkout/confirm/' % (self.orga.slug, self.event.slug), follow=True) doc = BeautifulSoup(response.rendered_content) self.assertEqual(len(doc.select(".thank-you")), 1) self.assertFalse(CartPosition.objects.filter(id=cr1.id).exists()) self.assertEqual(Order.objects.count(), 1) self.assertEqual(OrderPosition.objects.count(), 1) self.assertEqual(OrderPosition.objects.first().voucher, v) self.assertTrue(Voucher.objects.get(pk=v.pk).redeemed) def test_voucher_required(self): v = Voucher.objects.create(item=self.ticket, price=Decimal('12.00'), event=self.event, valid_until=now() + timedelta(days=2)) self.ticket.require_voucher = True self.ticket.save() CartPosition.objects.create( event=self.event, cart_id=self.session_key, item=self.ticket, price=12, expires=now() + timedelta(minutes=10), voucher=v ) self._set_session('payment', 'banktransfer') response = self.client.post('/%s/%s/checkout/confirm/' % (self.orga.slug, self.event.slug), follow=True) doc = BeautifulSoup(response.rendered_content) self.assertEqual(len(doc.select(".thank-you")), 1) self.assertTrue(Voucher.objects.get(pk=v.pk).redeemed) def test_voucher_price_changed(self): v = Voucher.objects.create(item=self.ticket, price=Decimal('12.00'), event=self.event, valid_until=now() + timedelta(days=2)) cr1 = CartPosition.objects.create( event=self.event, cart_id=self.session_key, item=self.ticket, price=13, expires=now() - timedelta(minutes=10), voucher=v ) self._set_session('payment', 'banktransfer') response = self.client.post('/%s/%s/checkout/confirm/' % (self.orga.slug, self.event.slug), follow=True) doc = BeautifulSoup(response.rendered_content) self.assertEqual(len(doc.select(".alert-danger")), 1) cr1 = CartPosition.objects.get(id=cr1.id) self.assertEqual(cr1.price, Decimal('12.00')) def test_voucher_expired(self): v = Voucher.objects.create(item=self.ticket, price=Decimal('12.00'), event=self.event, valid_until=now() - timedelta(days=2)) CartPosition.objects.create( event=self.event, cart_id=self.session_key, item=self.ticket, price=12, expires=now() - timedelta(minutes=10), voucher=v ) self._set_session('payment', 'banktransfer') response = self.client.post('/%s/%s/checkout/confirm/' % (self.orga.slug, self.event.slug), follow=True) doc = BeautifulSoup(response.rendered_content) self.assertIn("expired", doc.select(".alert-danger")[0].text) def test_voucher_redeemed(self): v = Voucher.objects.create(item=self.ticket, price=Decimal('12.00'), event=self.event, valid_until=now() + timedelta(days=2), redeemed=True) CartPosition.objects.create( event=self.event, cart_id=self.session_key, item=self.ticket, price=12, expires=now() - timedelta(minutes=10), voucher=v ) self._set_session('payment', 'banktransfer') response = self.client.post('/%s/%s/checkout/confirm/' % (self.orga.slug, self.event.slug), follow=True) doc = BeautifulSoup(response.rendered_content) self.assertIn("has already been", doc.select(".alert-danger")[0].text) def test_voucher_ignore_quota(self): self.quota_tickets.size = 0 self.quota_tickets.save() v = Voucher.objects.create(item=self.ticket, price=Decimal('12.00'), event=self.event, valid_until=now() + timedelta(days=2), allow_ignore_quota=True) cr1 = CartPosition.objects.create( event=self.event, cart_id=self.session_key, item=self.ticket, price=12, expires=now() - timedelta(minutes=10), voucher=v ) self._set_session('payment', 'banktransfer') response = self.client.post('/%s/%s/checkout/confirm/' % (self.orga.slug, self.event.slug), follow=True) doc = BeautifulSoup(response.rendered_content) self.assertEqual(len(doc.select(".thank-you")), 1) self.assertFalse(CartPosition.objects.filter(id=cr1.id).exists()) self.assertEqual(Order.objects.count(), 1) self.assertEqual(OrderPosition.objects.count(), 1) def test_voucher_block_quota(self): self.quota_tickets.size = 1 self.quota_tickets.save() v = Voucher.objects.create(item=self.ticket, price=Decimal('12.00'), event=self.event, valid_until=now() + timedelta(days=2), block_quota=True) cr1 = CartPosition.objects.create( event=self.event, cart_id=self.session_key, item=self.ticket, price=12, expires=now() - timedelta(minutes=10) ) self._set_session('payment', 'banktransfer') response = self.client.post('/%s/%s/checkout/confirm/' % (self.orga.slug, self.event.slug), follow=True) doc = BeautifulSoup(response.rendered_content) self.assertEqual(len(doc.select(".alert-danger")), 1) self.assertEqual(CartPosition.objects.filter(cart_id=self.session_key).count(), 1) cr1.voucher = v cr1.save() response = self.client.post('/%s/%s/checkout/confirm/' % (self.orga.slug, self.event.slug), follow=True) doc = BeautifulSoup(response.rendered_content) self.assertEqual(len(doc.select(".thank-you")), 1) self.assertFalse(CartPosition.objects.filter(id=cr1.id).exists()) self.assertEqual(Order.objects.count(), 1) self.assertEqual(OrderPosition.objects.count(), 1) def test_voucher_double(self): self.quota_tickets.size = 2 self.quota_tickets.save() v = Voucher.objects.create(item=self.ticket, event=self.event, valid_until=now() + timedelta(days=2), block_quota=True) CartPosition.objects.create( event=self.event, cart_id=self.session_key, item=self.ticket, price=23, expires=now() + timedelta(minutes=10), voucher=v ) CartPosition.objects.create( event=self.event, cart_id=self.session_key, item=self.ticket, price=23, expires=now() + timedelta(minutes=10), voucher=v ) self._set_session('payment', 'banktransfer') response = self.client.post('/%s/%s/checkout/confirm/' % (self.orga.slug, self.event.slug), follow=True) doc = BeautifulSoup(response.rendered_content) self.assertEqual(CartPosition.objects.filter(cart_id=self.session_key, voucher=v).count(), 1) self.assertEqual(len(doc.select(".alert-danger")), 1) self.assertFalse(Order.objects.exists()) response = self.client.post('/%s/%s/checkout/confirm/' % (self.orga.slug, self.event.slug), follow=True) doc = BeautifulSoup(response.rendered_content) self.assertFalse(CartPosition.objects.filter(cart_id=self.session_key, voucher=v).exists()) self.assertEqual(len(doc.select(".thank-you")), 1) self.assertEqual(Order.objects.count(), 1) self.assertEqual(OrderPosition.objects.count(), 1) def test_confirm_expired_partial(self): self.quota_tickets.size = 1 self.quota_tickets.save() CartPosition.objects.create( event=self.event, cart_id=self.session_key, item=self.ticket, price=23, expires=now() - timedelta(minutes=10) ) CartPosition.objects.create( event=self.event, cart_id=self.session_key, item=self.ticket, price=23, expires=now() - timedelta(minutes=10) ) self._set_session('payment', 'banktransfer') response = self.client.post('/%s/%s/checkout/confirm/' % (self.orga.slug, self.event.slug), follow=True) doc = BeautifulSoup(response.rendered_content) self.assertEqual(len(doc.select(".alert-danger")), 1) self.assertEqual(CartPosition.objects.filter(cart_id=self.session_key).count(), 1) def test_confirm_presale_over(self): self.event.presale_end = now() - datetime.timedelta(days=1) self.event.save() CartPosition.objects.create( event=self.event, cart_id=self.session_key, item=self.ticket, price=23, expires=now() + timedelta(minutes=10) ) self._set_session('payment', 'banktransfer') response = self.client.post('/%s/%s/checkout/confirm/' % (self.orga.slug, self.event.slug), follow=True) doc = BeautifulSoup(response.rendered_content) self.assertGreaterEqual(len(doc.select(".alert-danger")), 1) def test_confirm_require_voucher(self): self.ticket.require_voucher = True self.ticket.save() cr1 = CartPosition.objects.create( event=self.event, cart_id=self.session_key, item=self.ticket, price=23, expires=now() + timedelta(minutes=10) ) self._set_session('payment', 'banktransfer') response = self.client.post('/%s/%s/checkout/confirm/' % (self.orga.slug, self.event.slug), follow=True) doc = BeautifulSoup(response.rendered_content) self.assertGreaterEqual(len(doc.select(".alert-danger")), 1) self.assertFalse(CartPosition.objects.filter(id=cr1.id).exists()) def test_confirm_require_hide_without_voucher(self): self.ticket.require_voucher = True self.ticket.save() cr1 = CartPosition.objects.create( event=self.event, cart_id=self.session_key, item=self.ticket, price=23, expires=now() + timedelta(minutes=10) ) self._set_session('payment', 'banktransfer') response = self.client.post('/%s/%s/checkout/confirm/' % (self.orga.slug, self.event.slug), follow=True) doc = BeautifulSoup(response.rendered_content) self.assertGreaterEqual(len(doc.select(".alert-danger")), 1) self.assertFalse(CartPosition.objects.filter(id=cr1.id).exists()) def test_confirm_inactive(self): self.ticket.active = False self.ticket.save() cr1 = CartPosition.objects.create( event=self.event, cart_id=self.session_key, item=self.ticket, price=23, expires=now() - timedelta(minutes=10) ) self._set_session('payment', 'banktransfer') response = self.client.post('/%s/%s/checkout/confirm/' % (self.orga.slug, self.event.slug), follow=True) doc = BeautifulSoup(response.rendered_content) self.assertGreaterEqual(len(doc.select(".alert-danger")), 1) self.assertFalse(CartPosition.objects.filter(id=cr1.id).exists()) def test_confirm_expired_unavailable(self): self.quota_tickets.size = 0 self.quota_tickets.save() cr1 = CartPosition.objects.create( event=self.event, cart_id=self.session_key, item=self.ticket, price=23, expires=now() - timedelta(minutes=10) ) self._set_session('payment', 'banktransfer') response = self.client.post('/%s/%s/checkout/confirm/' % (self.orga.slug, self.event.slug), follow=True) doc = BeautifulSoup(response.rendered_content) self.assertGreaterEqual(len(doc.select(".alert-danger")), 1) self.assertFalse(CartPosition.objects.filter(id=cr1.id).exists()) def test_confirm_completely_unavailable(self): self.quota_tickets.items.remove(self.ticket) cr1 = CartPosition.objects.create( event=self.event, cart_id=self.session_key, item=self.ticket, price=23, expires=now() - timedelta(minutes=10) ) self._set_session('payment', 'banktransfer') response = self.client.post('/%s/%s/checkout/confirm/' % (self.orga.slug, self.event.slug), follow=True) doc = BeautifulSoup(response.rendered_content) self.assertGreaterEqual(len(doc.select(".alert-danger")), 1) self.assertFalse(CartPosition.objects.filter(id=cr1.id).exists())