#
# This file is part of pretix (Community Edition).
#
# Copyright (C) 2014-2020 Raphael Michel and contributors
# Copyright (C) 2020-2021 rami.io GmbH and contributors
#
# This program is free software: you can redistribute it and/or modify it under the terms of the GNU Affero General
# Public License as published by the Free Software Foundation in version 3 of the License.
#
# ADDITIONAL TERMS APPLY: Pursuant to Section 7 of the GNU Affero General Public License, additional terms are
# applicable granting you additional permissions and placing additional restrictions on your usage of this software.
# Please refer to the pretix LICENSE file to obtain the full terms applicable to this work. If you did not receive
# this file, see <https://pretix.eu/about/en/license>.
#
# This program is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even the implied
# warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU Affero General Public License for more
# details.
#
# You should have received a copy of the GNU Affero General Public License along with this program.  If not, see
# <https://www.gnu.org/licenses/>.
#

# This file is based on an earlier version of pretix which was released under the Apache License 2.0. The full text of
# the Apache License 2.0 can be obtained at <http://www.apache.org/licenses/LICENSE-2.0>.
#
# This file may have since been changed and any changes are released under the terms of AGPLv3 as described above. A
# full history of changes and contributors is available at <https://github.com/pretix/pretix>.
#
# This file contains Apache-licensed contributions copyrighted by: Clint, Enrique Saez, Sohalt
#
# Unless required by applicable law or agreed to in writing, software distributed under the Apache License 2.0 is
# distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
# License for the specific language governing permissions and limitations under the License.

import datetime
from decimal import Decimal

from django.utils.timezone import now
from django_scopes import scopes_disabled
from tests.base import SoupTest, extract_form_fields

from pretix.base.models import (
    Discount, Event, Item, ItemCategory, ItemVariation, Order, OrderPosition,
    Organizer, Question, Quota, Team, User,
)


class ItemFormTest(SoupTest):
    @scopes_disabled()
    def setUp(self):
        super().setUp()
        self.user = User.objects.create_user('dummy@dummy.dummy', 'dummy')
        self.orga1 = Organizer.objects.create(name='CCC', slug='ccc')
        self.orga2 = Organizer.objects.create(name='MRM', slug='mrm')
        self.event1 = Event.objects.create(
            organizer=self.orga1, name='30C3', slug='30c3',
            date_from=datetime.datetime(2013, 12, 26, tzinfo=datetime.timezone.utc),
        )
        self.item1 = Item.objects.create(event=self.event1, name="Standard", default_price=0, position=1)
        t = Team.objects.create(organizer=self.orga1, can_change_event_settings=True, can_change_items=True)
        t.members.add(self.user)
        t.limit_events.add(self.event1)
        self.client.login(email='dummy@dummy.dummy', password='dummy')


class CategoriesTest(ItemFormTest):

    def test_create(self):
        doc = self.get_doc('/control/event/%s/%s/categories/add' % (self.orga1.slug, self.event1.slug))
        form_data = extract_form_fields(doc.select('.container-fluid form')[0])
        form_data['name_0'] = 'Entry tickets'
        doc = self.post_doc('/control/event/%s/%s/categories/add' % (self.orga1.slug, self.event1.slug), form_data)
        assert doc.select(".alert-success")
        self.assertIn("Entry tickets", doc.select("#page-wrapper table")[0].text)

    def test_update(self):
        c = ItemCategory.objects.create(event=self.event1, name="Entry tickets")
        doc = self.get_doc('/control/event/%s/%s/categories/%s/' % (self.orga1.slug, self.event1.slug, c.id))
        form_data = extract_form_fields(doc.select('.container-fluid form')[0])
        form_data['name_0'] = 'T-Shirts'
        doc = self.post_doc('/control/event/%s/%s/categories/%s/' % (self.orga1.slug, self.event1.slug, c.id),
                            form_data)
        assert doc.select(".alert-success")
        self.assertIn("T-Shirts", doc.select("#page-wrapper table")[0].text)
        self.assertNotIn("Entry tickets", doc.select("#page-wrapper table")[0].text)
        with scopes_disabled():
            assert str(ItemCategory.objects.get(id=c.id).name) == 'T-Shirts'

    def test_sort(self):
        with scopes_disabled():
            c1 = ItemCategory.objects.create(event=self.event1, name="Entry tickets", position=0)
            ItemCategory.objects.create(event=self.event1, name="T-Shirts", position=1)
        doc = self.get_doc('/control/event/%s/%s/categories/' % (self.orga1.slug, self.event1.slug))
        self.assertIn("Entry tickets", doc.select("table > tbody > tr")[0].text)
        self.assertIn("T-Shirts", doc.select("table > tbody > tr")[1].text)

        self.client.post('/control/event/%s/%s/categories/%s/down' % (self.orga1.slug, self.event1.slug, c1.id))
        doc = self.get_doc('/control/event/%s/%s/categories/' % (self.orga1.slug, self.event1.slug))
        self.assertIn("Entry tickets", doc.select("table > tbody > tr")[1].text)
        self.assertIn("T-Shirts", doc.select("table > tbody > tr")[0].text)

        self.client.post('/control/event/%s/%s/categories/%s/up' % (self.orga1.slug, self.event1.slug, c1.id))
        doc = self.get_doc('/control/event/%s/%s/categories/' % (self.orga1.slug, self.event1.slug))
        self.assertIn("Entry tickets", doc.select("table > tbody > tr")[0].text)
        self.assertIn("T-Shirts", doc.select("table > tbody > tr")[1].text)

    def test_delete(self):
        with scopes_disabled():
            c = ItemCategory.objects.create(event=self.event1, name="Entry tickets")
        doc = self.get_doc('/control/event/%s/%s/categories/%s/delete' % (self.orga1.slug, self.event1.slug, c.id))
        form_data = extract_form_fields(doc.select('.container-fluid form')[0])
        doc = self.post_doc('/control/event/%s/%s/categories/%s/delete' % (self.orga1.slug, self.event1.slug, c.id),
                            form_data)
        assert doc.select(".alert-success")
        self.assertNotIn("Entry tickets", doc.select("#page-wrapper")[0].text)
        with scopes_disabled():
            assert not ItemCategory.objects.filter(id=c.id).exists()


