def test_statuses(self):
active_enrollment = Enrollment(status=Enrollment.ACTIVE_STATUS)
self.assertEquals(active_enrollment.is_active(), True)
inactive_enrollment = Enrollment(status=Enrollment.INACTIVE_STATUS)
self.assertEquals(inactive_enrollment.is_active(), False)
deleted_enrollment = Enrollment(status=Enrollment.DELETED_STATUS)
self.assertEquals(deleted_enrollment.is_active(), False)
completed_enrollment = Enrollment(status=Enrollment.COMPLETED_STATUS)
self.assertEquals(completed_enrollment.is_active(), False)
def test_roles(self):
student_enrollment = Enrollment(role=Enrollment.STUDENT_ROLE)
self.assertEquals(student_enrollment.is_student(), True)
self.assertEquals(student_enrollment.is_instructor(), False)
self.assertEquals(student_enrollment.is_auditor(), False)
instructor_enrollment = Enrollment(role=Enrollment.INSTRUCTOR_ROLE)
self.assertEquals(instructor_enrollment.is_student(), False)
self.assertEquals(instructor_enrollment.is_instructor(), True)
self.assertEquals(instructor_enrollment.is_auditor(), False)
auditor_enrollment = Enrollment(role=Enrollment.AUDITOR_ROLE)
self.assertEquals(auditor_enrollment.is_student(), False)
self.assertEquals(auditor_enrollment.is_instructor(), False)
self.assertEquals(auditor_enrollment.is_auditor(), True)
