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 testAddStudent(self): user_type = "student" first_name = "John" last_name = "Smith" self.assertEqual(api.create_student(user_type, first_name, last_name), 87)