class QuestionsTest(ItemFormTest):

    def test_create(self):
        doc = self.get_doc('/control/event/%s/%s/questions/add' % (self.orga1.slug, self.event1.slug))
        form_data = extract_form_fields(doc.select('.container-fluid form')[0])
        form_data['question_0'] = 'What is your shoe size?'
        form_data['type'] = 'N'
        form_data['items'] = self.item1.id
        doc = self.post_doc('/control/event/%s/%s/questions/add' % (self.orga1.slug, self.event1.slug), form_data)
        assert doc.select(".alert-success")
        self.assertIn("shoe size", doc.select("#page-wrapper table")[0].text)

    def test_update_choices(self):
        with scopes_disabled():
            c = Question.objects.create(event=self.event1, question="What country are you from?", type="C", required=True)
            o1 = c.options.create(answer='Germany')
        doc = self.get_doc('/control/event/%s/%s/questions/%s/change' % (self.orga1.slug, self.event1.slug, c.id))
        form_data = extract_form_fields(doc.select('.container-fluid form')[0])
        form_data['form-TOTAL_FORMS'] = '1'
        form_data['form-INITIAL_FORMS'] = '1'
        form_data['form-MIN_NUM_FORMS'] = '0'
        form_data['form-MAX_NUM_FORMS'] = '1'
        form_data['form-0-id'] = o1.pk
        form_data['items'] = self.item1.id
        form_data['form-0-answer_0'] = 'England'
        self.post_doc('/control/event/%s/%s/questions/%s/change' % (self.orga1.slug, self.event1.slug, c.id),
                      form_data)
        c.refresh_from_db()
        with scopes_disabled():
            assert c.options.exists()
            assert str(c.options.first().answer) == 'England'

    def test_delete_choices(self):
        with scopes_disabled():
            c = Question.objects.create(event=self.event1, question="What country are you from?", type="C", required=True)
            o1 = c.options.create(answer='Germany')
        doc = self.get_doc('/control/event/%s/%s/questions/%s/change' % (self.orga1.slug, self.event1.slug, c.id))
        form_data = extract_form_fields(doc.select('.container-fluid form')[0])
        form_data['form-TOTAL_FORMS'] = '1'
        form_data['form-INITIAL_FORMS'] = '1'
        form_data['form-MIN_NUM_FORMS'] = '0'
        form_data['form-MAX_NUM_FORMS'] = '1'
        form_data['items'] = self.item1.id
        form_data['form-0-id'] = o1.pk
        form_data['form-0-answer_0'] = 'England'
        form_data['form-0-DELETE'] = 'yes'
        self.post_doc('/control/event/%s/%s/questions/%s/change' % (self.orga1.slug, self.event1.slug, c.id),
                      form_data)
        c.refresh_from_db()
        with scopes_disabled():
            assert not c.options.exists()

    def test_add_choices(self):
        with scopes_disabled():
            c = Question.objects.create(event=self.event1, question="What country are you from?", type="N", required=True)
        doc = self.get_doc('/control/event/%s/%s/questions/%s/change' % (self.orga1.slug, self.event1.slug, c.id))
        form_data = extract_form_fields(doc.select('.container-fluid form')[0])
        form_data['type'] = 'C'
        form_data['form-TOTAL_FORMS'] = '1'
        form_data['form-INITIAL_FORMS'] = '0'
        form_data['form-MIN_NUM_FORMS'] = '0'
        form_data['form-MAX_NUM_FORMS'] = '1'
        form_data['items'] = self.item1.id
        form_data['form-0-id'] = ''
        form_data['form-0-answer_0'] = 'Germany'
        self.post_doc('/control/event/%s/%s/questions/%s/change' % (self.orga1.slug, self.event1.slug, c.id),
                      form_data)
        with scopes_disabled():
            c = Question.objects.get(id=c.id)
            assert c.options.exists()
            assert str(c.options.first().answer) == 'Germany'

    def test_update(self):
        with scopes_disabled():
            c = Question.objects.create(event=self.event1, question="What is your shoe size?", type="N", required=True)
        doc = self.get_doc('/control/event/%s/%s/questions/%s/change' % (self.orga1.slug, self.event1.slug, c.id))
        form_data = extract_form_fields(doc.select('.container-fluid form')[0])
        form_data['items'] = self.item1.id
        form_data['question_0'] = 'How old are you?'
        doc = self.post_doc('/control/event/%s/%s/questions/%s/change' % (self.orga1.slug, self.event1.slug, c.id),
                            form_data)
        self.assertIn("How old", doc.select("#page-wrapper table")[0].text)
        self.assertNotIn("shoe size", doc.select("#page-wrapper table")[0].text)
        with scopes_disabled():
            c = Question.objects.get(id=c.id)
            self.assertTrue(c.required)
            assert str(Question.objects.get(id=c.id).question) == 'How old are you?'

    def test_delete(self):
        with scopes_disabled():
            c = Question.objects.create(event=self.event1, question="What is your shoe size?", type="N", required=True)
        doc = self.get_doc('/control/event/%s/%s/questions/%s/delete' % (self.orga1.slug, self.event1.slug, c.id))
        form_data = extract_form_fields(doc.select('.container-fluid form')[0])
        doc = self.post_doc('/control/event/%s/%s/questions/%s/delete' % (self.orga1.slug, self.event1.slug, c.id),
                            form_data)
        assert doc.select(".alert-success")
        self.assertNotIn("shoe size", doc.select("#page-wrapper")[0].text)
        with scopes_disabled():
            assert not Question.objects.filter(id=c.id).exists()

    def test_question_view(self):
        with scopes_disabled():
            c = Question.objects.create(event=self.event1, question="What is your shoe size?", type="N", required=True)

            item1 = Item.objects.create(event=self.event1, name="Standard", default_price=0, position=1)
            o = Order.objects.create(code='FOO', event=self.event1, email='dummy@dummy.test',
                                     status=Order.STATUS_PENDING, datetime=now(),
                                     expires=now() + datetime.timedelta(days=10),
                                     total=14, locale='en')
            op = OrderPosition.objects.create(order=o, item=item1, variation=None, price=Decimal("14"),
                                              attendee_name_parts={'full_name': "Peter"})
            op.answers.create(question=c, answer='42')
            op = OrderPosition.objects.create(order=o, item=item1, variation=None, price=Decimal("14"),
                                              attendee_name_parts={'full_name': "Michael"})
            op.answers.create(question=c, answer='42')
            op = OrderPosition.objects.create(order=o, item=item1, variation=None, price=Decimal("14"),
                                              attendee_name_parts={'full_name': "Petra"})
            op.answers.create(question=c, answer='39')

        doc = self.get_doc('/control/event/%s/%s/questions/%s/' % (self.orga1.slug, self.event1.slug, c.id))
        tbl = doc.select('.container-fluid table.table-bordered tbody')[0]
        assert tbl.select('tr')[0].select('td')[0].text.strip() == '42'
        assert tbl.select('tr')[0].select('td')[1].text.strip() == '2'
        assert tbl.select('tr')[1].select('td')[0].text.strip() == '39'
        assert tbl.select('tr')[1].select('td')[1].text.strip() == '1'

        doc = self.get_doc('/control/event/%s/%s/questions/%s/?status=p' % (self.orga1.slug, self.event1.slug, c.id))
        assert not doc.select('.container-fluid table.table-bordered tbody')

        o.status = Order.STATUS_PAID
        o.save()
        doc = self.get_doc('/control/event/%s/%s/questions/%s/?status=p' % (self.orga1.slug, self.event1.slug, c.id))
        tbl = doc.select('.container-fluid table.table-bordered tbody')[0]
        assert tbl.select('tr')[0].select('td')[0].text.strip() == '42'

    def test_set_dependency(self):
        with scopes_disabled():
            q1 = Question.objects.create(event=self.event1, question="What country are you from?", type="C", required=True)
            q2 = Question.objects.create(event=self.event1, question="What city are you from?", type="T", required=True)
            o1 = q1.options.create(answer='Germany')
        doc = self.get_doc('/control/event/%s/%s/questions/%s/change' % (self.orga1.slug, self.event1.slug, q2.id))
        form_data = extract_form_fields(doc.select('.container-fluid form')[0])
        form_data['items'] = self.item1.id
        form_data['dependency_question'] = q1.pk
        form_data['dependency_values'] = o1.identifier
        doc = self.post_doc('/control/event/%s/%s/questions/%s/change' % (self.orga1.slug, self.event1.slug, q2.id),
                            form_data)
        assert doc.select(".alert-success")
        q2.refresh_from_db()
        assert q2.dependency_question == q1
        assert q2.dependency_values == [o1.identifier]

    def test_set_dependency_circular(self):
        with scopes_disabled():
            q1 = Question.objects.create(event=self.event1, question="What country are you from?", type="C", required=True)
            o1 = q1.options.create(answer='Germany')
            q2 = Question.objects.create(event=self.event1, question="What city are you from?", type="C", required=True,
                                         dependency_question=q1, dependency_values=[o1.identifier])
        doc = self.get_doc('/control/event/%s/%s/questions/%s/change' % (self.orga1.slug, self.event1.slug, q1.id))
        form_data = extract_form_fields(doc.select('.container-fluid form')[0])
        form_data['dependency_question'] = q2.pk
        form_data['dependency_values'] = '1'
        doc = self.post_doc('/control/event/%s/%s/questions/%s/change' % (self.orga1.slug, self.event1.slug, q1.id),
                            form_data)
        assert not doc.select(".alert-success")

    def test_set_dependency_to_non_choice(self):
        with scopes_disabled():
            q1 = Question.objects.create(event=self.event1, question="What country are you from?", type="N", required=True)
            q2 = Question.objects.create(event=self.event1, question="What city are you from?", type="T", required=True)
        doc = self.get_doc('/control/event/%s/%s/questions/%s/change' % (self.orga1.slug, self.event1.slug, q2.id))
        form_data = extract_form_fields(doc.select('.container-fluid form')[0])
        form_data['dependency_question'] = q1.pk
        form_data['dependency_values'] = '1'
        doc = self.post_doc('/control/event/%s/%s/questions/%s/change' % (self.orga1.slug, self.event1.slug, q2.id),
                            form_data)
        assert not doc.select(".alert-success")


