def test_generation_status(self, status, expected_response): """ Test handling of certificate statuses """ u = UserFactory() cr = CourseFactory() key = cr.id # pylint: disable=no-member cert = GeneratedCertificateFactory( user=u, course_id=key, mode=GeneratedCertificate.MODES.verified, status=status, ) self.assertEqual(_can_generate_allowlist_certificate_for_status(cert), expected_response)
def test_generation_status_for_none(self): """ Test handling of certificate statuses for a non-existent cert """ assert _can_generate_allowlist_certificate_for_status(None, None)