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")
