def addclass(): class_info = json.loads(request.data) cohort_name = (class_info.get("cohort")).get("name") teacher_id = session['user'] cohort_id = api.add_new_cohort(cohort_name, teacher_id) list_of_students = class_info.get("students") for student in list_of_students: user_type = "student" first_name = student.get("first_name") last_name = student.get("last_name") student_id = api.create_student(user_type, first_name, last_name) api.add_student_to_cohort(student_id, cohort_id) return "Success"
def testAddStudentToCohort(self): student_id = 87 cohort_id = 7 self.assertEqual(api.add_student_to_cohort(student_id, cohort_id), "Successfully Added!")