class QuotaTest(ItemFormTest):

    def test_create(self):
        doc = self.get_doc('/control/event/%s/%s/quotas/add' % (self.orga1.slug, self.event1.slug))
        form_data = extract_form_fields(doc.select('.container-fluid form')[0])
        form_data['name'] = 'Full house'
        form_data['size'] = '500'
        form_data['itemvars'] = str(self.item1.pk)
        doc = self.post_doc('/control/event/%s/%s/quotas/add' % (self.orga1.slug, self.event1.slug), form_data)
        assert doc.select(".alert-success")
        self.assertIn("Full house", doc.select("#page-wrapper table")[0].text)

    def test_update(self):
        with scopes_disabled():
            c = Quota.objects.create(event=self.event1, name="Full house", size=500)
            item1 = Item.objects.create(event=self.event1, name="Standard", default_price=0)
            item2 = Item.objects.create(event=self.event1, name="Business", default_price=0)
            ItemVariation.objects.create(item=item2, value="Silver")
            ItemVariation.objects.create(item=item2, value="Gold")
        doc = self.get_doc('/control/event/%s/%s/quotas/%s/change' % (self.orga1.slug, self.event1.slug, c.id))
        [i for i in doc.select('[name=itemvars]') if i.get('value') == str(item1.id)][0]['checked'] = 'checked'
        form_data = extract_form_fields(doc.select('.container-fluid form')[0])
        form_data['size'] = '350'
        doc = self.post_doc('/control/event/%s/%s/quotas/%s/change' % (self.orga1.slug, self.event1.slug, c.id),
                            form_data)
        doc = self.get_doc('/control/event/%s/%s/quotas/' % (self.orga1.slug, self.event1.slug))
        self.assertIn("350", doc.select("#page-wrapper table")[0].text)
        self.assertNotIn("500", doc.select("#page-wrapper table")[0].text)
        with scopes_disabled():
            assert Quota.objects.get(id=c.id).size == 350
            assert item1 in Quota.objects.get(id=c.id).items.all()

    def test_update_subevent(self):
        self.event1.has_subevents = True
        self.event1.save()
        with scopes_disabled():
            se1 = self.event1.subevents.create(name="Foo", date_from=now())
            se2 = self.event1.subevents.create(name="Bar", date_from=now())
            c = Quota.objects.create(event=self.event1, name="Full house", size=500, subevent=se1)
            c.items.add(self.item1)
        doc = self.get_doc('/control/event/%s/%s/quotas/%s/change' % (self.orga1.slug, self.event1.slug, c.id))
        form_data = extract_form_fields(doc.select('.container-fluid form')[0])
        form_data['subevent'] = se2.pk
        self.post_doc('/control/event/%s/%s/quotas/%s/change' % (self.orga1.slug, self.event1.slug, c.id),
                      form_data)
        with scopes_disabled():
            assert Quota.objects.get(id=c.id).subevent == se2

    def test_delete(self):
        with scopes_disabled():
            c = Quota.objects.create(event=self.event1, name="Full house", size=500)
        doc = self.get_doc('/control/event/%s/%s/quotas/%s/delete' % (self.orga1.slug, self.event1.slug, c.id))
        form_data = extract_form_fields(doc.select('.container-fluid form')[0])
        doc = self.post_doc('/control/event/%s/%s/quotas/%s/delete' % (self.orga1.slug, self.event1.slug, c.id),
                            form_data)
        assert doc.select(".alert-success")
        self.assertNotIn("Full house", doc.select("#page-wrapper")[0].text)
        with scopes_disabled():
            assert not Quota.objects.filter(id=c.id).exists()

    def test_reopen(self):
        with scopes_disabled():
            c = Quota.objects.create(event=self.event1, name="Full house", size=500,
                                     close_when_sold_out=True, closed=True)
        self.post_doc('/control/event/%s/%s/quotas/%s/' % (self.orga1.slug, self.event1.slug, c.id),
                      {'reopen': 'true'})
        with scopes_disabled():
            c.refresh_from_db()
            assert not c.closed
            assert c.close_when_sold_out

    def test_reopen_and_disable(self):
        with scopes_disabled():
            c = Quota.objects.create(event=self.event1, name="Full house", size=500,
                                     close_when_sold_out=True, closed=True)
        self.post_doc('/control/event/%s/%s/quotas/%s/' % (self.orga1.slug, self.event1.slug, c.id),
                      {'disable': 'true'})
        with scopes_disabled():
            c.refresh_from_db()
            assert not c.closed
            assert not c.close_when_sold_out


