Merge pull request #4268 from edx/usman/lms6631-certificate-mode

Moved GeneratedCertificates.MODES back inside model.

lms/djangoapps/certificates/models.py

@@ -80,9 +80,10 @@ class CertificateWhitelist(models.Model):
     whitelist = models.BooleanField(default=0)
 
 
-MODES = Choices('verified', 'honor', 'audit')
-
 class GeneratedCertificate(models.Model):
+
+    MODES = Choices('verified', 'honor', 'audit')
+
     user = models.ForeignKey(User)
     course_id = CourseKeyField(max_length=255, blank=True, default=None)
     verify_uuid = models.CharField(max_length=32, blank=True, default='')

lms/djangoapps/certificates/tests/factories.py

@@ -2,7 +2,7 @@ from factory.django import DjangoModelFactory
 
 from opaque_keys.edx.locations import SlashSeparatedCourseKey
 
-from certificates.models import GeneratedCertificate, CertificateStatuses, MODES
+from certificates.models import GeneratedCertificate, CertificateStatuses
 
 # Factories don't have __init__ methods, and are self documenting
 # pylint: disable=W0232
@@ -12,5 +12,5 @@ class GeneratedCertificateFactory(DjangoModelFactory):
 
     course_id = None
     status = CertificateStatuses.unavailable
-    mode = MODES.honor
+    mode = GeneratedCertificate.MODES.honor
     name = ''

lms/djangoapps/certificates/tests/tests.py

@@ -0,0 +1,24 @@
+"""
+Tests for the certificates models.
+"""
+
+from django.test import TestCase
+
+from xmodule.modulestore.tests.factories import CourseFactory
+
+from student.tests.factories import UserFactory
+from certificates.models import CertificateStatuses, GeneratedCertificate, certificate_status_for_student
+
+
+class CertificatesModelTest(TestCase):
+    """
+    Tests for the GeneratedCertificate model
+    """
+
+    def test_certificate_status_for_student(self):
+        student = UserFactory()
+        course = CourseFactory.create(org='edx', number='verified', display_name='Verified Course')
+
+        certificate_status = certificate_status_for_student(student, course.id)
+        self.assertEqual(certificate_status['status'], CertificateStatuses.unavailable)
+        self.assertEqual(certificate_status['mode'], GeneratedCertificate.MODES.honor)