fix stupid bug
[wolnelektury.git] / src / funding / tests.py
1 # -*- coding: utf-8 -*-
2 # This file is part of Wolnelektury, licensed under GNU Affero GPLv3 or later.
3 # Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information.
4 #
5 from datetime import date, timedelta
6 from django.test import TestCase
7 from .models import Offer, Perk
8
9
10 class PerksTest(TestCase):
11     def setUp(self):
12         self.today = date.today()
13         self.offer1 = Offer.objects.create(
14             author='author1', title='title1', slug='slug1',
15             target=100, start=self.today, end=self.today)
16
17     def test_perks(self):
18         perk = Perk.objects.create(price=20, name='Perk 20')
19         perk1 = Perk.objects.create(offer=self.offer1, price=50, name='Perk 50')
20         offer2 = Offer.objects.create(
21             author='author2', title='title2', slug='slug2',
22             target=100, start=self.today-timedelta(1), end=self.today-timedelta(1))
23         perk2 = Perk.objects.create(offer=offer2, price=1, name='Perk 1')
24
25         self.assertEqual(
26             set(self.offer1.get_perks(10)),
27             set())
28         self.assertEqual(
29             set(self.offer1.get_perks(70)),
30             {perk, perk1})
31
32
33 class FundingTest(TestCase):
34     def setUp(self):
35         self.today = date.today()
36         self.offer_past = Offer.objects.create(
37             author='an-author', title='past', slug='past',
38             target=100, start=self.today-timedelta(1), end=self.today-timedelta(1))
39         self.offer_current = Offer.objects.create(
40             author='an-author', title='current', slug='current',
41             target=100, start=self.today, end=self.today)
42         self.offer_future = Offer.objects.create(
43             author='an-author', title='future', slug='future',
44             target=100, start=self.today+timedelta(1), end=self.today+timedelta(1))
45
46     def test_current(self):
47         self.assertTrue(self.offer_current.is_current())
48         self.assertFalse(self.offer_past.is_current())
49         self.assertEqual(Offer.current(), self.offer_current)
50         self.assertEqual(
51             set(Offer.past()),
52             {self.offer_past}
53         )
54         self.assertEqual(
55             set(Offer.public()),
56             {self.offer_past, self.offer_current}
57         )
58
59     def test_interrupt(self):
60         # A new offer starts, ending the previously current one.
61         offer_interrupt = Offer.objects.create(
62             author='an-author', title='interrupt', slug='interrupt',
63             target=100, start=self.today-timedelta(1), end=self.today+timedelta(1))
64
65         self.assertTrue(offer_interrupt.is_current())
66         self.assertFalse(self.offer_current.is_current())
67         self.assertEqual(Offer.current(), offer_interrupt)
68         self.assertEqual(
69             set(Offer.past()),
70             {self.offer_past, self.offer_current}
71         )
72         self.assertEqual(
73             set(Offer.public()),
74             {self.offer_past, self.offer_current, offer_interrupt}
75         )