New tests and minor fixes

src/tests/control/test_orders.py

@@ -2,12 +2,16 @@ from datetime import timedelta
 from decimal import Decimal
 
 import pytest
+from django.core import mail
 from django.utils.timezone import now
 from tests.base import SoupTest
 
 from pretix.base.models import (
-    CachedTicket, Event, EventPermission, Item, Order, OrderPosition,
-    Organizer, Quota, User,
+    CachedTicket, Event, EventPermission, InvoiceAddress, Item, Order,
+    OrderPosition, Organizer, Quota, User,
+)
+from pretix.base.services.invoices import (
+    generate_cancellation, generate_invoice,
 )
 
 
@@ -30,7 +34,7 @@ def env():
         code='FOO', event=event, email='dummy@dummy.test',
         status=Order.STATUS_PENDING,
         datetime=now(), expires=now() + timedelta(days=10),
-        total=0, payment_provider='banktransfer'
+        total=14, payment_provider='banktransfer', locale='en'
     )
     ticket = Item.objects.create(event=event, name='Early-bird ticket',
                                  category=None, default_price=23,
@@ -61,10 +65,23 @@ def test_order_list(client, env):
     assert 'FOO' not in response.rendered_content
     response = client.get('/control/event/dummy/dummy/orders/?status=n')
     assert 'FOO' in response.rendered_content
+    response = client.get('/control/event/dummy/dummy/orders/?status=ne')
+    assert 'FOO' in response.rendered_content
     response = client.get('/control/event/dummy/dummy/orders/?item=15')
     assert 'FOO' not in response.rendered_content
     response = client.get('/control/event/dummy/dummy/orders/?item=%s' % env[3].id)
     assert 'FOO' in response.rendered_content
+    response = client.get('/control/event/dummy/dummy/orders/?provider=foo')
+    assert 'FOO' not in response.rendered_content
+    response = client.get('/control/event/dummy/dummy/orders/?provider=banktransfer')
+    assert 'FOO' in response.rendered_content
+
+    response = client.get('/control/event/dummy/dummy/orders/?status=o')
+    assert 'FOO' not in response.rendered_content
+    env[2].expires = now() - timedelta(days=10)
+    env[2].save()
+    response = client.get('/control/event/dummy/dummy/orders/?status=o')
+    assert 'FOO' in response.rendered_content
 
 
 @pytest.mark.django_db
@@ -75,6 +92,42 @@ def test_order_detail(client, env):
     assert 'Peter' in response.rendered_content
 
 
+@pytest.mark.django_db
+def test_order_set_contact(client, env):
+    q = Quota.objects.create(event=env[0], size=0)
+    q.items.add(env[3])
+    client.login(email='dummy@dummy.dummy', password='dummy')
+    client.post('/control/event/dummy/dummy/orders/FOO/contact', {
+        'email': 'admin@rami.io'
+    })
+    o = Order.objects.get(id=env[2].id)
+    assert o.email == 'admin@rami.io'
+
+
+@pytest.mark.django_db
+def test_order_set_comment(client, env):
+    q = Quota.objects.create(event=env[0], size=0)
+    q.items.add(env[3])
+    client.login(email='dummy@dummy.dummy', password='dummy')
+    client.post('/control/event/dummy/dummy/orders/FOO/comment', {
+        'comment': 'Foo'
+    })
+    o = Order.objects.get(id=env[2].id)
+    assert o.comment == 'Foo'
+
+
+@pytest.mark.django_db
+def test_order_transition_to_expired_success(client, env):
+    q = Quota.objects.create(event=env[0], size=0)
+    q.items.add(env[3])
+    client.login(email='dummy@dummy.dummy', password='dummy')
+    client.post('/control/event/dummy/dummy/orders/FOO/transition', {
+        'status': 'e'
+    })
+    o = Order.objects.get(id=env[2].id)
+    assert o.status == Order.STATUS_EXPIRED
+
+
 @pytest.mark.django_db
 def test_order_transition_to_paid_in_time_success(client, env):
     q = Quota.objects.create(event=env[0], size=0)
@@ -103,45 +156,35 @@ def test_order_transition_to_paid_expired_quota_left(client, env):
     assert o.status == Order.STATUS_PAID
 
 
-@pytest.mark.django_db
-def test_order_transition_to_paid_expired_quota_full(client, env):
-    o = Order.objects.get(id=env[2].id)
-    o.status = Order.STATUS_EXPIRED
-    o.save()
-    q = Quota.objects.create(event=env[0], size=0)
-    q.items.add(env[3])
-    client.login(email='dummy@dummy.dummy', password='dummy')
-    client.post('/control/event/dummy/dummy/orders/FOO/transition', {
-        'status': 'p'
-    })
-    o = Order.objects.get(id=env[2].id)
-    assert o.status == Order.STATUS_EXPIRED
-
-
 @pytest.mark.django_db
 @pytest.mark.parametrize("process", [
     # (Old status, new status, success expected)
     (Order.STATUS_CANCELLED, Order.STATUS_PAID, False),
     (Order.STATUS_CANCELLED, Order.STATUS_PENDING, False),
     (Order.STATUS_CANCELLED, Order.STATUS_REFUNDED, False),
+    (Order.STATUS_CANCELLED, Order.STATUS_EXPIRED, False),
 
     (Order.STATUS_PAID, Order.STATUS_PENDING, True),
     (Order.STATUS_PAID, Order.STATUS_CANCELLED, False),
     (Order.STATUS_PAID, Order.STATUS_REFUNDED, True),
+    (Order.STATUS_PAID, Order.STATUS_EXPIRED, False),
 
     (Order.STATUS_PENDING, Order.STATUS_CANCELLED, True),
     (Order.STATUS_PENDING, Order.STATUS_PAID, True),
     (Order.STATUS_PENDING, Order.STATUS_REFUNDED, False),
+    (Order.STATUS_PENDING, Order.STATUS_EXPIRED, True),
 
     (Order.STATUS_REFUNDED, Order.STATUS_CANCELLED, False),
     (Order.STATUS_REFUNDED, Order.STATUS_PAID, False),
-    (Order.STATUS_REFUNDED, Order.STATUS_PENDING, False)
+    (Order.STATUS_REFUNDED, Order.STATUS_PENDING, False),
+    (Order.STATUS_REFUNDED, Order.STATUS_EXPIRED, False),
 ])
 def test_order_transition(client, env, process):
     o = Order.objects.get(id=env[2].id)
     o.status = process[0]
     o.save()
     client.login(email='dummy@dummy.dummy', password='dummy')
+    client.get('/control/event/dummy/dummy/orders/FOO/transition?status=' + process[1])
     client.post('/control/event/dummy/dummy/orders/FOO/transition', {
         'status': process[1]
     })
@@ -278,6 +321,101 @@ def test_order_download_success(client, env, mocker):
     assert dl == response['Location']
 
 
+@pytest.mark.django_db
+def test_order_invoice_create_forbidden(client, env):
+    client.login(email='dummy@dummy.dummy', password='dummy')
+    env[0].settings.set('invoice_generate', 'no')
+    response = client.post('/control/event/dummy/dummy/orders/FOO/invoice', {}, follow=True)
+    assert 'alert-danger' in response.rendered_content
+
+
+@pytest.mark.django_db
+def test_order_invoice_create_duplicate(client, env):
+    client.login(email='dummy@dummy.dummy', password='dummy')
+    generate_invoice(env[2])
+    env[0].settings.set('invoice_generate', 'admin')
+    response = client.post('/control/event/dummy/dummy/orders/FOO/invoice', {}, follow=True)
+    assert 'alert-danger' in response.rendered_content
+
+
+@pytest.mark.django_db
+def test_order_invoice_create_ok(client, env):
+    client.login(email='dummy@dummy.dummy', password='dummy')
+    env[0].settings.set('invoice_generate', 'admin')
+    response = client.post('/control/event/dummy/dummy/orders/FOO/invoice', {}, follow=True)
+    assert 'alert-success' in response.rendered_content
+    assert env[2].invoices.exists()
+
+
+@pytest.mark.django_db
+def test_order_invoice_regenerate(client, env):
+    client.login(email='dummy@dummy.dummy', password='dummy')
+    i = generate_invoice(env[2])
+    InvoiceAddress.objects.create(name='Foo', order=env[2])
+    env[0].settings.set('invoice_generate', 'admin')
+    response = client.post('/control/event/dummy/dummy/orders/FOO/invoices/%d/regenerate' % i.pk, {}, follow=True)
+    assert 'alert-success' in response.rendered_content
+    i.refresh_from_db()
+    assert 'Foo' in i.invoice_to
+    assert env[2].invoices.exists()
+
+
+@pytest.mark.django_db
+def test_order_invoice_regenerate_canceled(client, env):
+    client.login(email='dummy@dummy.dummy', password='dummy')
+    i = generate_invoice(env[2])
+    generate_cancellation(i)
+    response = client.post('/control/event/dummy/dummy/orders/FOO/invoices/%d/regenerate' % i.pk, {}, follow=True)
+    assert 'alert-danger' in response.rendered_content
+
+
+@pytest.mark.django_db
+def test_order_invoice_regenerate_unknown(client, env):
+    client.login(email='dummy@dummy.dummy', password='dummy')
+    response = client.post('/control/event/dummy/dummy/orders/FOO/invoices/%d/regenerate' % 3, {}, follow=True)
+    assert 'alert-danger' in response.rendered_content
+
+
+@pytest.mark.django_db
+def test_order_invoice_reissue(client, env):
+    client.login(email='dummy@dummy.dummy', password='dummy')
+    i = generate_invoice(env[2])
+    InvoiceAddress.objects.create(name='Foo', order=env[2])
+    env[0].settings.set('invoice_generate', 'admin')
+    response = client.post('/control/event/dummy/dummy/orders/FOO/invoices/%d/reissue' % i.pk, {}, follow=True)
+    assert 'alert-success' in response.rendered_content
+    i.refresh_from_db()
+    assert env[2].invoices.count() == 3
+    assert 'Foo' not in env[2].invoices.all()[0].invoice_to
+    assert 'Foo' not in env[2].invoices.all()[1].invoice_to
+    assert 'Foo' in env[2].invoices.all()[2].invoice_to
+
+
+@pytest.mark.django_db
+def test_order_invoice_reissue_canceled(client, env):
+    client.login(email='dummy@dummy.dummy', password='dummy')
+    i = generate_invoice(env[2])
+    generate_cancellation(i)
+    response = client.post('/control/event/dummy/dummy/orders/FOO/invoices/%d/reissue' % i.pk, {}, follow=True)
+    assert 'alert-danger' in response.rendered_content
+
+
+@pytest.mark.django_db
+def test_order_invoice_reissue_unknown(client, env):
+    client.login(email='dummy@dummy.dummy', password='dummy')
+    response = client.post('/control/event/dummy/dummy/orders/FOO/invoices/%d/reissue' % 3, {}, follow=True)
+    assert 'alert-danger' in response.rendered_content
+
+
+@pytest.mark.django_db
+def test_order_resend_link(client, env):
+    mail.outbox = []
+    client.login(email='dummy@dummy.dummy', password='dummy')
+    response = client.post('/control/event/dummy/dummy/orders/FOO/resend', {}, follow=True)
+    assert 'alert-success' in response.rendered_content
+    assert 'FOO' in mail.outbox[0].body
+
+
 @pytest.mark.django_db
 def test_order_extend_not_pending(client, env):
     o = Order.objects.get(id=env[2].id)