def test_assign_ta_too_many(self): ta1 = models.User() ta1.email = "*****@*****.**" ta1.type = "ta" ta1.save() ta2 = models.User() ta2.email = "*****@*****.**" ta2.type = "ta" ta2.save() ta3 = models.User() ta3.email = "*****@*****.**" ta3.type = "ta" ta3.save() course = models.Course() course.num_labs = 2 course.current_num_TA = 0 course.num_lectures = 1 course.instructor = "DEFAULT" course.course_id = "301" course.course_department = "COMPSCI" course.save() proper = Commands.assign_ta_to_course(ta1.email, course.course_id, course.course_department) self.assertEqual(proper, "TA Assigned!") proper = Commands.assign_ta_to_course(ta2.email, course.course_id, course.course_department) self.assertEqual(proper, "TA Assigned!") proper = Commands.assign_ta_to_course(ta3.email, course.course_id, course.course_department) self.assertEqual(proper, "Too Many TA's Assigned")
def post(self, request): request.session.set_expiry(300) email = request.POST["email"] course_id = request.POST["course_id"] course_department = request.POST["course_department"] response = Commands.assign_ta_to_course(email, course_id, course_department) if response == "TA Assigned!": messages.success(request, response) else: messages.error(request, response) return render(request, 'main/assign_ta.html')
def test_assign_ta_ins(self): ins1 = models.User() ins1.email = "*****@*****.**" ins1.type = "instructor" ins1.save() course = models.Course() course.num_labs = 2 course.current_num_TA = 0 course.num_lectures = 1 course.instructor = "DEFAULT" course.course_id = "301" course.course_department = "COMPSCI" course.save() proper = Commands.assign_ta_to_course(ins1.email, course.course_id, course.course_department) self.assertEqual(proper, "no such ta")