Commit 6d4f499b by Andy Armstrong Committed by GitHub

Merge pull request #15050 from edx/andya/fix-query-count-test

Fix broken Python certificate tests
parents 3d51bd08 480cf570
...@@ -302,6 +302,12 @@ class GeneratedCertificate(models.Model): ...@@ -302,6 +302,12 @@ class GeneratedCertificate(models.Model):
else: else:
return query.values('status').annotate(count=Count('status')) return query.values('status').annotate(count=Count('status'))
def __repr__(self):
return "<GeneratedCertificate: {course_id}, user={user}>".format(
course_id=self.course_id,
user=self.user
)
def invalidate(self): def invalidate(self):
""" """
Invalidate Generated Certificate by marking it 'unavailable'. Invalidate Generated Certificate by marking it 'unavailable'.
......
...@@ -369,6 +369,7 @@ class CertificateGetTests(SharedModuleStoreTestCase): ...@@ -369,6 +369,7 @@ class CertificateGetTests(SharedModuleStoreTestCase):
@classmethod @classmethod
def tearDownClass(cls): def tearDownClass(cls):
super(CertificateGetTests, cls).tearDownClass()
cls.freezer.stop() cls.freezer.stop()
def test_get_certificate_for_user(self): def test_get_certificate_for_user(self):
......
...@@ -721,7 +721,7 @@ class CertificatesViewsTests(CommonCertificatesTestCase): ...@@ -721,7 +721,7 @@ class CertificatesViewsTests(CommonCertificatesTestCase):
course_id=unicode(self.course.id) course_id=unicode(self.course.id)
) )
self.cert.delete() self.cert.delete()
self.assertEqual(len(GeneratedCertificate.eligible_certificates.all()), 0) self.assertListEqual(list(GeneratedCertificate.eligible_certificates.all()), [])
response = self.client.get(test_url) response = self.client.get(test_url)
self.assertIn('invalid', response.content) self.assertIn('invalid', response.content)
...@@ -744,7 +744,7 @@ class CertificatesViewsTests(CommonCertificatesTestCase): ...@@ -744,7 +744,7 @@ class CertificatesViewsTests(CommonCertificatesTestCase):
preview mode. Either the certificate is marked active or not. preview mode. Either the certificate is marked active or not.
""" """
self.cert.delete() self.cert.delete()
self.assertEqual(len(GeneratedCertificate.eligible_certificates.all()), 0) self.assertListEqual(list(GeneratedCertificate.eligible_certificates.all()), [])
self._add_course_certificates(count=1, signatory_count=2) self._add_course_certificates(count=1, signatory_count=2)
test_url = get_certificate_url( test_url = get_certificate_url(
user_id=self.user.id, user_id=self.user.id,
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment