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_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 create_mock_enrollment(**kwargs): people = kwargs.pop('people', [create_mock_user(), create_mock_user()]) enrollment = Enrollment.create_one( course=create_mock_course(), start_date=datetime.utcnow(), end_date=datetime.utcnow() + timedelta(days=60), status='open', ) for user in people: enrollment.people.append(user) db.session.commit() return enrollment
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_relationship__course(self): enrollment = Enrollment.create_one(**enrollment_data[0], course=self.courses[1]) db.session.commit() assert enrollment.course == self.courses[1]