def test_update(self): enrollment = Enrollment.create_one(**enrollment_data[0], course=self.courses[0]) db.session.commit() assert enrollment.status == 'open' enrollment.update(status='closed') db.session.commit() assert Enrollment.get_one(guid=enrollment.guid).status == 'closed'
def test_relationship__people(self): enrollment = Enrollment.create_one(**enrollment_data[0], course=self.courses[0]) db.session.commit() assert enrollment.people == [] enrollment.people.append(self.users[0]) enrollment.people.append(self.users[1]) db.session.commit() assert Enrollment.get_one(guid=enrollment.guid).people == self.users enrollment.people.remove(enrollment.people[0]) db.session.commit() assert Enrollment.get_one(guid=enrollment.guid).people == [ self.users[1] ]
def test_create(self): enrollment = Enrollment.create_one(**enrollment_data[0], guid=self.enrollment_guid, course=self.courses[0]) db.session.commit() enrollment = Enrollment.get_one(guid=enrollment.guid) assert enrollment.status == 'open' assert enrollment.start_date == enrollment_data[0]['start_date'] assert enrollment.end_date == enrollment_data[0]['end_date'] assert enrollment.course.guid == self.courses[0].guid assert enrollment.people == []
def get(guid): return Enrollment.get_one(guid=guid)
def test_delete(self): enrollment = Enrollment.get_one(guid=self.enrollment_guid) enrollment.delete() db.session.commit() assert Enrollment.get_one(guid=enrollment.guid) == None