class ItemsTest(ItemFormTest):

    @scopes_disabled()
    def setUp(self):
        super().setUp()
        self.item2 = Item.objects.create(event=self.event1, name="Business", default_price=0, position=2,
                                         description="If your ticket is paid by your employer",
                                         active=True, available_until=now() + datetime.timedelta(days=4),
                                         require_voucher=True, allow_cancel=False)
        self.var1 = ItemVariation.objects.create(item=self.item2, value="Silver")
        self.var2 = ItemVariation.objects.create(item=self.item2, value="Gold")
        self.addoncat = ItemCategory.objects.create(event=self.event1, name="Item category")

    def test_move(self):
        self.client.post('/control/event/%s/%s/items/%s/down' % (self.orga1.slug, self.event1.slug, self.item1.id),)
        self.item1.refresh_from_db()
        self.item2.refresh_from_db()
        assert self.item1.position > self.item2.position
        self.client.post('/control/event/%s/%s/items/%s/up' % (self.orga1.slug, self.event1.slug, self.item1.id),)
        self.item1.refresh_from_db()
        self.item2.refresh_from_db()
        assert self.item1.position < self.item2.position

    def test_reorder(self):
        self.client.post('/control/event/%s/%s/items/reorder/0/' % (self.orga1.slug, self.event1.slug), {
            'ids': [str(self.item2.id), str(self.item1.id)],
        }, content_type='application/json')
        self.item1.refresh_from_db()
        self.item2.refresh_from_db()
        assert self.item1.position > self.item2.position
        assert self.item1.category is None
        assert self.item2.category is None
        self.client.post('/control/event/%s/%s/items/reorder/%s/' % (self.orga1.slug, self.event1.slug, self.addoncat.id), {
            'ids': [str(self.item1.id), str(self.item2.id)],
        }, content_type='application/json')
        self.item1.refresh_from_db()
        self.item2.refresh_from_db()
        assert self.item1.position < self.item2.position
        assert self.item1.category.id == self.addoncat.id
        assert self.item2.category.id == self.addoncat.id

    def test_create(self):
        self.client.post('/control/event/%s/%s/items/add' % (self.orga1.slug, self.event1.slug), {
            'name_0': 'T-Shirt',
            'default_price': '23.00',
            'tax_rate': '19.00'
        })
        resp = self.client.get('/control/event/%s/%s/items/' % (self.orga1.slug, self.event1.slug))
        assert 'T-Shirt' in resp.content.decode()

    def test_update(self):
        doc = self.get_doc('/control/event/%s/%s/items/%d/' % (self.orga1.slug, self.event1.slug, self.item2.id))
        d = extract_form_fields(doc.select('.container-fluid form')[0])
        d.update({
            'name_0': 'Standard',
            'default_price': '23.00',
            'tax_rate': '19.00',
            'active': 'yes',
            'allow_cancel': 'yes',
            'sales_channels': 'web'
        })
        self.client.post('/control/event/%s/%s/items/%d/' % (self.orga1.slug, self.event1.slug, self.item1.id), d)
        self.item1.refresh_from_db()
        assert self.item1.default_price == Decimal('23.00')

    def test_update_validate_giftcard(self):
        doc = self.get_doc('/control/event/%s/%s/items/%d/' % (self.orga1.slug, self.event1.slug, self.item2.id))
        d = extract_form_fields(doc.select('.container-fluid form')[0])
        d.update({
            'name_0': 'Standard',
            'default_price': '23.00',
            'admission': 'on',
            'issue_giftcard': 'on',
            'active': 'yes',
            'allow_cancel': 'yes',
            'sales_channels': 'web'
        })
        self.client.post('/control/event/%s/%s/items/%d/' % (self.orga1.slug, self.event1.slug, self.item1.id), d)
        self.item1.refresh_from_db()
        assert not self.item1.issue_giftcard

    def test_manipulate_addons(self):
        doc = self.get_doc('/control/event/%s/%s/items/%d/' % (self.orga1.slug, self.event1.slug, self.item2.id))
        d = extract_form_fields(doc.select('.container-fluid form')[0])

        d.update({
            'addons-TOTAL_FORMS': '1',
            'addons-INITIAL_FORMS': '0',
            'addons-MIN_NUM_FORMS': '0',
            'addons-MAX_NUM_FORMS': '1000',
            'addons-0-id': '',
            'addons-0-addon_category': str(self.addoncat.pk),
            'addons-0-min_count': '1',
            'addons-0-max_count': '2',
        })
        self.client.post('/control/event/%s/%s/items/%d/' % (self.orga1.slug, self.event1.slug, self.item2.id), d)
        with scopes_disabled():
            assert self.item2.addons.exists()
            assert self.item2.addons.first().addon_category == self.addoncat
            a = self.item2.addons.first()
        d.update({
            'addons-TOTAL_FORMS': '1',
            'addons-INITIAL_FORMS': '1',
            'addons-MIN_NUM_FORMS': '0',
            'addons-MAX_NUM_FORMS': '1000',
            'addons-0-id': str(a.pk),
            'addons-0-addon_category': str(self.addoncat.pk),
            'addons-0-min_count': '1',
            'addons-0-max_count': '2',
            'addons-0-DELETE': 'on',
        })
        self.client.post('/control/event/%s/%s/items/%d/' % (self.orga1.slug, self.event1.slug, self.item2.id), d)
        with scopes_disabled():
            assert not self.item2.addons.exists()

        # Do not allow duplicates
        d.update({
            'addons-TOTAL_FORMS': '2',
            'addons-INITIAL_FORMS': '0',
            'addons-MIN_NUM_FORMS': '0',
            'addons-MAX_NUM_FORMS': '1000',
            'addons-0-id': '',
            'addons-0-addon_category': str(self.addoncat.pk),
            'addons-0-min_count': '1',
            'addons-0-max_count': '2',
            'addons-1-id': '',
            'addons-1-addon_category': str(self.addoncat.pk),
            'addons-1-min_count': '1',
            'addons-1-max_count': '2',
        })
        self.client.post('/control/event/%s/%s/items/%d/' % (self.orga1.slug, self.event1.slug, self.item2.id), d)
        with scopes_disabled():
            assert self.item2.addons.count() == 1

    def test_manipulate_bundles(self):
        doc = self.get_doc('/control/event/%s/%s/items/%d/' % (self.orga1.slug, self.event1.slug, self.item2.id))
        d = extract_form_fields(doc.select('.container-fluid form')[0])

        d.update({
            'bundles-TOTAL_FORMS': '1',
            'bundles-INITIAL_FORMS': '0',
            'bundles-MIN_NUM_FORMS': '0',
            'bundles-MAX_NUM_FORMS': '1000',
            'bundles-0-id': '',
            'bundles-0-itemvar': str(self.item1.pk),
            'bundles-0-count': '2',
            'bundles-0-designated_price': '',
        })
        self.client.post('/control/event/%s/%s/items/%d/' % (self.orga1.slug, self.event1.slug, self.item2.id), d)
        with scopes_disabled():
            assert self.item2.bundles.exists()
            assert self.item2.bundles.first().bundled_item == self.item1
            assert self.item2.bundles.first().designated_price == Decimal('0.000')
        d.update({
            'bundles-TOTAL_FORMS': '1',
            'bundles-INITIAL_FORMS': '1',
            'bundles-MIN_NUM_FORMS': '0',
            'bundles-MAX_NUM_FORMS': '1000',
            'bundles-0-id': str(self.item2.bundles.first().pk),
            'bundles-0-itemvar': str(self.item1.pk),
            'bundles-0-count': '2',
            'bundles-0-designated_price': '2.00',
            'bundles-0-DELETE': 'on',
        })
        self.client.post('/control/event/%s/%s/items/%d/' % (self.orga1.slug, self.event1.slug, self.item2.id), d)
        with scopes_disabled():
            assert not self.item2.bundles.exists()

        # Do not allow self-reference
        d.update({
            'bundles-TOTAL_FORMS': '1',
            'bundles-INITIAL_FORMS': '0',
            'bundles-MIN_NUM_FORMS': '0',
            'bundles-MAX_NUM_FORMS': '1000',
            'bundles-0-id': '',
            'bundles-0-itemvar': str(self.item2.pk),
            'bundles-0-count': '2',
            'bundles-0-designated_price': '2.00',
        })
        self.client.post('/control/event/%s/%s/items/%d/' % (self.orga1.slug, self.event1.slug, self.item2.id), d)
        with scopes_disabled():
            assert not self.item2.bundles.exists()

        # Do not allow multi-level bundles
        with scopes_disabled():
            self.item1.bundles.create(bundled_item=self.item1, count=1, designated_price=0)
        d.update({
            'bundles-TOTAL_FORMS': '1',
            'bundles-INITIAL_FORMS': '0',
            'bundles-MIN_NUM_FORMS': '0',
            'bundles-MAX_NUM_FORMS': '1000',
            'bundles-0-id': '',
            'bundles-0-itemvar': str(self.item1.pk),
            'bundles-0-count': '2',
            'bundles-0-designated_price': '2.00',
        })
        self.client.post('/control/event/%s/%s/items/%d/' % (self.orga1.slug, self.event1.slug, self.item2.id), d)
        with scopes_disabled():
            assert not self.item2.bundles.exists()

    def test_update_variations(self):
        doc = self.get_doc('/control/event/%s/%s/items/%d/' % (self.orga1.slug, self.event1.slug, self.item2.id))
        d = extract_form_fields(doc.select('.container-fluid form')[0])
        d.update({
            'variations-TOTAL_FORMS': '2',
            'variations-INITIAL_FORMS': '2',
            'variations-MIN_NUM_FORMS': '0',
            'variations-MAX_NUM_FORMS': '1000',
            'variations-0-id': str(self.var1.pk),
            'variations-0-value_0': 'Bronze',
            'variations-0-active': 'yes',
            'variations-1-id': str(self.var2.pk),
            'variations-1-value_0': 'Gold',
            'variations-1-active': 'yes',
        })
        self.client.post('/control/event/%s/%s/items/%d/' % (self.orga1.slug, self.event1.slug, self.item2.id), d)
        self.var1.refresh_from_db()
        assert str(self.var1.value) == 'Bronze'

    def test_delete_variation(self):
        doc = self.get_doc('/control/event/%s/%s/items/%d/' % (self.orga1.slug, self.event1.slug, self.item2.id))
        d = extract_form_fields(doc.select('.container-fluid form')[0])
        d.update({
            'variations-TOTAL_FORMS': '2',
            'variations-INITIAL_FORMS': '2',
            'variations-MIN_NUM_FORMS': '0',
            'variations-MAX_NUM_FORMS': '1000',
            'variations-0-id': str(self.var1.pk),
            'variations-0-value_0': 'Bronze',
            'variations-0-active': 'yes',
            'variations-1-id': str(self.var2.pk),
            'variations-1-value_0': 'Gold',
            'variations-1-active': 'yes',
            'variations-1-DELETE': 'yes',
        })
        self.client.post('/control/event/%s/%s/items/%d/' % (self.orga1.slug, self.event1.slug, self.item2.id), d)
        with scopes_disabled():
            assert not self.item2.variations.filter(pk=self.var2.pk).exists()

    def test_delete(self):
        self.client.post('/control/event/%s/%s/items/%d/delete' % (self.orga1.slug, self.event1.slug, self.item1.id),
                         {})
        with scopes_disabled():
            assert not self.event1.items.filter(pk=self.item1.pk).exists()
        self.client.post('/control/event/%s/%s/items/%d/delete' % (self.orga1.slug, self.event1.slug, self.item2.id),
                         {})
        with scopes_disabled():
            assert not self.event1.items.filter(pk=self.item2.pk).exists()

    def test_delete_ordered(self):
        with scopes_disabled():
            o = Order.objects.create(
                code='FOO', event=self.event1, email='dummy@dummy.test',
                status=Order.STATUS_PENDING,
                datetime=now(), expires=now() + datetime.timedelta(days=10),
                total=14, locale='en'
            )
            OrderPosition.objects.create(
                order=o,
                item=self.item1,
                variation=None,
                price=Decimal("14"),
                attendee_name_parts={'full_name': "Peter"}
            )
        self.client.post('/control/event/%s/%s/items/%d/delete' % (self.orga1.slug, self.event1.slug, self.item1.id),
                         {})
        with scopes_disabled():
            assert self.event1.items.filter(pk=self.item1.pk).exists()
        self.item1.refresh_from_db()
        assert not self.item1.active

    def test_create_copy(self):
        with scopes_disabled():
            q = Question.objects.create(event=self.event1, question="Size", type="N")
            q.items.add(self.item2)
        self.item2.sales_channels = ["web", "bar"]
        prop = self.event1.item_meta_properties.create(name="Foo")
        self.item2.meta_values.create(property=prop, value="Bar")

        doc = self.get_doc('/control/event/%s/%s/items/add?copy_from=%d' % (self.orga1.slug, self.event1.slug, self.item2.pk))
        data = extract_form_fields(doc.select("form")[0])
        self.client.post('/control/event/%s/%s/items/add' % (self.orga1.slug, self.event1.slug), {
            **data,
            'name_0': 'Intermediate',
            'default_price': '23.00',
            'tax_rate': '19.00',
            'copy_from': str(self.item2.pk),
            'has_variations': '1'
        })
        with scopes_disabled():
            i_old = Item.objects.get(name__icontains='Business')
            i_new = Item.objects.get(name__icontains='Intermediate')
            assert i_new.category == i_old.category
            assert i_new.description == i_old.description
            assert i_new.active == i_old.active
            assert i_new.available_from == i_old.available_from
            assert i_new.available_until == i_old.available_until
            assert i_new.require_voucher == i_old.require_voucher
            assert i_new.hide_without_voucher == i_old.hide_without_voucher
            assert i_new.allow_cancel == i_old.allow_cancel
            assert i_new.sales_channels == i_old.sales_channels
            assert i_new.meta_data == i_old.meta_data == {"Foo": "Bar"}
            assert set(i_new.questions.all()) == set(i_old.questions.all())
            assert set([str(v.value) for v in i_new.variations.all()]) == set([str(v.value) for v in i_old.variations.all()])

    def test_add_to_existing_quota(self):
        with scopes_disabled():
            q = Quota.objects.create(event=self.event1, name="New Test Quota", size=50)

        doc = self.get_doc('/control/event/%s/%s/items/add' % (self.orga1.slug, self.event1.slug))
        form_data = extract_form_fields(doc.select('.container-fluid form')[0])
        form_data['name_0'] = 'Existing'
        form_data['default_price'] = '2.00'
        form_data['quota_option'] = 'existing'
        form_data['quota_add_existing'] = str(q.pk)
        doc = self.post_doc('/control/event/%s/%s/items/add' % (self.orga1.slug, self.event1.slug), form_data)

        with scopes_disabled():
            i = Item.objects.get(name__icontains='Existing')

            assert doc.select(".alert-success")
            assert q.items.filter(pk=i.pk).exists()

    def test_add_to_new_quota(self):
        doc = self.get_doc('/control/event/%s/%s/items/add' % (self.orga1.slug, self.event1.slug))
        form_data = extract_form_fields(doc.select('.container-fluid form')[0])
        form_data['name_0'] = 'New Item'
        form_data['default_price'] = '2.00'
        form_data['quota_option'] = 'new'
        form_data['quota_add_new_name'] = 'New Quota'
        form_data['quota_add_new_size'] = '200'
        doc = self.post_doc('/control/event/%s/%s/items/add' % (self.orga1.slug, self.event1.slug), form_data)

        assert doc.select(".alert-success")
        with scopes_disabled():
            assert Quota.objects.filter(name__icontains='New Quota').exists()
            assert Item.objects.filter(name__icontains='New Item').exists()
            i = Item.objects.get(name__icontains='New Item')
            q = Quota.objects.get(name__icontains='New Quota')
            assert q.items.filter(pk=i.pk).exists()


