10
10
11
11
12
12
class EventsViewsTests (TestCase ):
13
- def setUp (self ):
14
- self .user = get_user_model ().objects .create_user (username = 'username' , password = 'password' )
15
- self .calendar = Calendar .objects .create (creator = self .user , slug = "test-calendar" )
16
- self .event = Event .objects .create (creator = self .user , calendar = self .calendar )
13
+ @classmethod
14
+ def setUpTestData (cls ):
15
+ cls .user = get_user_model ().objects .create_user (username = 'username' , password = 'password' )
16
+ cls .calendar = Calendar .objects .create (creator = cls .user , slug = "test-calendar" )
17
+ cls .event = Event .objects .create (creator = cls .user , calendar = cls .calendar )
18
+ cls .event_past = Event .objects .create (title = 'Past Event' , creator = cls .user , calendar = cls .calendar )
17
19
18
- self .now = timezone .now ()
20
+ cls .now = timezone .now ()
19
21
20
- recurring_time_dtstart = self .now + datetime .timedelta (days = 3 )
22
+ recurring_time_dtstart = cls .now + datetime .timedelta (days = 3 )
21
23
recurring_time_dtend = recurring_time_dtstart + datetime .timedelta (days = 5 )
22
24
23
- self .rule = RecurringRule .objects .create (
24
- event = self .event ,
25
+ cls .rule = RecurringRule .objects .create (
26
+ event = cls .event ,
25
27
begin = recurring_time_dtstart ,
26
28
finish = recurring_time_dtend ,
27
29
)
30
+ cls .rule_past = RecurringRule .objects .create (
31
+ event = cls .event_past ,
32
+ begin = cls .now - datetime .timedelta (days = 2 ),
33
+ finish = cls .now - datetime .timedelta (days = 1 ),
34
+ )
28
35
29
36
def test_calendar_list (self ):
30
37
calendars_count = Calendar .objects .count ()
@@ -50,15 +57,6 @@ def test_event_list_past(self):
50
57
url = reverse ('events:event_list_past' , kwargs = {"calendar_slug" : self .calendar .slug })
51
58
response = self .client .get (url )
52
59
53
- self .assertEqual (response .status_code , 200 )
54
- self .assertEqual (len (response .context ['object_list' ]), 0 )
55
-
56
- self .rule .begin = self .now - datetime .timedelta (days = 3 )
57
- self .rule .finish = self .now - datetime .timedelta (days = 2 )
58
- self .rule .save ()
59
-
60
- response = self .client .get (url )
61
-
62
60
self .assertEqual (response .status_code , 200 )
63
61
self .assertEqual (len (response .context ['object_list' ]), 1 )
64
62
@@ -108,7 +106,7 @@ def test_event_list_date(self):
108
106
response = self .client .get (url )
109
107
self .assertEqual (response .status_code , 200 )
110
108
self .assertEqual (response .context ['object' ], dt .date ())
111
- self .assertEqual (len (response .context ['object_list' ]), 1 )
109
+ self .assertEqual (len (response .context ['object_list' ]), 2 )
112
110
113
111
def test_eventlocation_list (self ):
114
112
venue = EventLocation .objects .create (
0 commit comments