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