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]
