forked from CGM_Public/pretix_original
363 lines
19 KiB
Python
363 lines
19 KiB
Python
import datetime
|
|
from datetime import timedelta
|
|
|
|
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, ItemVariation, Organizer,
|
|
Property, PropertyValue, Question, QuestionAnswer, Quota, User,
|
|
)
|
|
|
|
|
|
class CartTestMixin:
|
|
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)
|
|
)
|
|
self.category = ItemCategory.objects.create(event=self.event, name="Everything", position=0)
|
|
self.quota_shirts = Quota.objects.create(event=self.event, name='Shirts', size=2)
|
|
self.shirt = Item.objects.create(event=self.event, name='T-Shirt', category=self.category, default_price=12)
|
|
prop1 = Property.objects.create(event=self.event, name="Color", item=self.shirt)
|
|
val1 = PropertyValue.objects.create(prop=prop1, value="Red", position=0)
|
|
val2 = PropertyValue.objects.create(prop=prop1, value="Black", position=1)
|
|
self.quota_shirts.items.add(self.shirt)
|
|
self.shirt_red = ItemVariation.objects.create(item=self.shirt, default_price=14)
|
|
self.shirt_red.values.add(val1)
|
|
var2 = ItemVariation.objects.create(item=self.shirt)
|
|
var2.values.add(val2)
|
|
self.quota_shirts.variations.add(self.shirt_red)
|
|
self.quota_shirts.variations.add(var2)
|
|
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)
|
|
self.quota_tickets.items.add(self.ticket)
|
|
|
|
self.client.get('/%s/%s/' % (self.orga.slug, self.event.slug))
|
|
self.session_key = self.client.cookies.get(settings.SESSION_COOKIE_NAME).value
|
|
|
|
|
|
class CartTest(CartTestMixin, TestCase):
|
|
def test_simple(self):
|
|
response = self.client.post('/%s/%s/cart/add' % (self.orga.slug, self.event.slug), {
|
|
'item_' + self.ticket.identity: '1'
|
|
}, follow=True)
|
|
self.assertRedirects(response, '/%s/%s/' % (self.orga.slug, self.event.slug),
|
|
target_status_code=200)
|
|
doc = BeautifulSoup(response.rendered_content)
|
|
self.assertIn('Early-bird', doc.select('.cart .cart-row')[0].select('strong')[0].text)
|
|
self.assertIn('1', doc.select('.cart .cart-row')[0].select('.count')[0].text)
|
|
self.assertIn('23', doc.select('.cart .cart-row')[0].select('.price')[0].text)
|
|
self.assertIn('23', doc.select('.cart .cart-row')[0].select('.price')[1].text)
|
|
objs = list(CartPosition.objects.filter(cart_id=self.session_key, event=self.event))
|
|
self.assertEqual(len(objs), 1)
|
|
self.assertEqual(objs[0].item, self.ticket)
|
|
self.assertIsNone(objs[0].variation)
|
|
self.assertEqual(objs[0].price, 23)
|
|
|
|
def test_variation(self):
|
|
response = self.client.post('/%s/%s/cart/add' % (self.orga.slug, self.event.slug), {
|
|
'variation_' + self.shirt.identity + '_' + self.shirt_red.identity: '1'
|
|
}, follow=True)
|
|
self.assertRedirects(response, '/%s/%s/' % (self.orga.slug, self.event.slug),
|
|
target_status_code=200)
|
|
doc = BeautifulSoup(response.rendered_content)
|
|
self.assertIn('Shirt', doc.select('.cart .cart-row')[0].select('strong')[0].text)
|
|
self.assertIn('Red', doc.select('.cart .cart-row')[0].text)
|
|
self.assertIn('1', doc.select('.cart .cart-row')[0].select('.count')[0].text)
|
|
self.assertIn('14', doc.select('.cart .cart-row')[0].select('.price')[0].text)
|
|
self.assertIn('14', doc.select('.cart .cart-row')[0].select('.price')[1].text)
|
|
objs = list(CartPosition.objects.filter(cart_id=self.session_key, event=self.event))
|
|
self.assertEqual(len(objs), 1)
|
|
self.assertEqual(objs[0].item, self.shirt)
|
|
self.assertEqual(objs[0].variation, self.shirt_red)
|
|
self.assertEqual(objs[0].price, 14)
|
|
|
|
def test_count(self):
|
|
response = self.client.post('/%s/%s/cart/add' % (self.orga.slug, self.event.slug), {
|
|
'item_' + self.ticket.identity: '2'
|
|
}, follow=True)
|
|
self.assertRedirects(response, '/%s/%s/' % (self.orga.slug, self.event.slug),
|
|
target_status_code=200)
|
|
doc = BeautifulSoup(response.rendered_content)
|
|
self.assertIn('Early-bird', doc.select('.cart .cart-row')[0].select('strong')[0].text)
|
|
self.assertIn('2', doc.select('.cart .cart-row')[0].select('.count')[0].text)
|
|
self.assertIn('23', doc.select('.cart .cart-row')[0].select('.price')[0].text)
|
|
self.assertIn('46', doc.select('.cart .cart-row')[0].select('.price')[1].text)
|
|
objs = list(CartPosition.objects.filter(cart_id=self.session_key, event=self.event))
|
|
self.assertEqual(len(objs), 2)
|
|
for obj in objs:
|
|
self.assertEqual(obj.item, self.ticket)
|
|
self.assertIsNone(obj.variation)
|
|
self.assertEqual(obj.price, 23)
|
|
|
|
def test_multiple(self):
|
|
response = self.client.post('/%s/%s/cart/add' % (self.orga.slug, self.event.slug), {
|
|
'item_' + self.ticket.identity: '2',
|
|
'variation_' + self.shirt.identity + '_' + self.shirt_red.identity: '1'
|
|
}, follow=True)
|
|
self.assertRedirects(response, '/%s/%s/' % (self.orga.slug, self.event.slug),
|
|
target_status_code=200)
|
|
doc = BeautifulSoup(response.rendered_content)
|
|
self.assertIn('Early-bird', doc.select('.cart')[0].text)
|
|
self.assertIn('Shirt', doc.select('.cart')[0].text)
|
|
objs = list(CartPosition.objects.filter(cart_id=self.session_key, event=self.event))
|
|
self.assertEqual(len(objs), 3)
|
|
self.assertIn(self.shirt, [obj.item for obj in objs])
|
|
self.assertIn(self.shirt_red, [obj.variation for obj in objs])
|
|
self.assertIn(self.ticket, [obj.item for obj in objs])
|
|
|
|
def test_fuzzy_input(self):
|
|
response = self.client.post('/%s/%s/cart/add' % (self.orga.slug, self.event.slug), {
|
|
'item_' + self.ticket.identity: 'a',
|
|
}, follow=True)
|
|
self.assertRedirects(response, '/%s/%s/' % (self.orga.slug, self.event.slug),
|
|
target_status_code=200)
|
|
doc = BeautifulSoup(response.rendered_content)
|
|
self.assertIn('numbers only', doc.select('.alert-danger')[0].text)
|
|
self.assertFalse(CartPosition.objects.filter(cart_id=self.session_key, event=self.event).exists())
|
|
|
|
response = self.client.post('/%s/%s/cart/add' % (self.orga.slug, self.event.slug), {
|
|
}, follow=True)
|
|
self.assertRedirects(response, '/%s/%s/' % (self.orga.slug, self.event.slug),
|
|
target_status_code=200)
|
|
doc = BeautifulSoup(response.rendered_content)
|
|
self.assertIn('did not select any products', doc.select('.alert-warning')[0].text)
|
|
self.assertFalse(CartPosition.objects.filter(cart_id=self.session_key, event=self.event).exists())
|
|
|
|
def test_wrong_event(self):
|
|
event2 = Event.objects.create(
|
|
organizer=self.orga, name='MRMCD', slug='mrmcd',
|
|
date_from=datetime.datetime(2014, 9, 6, tzinfo=datetime.timezone.utc)
|
|
)
|
|
shirt2 = Item.objects.create(event=event2, name='T-Shirt', default_price=12)
|
|
response = self.client.post('/%s/%s/cart/add' % (self.orga.slug, self.event.slug), {
|
|
'item_' + shirt2.identity: '1',
|
|
}, follow=True)
|
|
self.assertRedirects(response, '/%s/%s/' % (self.orga.slug, self.event.slug),
|
|
target_status_code=200)
|
|
doc = BeautifulSoup(response.rendered_content)
|
|
self.assertIn('not available', doc.select('.alert-danger')[0].text)
|
|
self.assertFalse(CartPosition.objects.filter(cart_id=self.session_key, event=self.event).exists())
|
|
|
|
def test_no_quota(self):
|
|
shirt2 = Item.objects.create(event=self.event, name='T-Shirt', default_price=12)
|
|
response = self.client.post('/%s/%s/cart/add' % (self.orga.slug, self.event.slug), {
|
|
'item_' + shirt2.identity: '1',
|
|
}, follow=True)
|
|
self.assertRedirects(response, '/%s/%s/' % (self.orga.slug, self.event.slug),
|
|
target_status_code=200)
|
|
doc = BeautifulSoup(response.rendered_content)
|
|
self.assertIn('no longer available', doc.select('.alert-danger')[0].text)
|
|
self.assertFalse(CartPosition.objects.filter(cart_id=self.session_key, event=self.event).exists())
|
|
|
|
def test_in_time_available(self):
|
|
self.ticket.available_until = now() + timedelta(days=2)
|
|
self.ticket.available_from = now() - timedelta(days=2)
|
|
self.ticket.save()
|
|
self.client.post('/%s/%s/cart/add' % (self.orga.slug, self.event.slug), {
|
|
'item_' + self.ticket.identity: '1',
|
|
}, follow=True)
|
|
self.assertEqual(CartPosition.objects.filter(cart_id=self.session_key, event=self.event).count(), 1)
|
|
|
|
def test_no_longer_available(self):
|
|
self.ticket.available_until = now() - timedelta(days=2)
|
|
self.ticket.save()
|
|
self.client.post('/%s/%s/cart/add' % (self.orga.slug, self.event.slug), {
|
|
'item_' + self.ticket.identity: '1',
|
|
}, follow=True)
|
|
self.assertEqual(CartPosition.objects.filter(cart_id=self.session_key, event=self.event).count(), 0)
|
|
|
|
def test_not_yet_available(self):
|
|
self.ticket.available_from = now() + timedelta(days=2)
|
|
self.ticket.save()
|
|
self.client.post('/%s/%s/cart/add' % (self.orga.slug, self.event.slug), {
|
|
'item_' + self.ticket.identity: '1',
|
|
}, follow=True)
|
|
self.assertEqual(CartPosition.objects.filter(cart_id=self.session_key, event=self.event).count(), 0)
|
|
|
|
def test_max_items(self):
|
|
CartPosition.objects.create(
|
|
event=self.event, cart_id=self.session_key, item=self.ticket,
|
|
price=23, expires=now() + timedelta(minutes=10)
|
|
)
|
|
self.event.settings.max_items_per_order = 5
|
|
response = self.client.post('/%s/%s/cart/add' % (self.orga.slug, self.event.slug), {
|
|
'item_' + self.ticket.identity: '5',
|
|
}, follow=True)
|
|
self.assertRedirects(response, '/%s/%s/' % (self.orga.slug, self.event.slug),
|
|
target_status_code=200)
|
|
doc = BeautifulSoup(response.rendered_content)
|
|
self.assertIn('more than', doc.select('.alert-danger')[0].text)
|
|
self.assertEqual(CartPosition.objects.filter(cart_id=self.session_key, event=self.event).count(), 1)
|
|
|
|
def test_quota_full(self):
|
|
self.quota_tickets.size = 0
|
|
self.quota_tickets.save()
|
|
response = self.client.post('/%s/%s/cart/add' % (self.orga.slug, self.event.slug), {
|
|
'item_' + self.ticket.identity: '1',
|
|
}, follow=True)
|
|
self.assertRedirects(response, '/%s/%s/' % (self.orga.slug, self.event.slug),
|
|
target_status_code=200)
|
|
doc = BeautifulSoup(response.rendered_content)
|
|
self.assertIn('no longer available', doc.select('.alert-danger')[0].text)
|
|
self.assertFalse(CartPosition.objects.filter(cart_id=self.session_key, event=self.event).exists())
|
|
|
|
def test_quota_partly(self):
|
|
self.quota_tickets.size = 1
|
|
self.quota_tickets.save()
|
|
response = self.client.post('/%s/%s/cart/add' % (self.orga.slug, self.event.slug), {
|
|
'item_' + self.ticket.identity: '2'
|
|
}, follow=True)
|
|
self.assertRedirects(response, '/%s/%s/' % (self.orga.slug, self.event.slug),
|
|
target_status_code=200)
|
|
doc = BeautifulSoup(response.rendered_content)
|
|
self.assertIn('no longer available', doc.select('.alert-danger')[0].text)
|
|
self.assertIn('Early-bird', doc.select('.cart .cart-row')[0].select('strong')[0].text)
|
|
self.assertIn('1', doc.select('.cart .cart-row')[0].select('.count')[0].text)
|
|
self.assertIn('23', doc.select('.cart .cart-row')[0].select('.price')[0].text)
|
|
self.assertIn('23', doc.select('.cart .cart-row')[0].select('.price')[1].text)
|
|
objs = list(CartPosition.objects.filter(cart_id=self.session_key, event=self.event))
|
|
self.assertEqual(len(objs), 1)
|
|
self.assertEqual(objs[0].item, self.ticket)
|
|
self.assertIsNone(objs[0].variation)
|
|
self.assertEqual(objs[0].price, 23)
|
|
|
|
def test_renew_in_time(self):
|
|
cp = CartPosition.objects.create(
|
|
event=self.event, cart_id=self.session_key, item=self.ticket,
|
|
price=23, expires=now() + timedelta(minutes=10)
|
|
)
|
|
self.client.post('/%s/%s/cart/add' % (self.orga.slug, self.event.slug), {
|
|
}, follow=True)
|
|
cp = CartPosition.objects.current.get(identity=cp.identity)
|
|
self.assertGreater(cp.expires, now())
|
|
|
|
def test_renew_expired_successfully(self):
|
|
cp1 = CartPosition.objects.create(
|
|
event=self.event, cart_id=self.session_key, item=self.ticket,
|
|
price=23, expires=now() - timedelta(minutes=10)
|
|
)
|
|
self.client.post('/%s/%s/cart/add' % (self.orga.slug, self.event.slug), {
|
|
'variation_%s_%s' % (self.shirt.identity, self.shirt_red.identity): '1'
|
|
}, follow=True)
|
|
obj = CartPosition.objects.current.get(identity=cp1.identity)
|
|
self.assertEqual(obj.item, self.ticket)
|
|
self.assertIsNone(obj.variation)
|
|
self.assertEqual(obj.price, 23)
|
|
self.assertGreater(obj.expires, now())
|
|
|
|
def test_renew_questions(self):
|
|
cr1 = CartPosition.objects.create(
|
|
event=self.event, cart_id=self.session_key, item=self.ticket,
|
|
price=23, expires=now() - timedelta(minutes=10)
|
|
)
|
|
q1 = Question.objects.create(
|
|
event=self.event, question='Age', type=Question.TYPE_NUMBER,
|
|
required=True
|
|
)
|
|
self.ticket.questions.add(q1)
|
|
cr1.answers.add(QuestionAnswer.objects.create(
|
|
cartposition=cr1, question=q1, answer='23'
|
|
))
|
|
self.client.post('/%s/%s/cart/add' % (self.orga.slug, self.event.slug), {
|
|
'item_' + self.ticket.identity: '1',
|
|
}, follow=True)
|
|
obj = CartPosition.objects.current.get(identity=cr1.identity)
|
|
self.assertEqual(obj.answers.get(question=q1).answer, '23')
|
|
|
|
def test_renew_expired_failed(self):
|
|
self.quota_tickets.size = 0
|
|
self.quota_tickets.save()
|
|
cp1 = CartPosition.objects.create(
|
|
event=self.event, cart_id=self.session_key, item=self.ticket,
|
|
price=23, expires=now() - timedelta(minutes=10)
|
|
)
|
|
response = self.client.post('/%s/%s/cart/add' % (self.orga.slug, self.event.slug), {
|
|
'item_' + self.ticket.identity: '1',
|
|
}, follow=True)
|
|
doc = BeautifulSoup(response.rendered_content)
|
|
self.assertIn('no longer available', doc.select('.alert-danger')[0].text)
|
|
self.assertFalse(CartPosition.objects.current.filter(identity=cp1.identity).exists())
|
|
|
|
def test_remove_simple(self):
|
|
CartPosition.objects.create(
|
|
event=self.event, cart_id=self.session_key, item=self.ticket,
|
|
price=23, expires=now() + timedelta(minutes=10)
|
|
)
|
|
response = self.client.post('/%s/%s/cart/remove' % (self.orga.slug, self.event.slug), {
|
|
'item_' + self.ticket.identity: '1',
|
|
}, follow=True)
|
|
doc = BeautifulSoup(response.rendered_content)
|
|
self.assertIn('updated', doc.select('.alert-success')[0].text)
|
|
self.assertFalse(CartPosition.objects.current.filter(cart_id=self.session_key, event=self.event).exists())
|
|
|
|
def test_remove_variation(self):
|
|
CartPosition.objects.create(
|
|
event=self.event, cart_id=self.session_key, item=self.shirt, variation=self.shirt_red,
|
|
price=14, expires=now() + timedelta(minutes=10)
|
|
)
|
|
response = self.client.post('/%s/%s/cart/remove' % (self.orga.slug, self.event.slug), {
|
|
'variation_' + self.shirt.identity + '_' + self.shirt_red.identity: '1',
|
|
}, follow=True)
|
|
doc = BeautifulSoup(response.rendered_content)
|
|
self.assertIn('updated', doc.select('.alert-success')[0].text)
|
|
self.assertFalse(CartPosition.objects.current.filter(cart_id=self.session_key, event=self.event).exists())
|
|
|
|
def test_remove_one_of_multiple(self):
|
|
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)
|
|
)
|
|
response = self.client.post('/%s/%s/cart/remove' % (self.orga.slug, self.event.slug), {
|
|
'item_' + self.ticket.identity: '1',
|
|
}, follow=True)
|
|
doc = BeautifulSoup(response.rendered_content)
|
|
self.assertIn('updated', doc.select('.alert-success')[0].text)
|
|
self.assertEqual(CartPosition.objects.current.filter(cart_id=self.session_key, event=self.event).count(), 1)
|
|
|
|
def test_remove_multiple(self):
|
|
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)
|
|
)
|
|
response = self.client.post('/%s/%s/cart/remove' % (self.orga.slug, self.event.slug), {
|
|
'item_' + self.ticket.identity: '2',
|
|
}, follow=True)
|
|
doc = BeautifulSoup(response.rendered_content)
|
|
self.assertIn('updated', doc.select('.alert-success')[0].text)
|
|
self.assertFalse(CartPosition.objects.current.filter(cart_id=self.session_key, event=self.event).exists())
|
|
|
|
def test_remove_most_expensive(self):
|
|
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=20, expires=now() + timedelta(minutes=10)
|
|
)
|
|
response = self.client.post('/%s/%s/cart/remove' % (self.orga.slug, self.event.slug), {
|
|
'item_' + self.ticket.identity: '1',
|
|
}, follow=True)
|
|
doc = BeautifulSoup(response.rendered_content)
|
|
self.assertIn('updated', doc.select('.alert-success')[0].text)
|
|
objs = list(CartPosition.objects.current.filter(cart_id=self.session_key, event=self.event))
|
|
self.assertEqual(len(objs), 1)
|
|
self.assertEqual(objs[0].item, self.ticket)
|
|
self.assertIsNone(objs[0].variation)
|
|
self.assertEqual(objs[0].price, 20)
|