def test_get_rows_for_template(self):
self.talk.room = self.room2
timetable = Timetable()
timetable.add_talk(self.talk)
number_of_sessions = len(Talk.get_talk_sessions())
self.assertEquals(number_of_sessions, len(timetable.get_rows_for_template()))
for row in timetable.get_rows_for_template():
self.assertEquals(2, len(row.talks_by_room()))
if row.session == 1:
self.assertEquals(row.talks_by_room()[0], None)
self.assertEquals(row.talks_by_room()[1], self.talk)
def show_timetable(request):
user = request.session.get('user')
timetable = Timetable()
rooms = Room.get_rooms()
sessions = Talk.get_talk_sessions()
return render_to_response('timetable_app.html', {'user': user, 'rooms': rooms, 'sessions': sessions, 'rows': timetable.get_rows_for_template()})
