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()})