class DiscountTest(ItemFormTest):

    def test_create(self):
        doc = self.get_doc('/control/event/%s/%s/discounts/add' % (self.orga1.slug, self.event1.slug))
        form_data = extract_form_fields(doc.select('.container-fluid form')[0])
        form_data['internal_name'] = 'Group discount'
        form_data['condition_min_count'] = '2'
        form_data['benefit_discount_matching_percent'] = '20'
        doc = self.post_doc('/control/event/%s/%s/discounts/add' % (self.orga1.slug, self.event1.slug), form_data)
        assert doc.select(".alert-success")
        self.assertIn("Group discount", doc.select("#page-wrapper table")[0].text)

    def test_update(self):
        c = Discount.objects.create(event=self.event1, internal_name="2 for 1")
        doc = self.get_doc('/control/event/%s/%s/discounts/%s/' % (self.orga1.slug, self.event1.slug, c.id))
        form_data = extract_form_fields(doc.select('.container-fluid form')[0])
        form_data['internal_name'] = 'Group discount'
        form_data['condition_min_count'] = '2'
        form_data['benefit_discount_matching_percent'] = '20'
        doc = self.post_doc('/control/event/%s/%s/discounts/%s/' % (self.orga1.slug, self.event1.slug, c.id),
                            form_data)
        assert doc.select(".alert-success")
        self.assertIn("Group discount", doc.select("#page-wrapper table")[0].text)
        self.assertNotIn("2 for 1", doc.select("#page-wrapper table")[0].text)
        with scopes_disabled():
            assert str(Discount.objects.get(id=c.id).benefit_discount_matching_percent) == '20.00'

    def test_sort(self):
        with scopes_disabled():
            c1 = Discount.objects.create(event=self.event1, internal_name="Group discount", condition_min_value=2,
                                         benefit_discount_matching_percent=20)
            Discount.objects.create(event=self.event1, internal_name="Big group", condition_min_value=5,
                                    benefit_discount_matching_percent=40)
        doc = self.get_doc('/control/event/%s/%s/discounts/' % (self.orga1.slug, self.event1.slug))
        self.assertIn("Group discount", doc.select("table > tbody > tr")[0].text)
        self.assertIn("Big group", doc.select("table > tbody > tr")[1].text)

        self.client.post('/control/event/%s/%s/discounts/%s/down' % (self.orga1.slug, self.event1.slug, c1.id))
        doc = self.get_doc('/control/event/%s/%s/discounts/' % (self.orga1.slug, self.event1.slug))
        self.assertIn("Group discount", doc.select("table > tbody > tr")[1].text)
        self.assertIn("Big group", doc.select("table > tbody > tr")[0].text)

        self.client.post('/control/event/%s/%s/discounts/%s/up' % (self.orga1.slug, self.event1.slug, c1.id))
        doc = self.get_doc('/control/event/%s/%s/discounts/' % (self.orga1.slug, self.event1.slug))
        self.assertIn("Group discount", doc.select("table > tbody > tr")[0].text)
        self.assertIn("Big group", doc.select("table > tbody > tr")[1].text)

    def test_delete(self):
        with scopes_disabled():
            c = Discount.objects.create(event=self.event1, internal_name="Group discount", condition_min_value=2,
                                        benefit_discount_matching_percent=20)
        doc = self.get_doc('/control/event/%s/%s/discounts/%s/delete' % (self.orga1.slug, self.event1.slug, c.id))
        form_data = extract_form_fields(doc.select('.container-fluid form')[0])
        doc = self.post_doc('/control/event/%s/%s/discounts/%s/delete' % (self.orga1.slug, self.event1.slug, c.id),
                            form_data)
        assert doc.select(".alert-success")
        self.assertNotIn("Group discount", doc.select("#page-wrapper")[0].text)
        with scopes_disabled():
            assert not Discount.objects.filter(id=c.id).exists()