def test_get_cached_edx_data(self): """ Test for get_cached_edx_data """ with self.assertRaises(ValueError): CachedEdxDataApi.get_cached_edx_data(self.user, 'footype') self.assert_cache_in_db() for run in self.all_runs: CachedEnrollmentFactory.create(user=self.user, course_run=run) CachedCertificateFactory.create(user=self.user, course_run=run) CachedCurrentGradeFactory.create(user=self.user, course_run=run) self.assert_cache_in_db(self.all_course_run_ids, self.all_course_run_ids, self.all_course_run_ids)
def assert_cache_in_db(self, enrollment_keys=None, certificate_keys=None, grades_keys=None): """ Helper function to assert the course keys in the database cache """ enrollment_keys = enrollment_keys or [] certificate_keys = certificate_keys or [] grades_keys = grades_keys or [] enrollments = CachedEdxDataApi.get_cached_edx_data(self.user, CachedEdxDataApi.ENROLLMENT) certificates = CachedEdxDataApi.get_cached_edx_data(self.user, CachedEdxDataApi.CERTIFICATE) grades = CachedEdxDataApi.get_cached_edx_data(self.user, CachedEdxDataApi.CURRENT_GRADE) assert sorted(list(enrollments.enrollments.keys())) == sorted(enrollment_keys) assert sorted(list(certificates.certificates.keys())) == sorted(certificate_keys) assert sorted(list(grades.current_grades.keys())) == sorted(grades_keys)
def assert_cache_in_db(self, enrollment_keys=None, certificate_keys=None, grades_keys=None): """ Helper function to assert the course keys in the database cache """ enrollment_keys = enrollment_keys or [] certificate_keys = certificate_keys or [] grades_keys = grades_keys or [] enrollments = CachedEdxDataApi.get_cached_edx_data( self.user, CachedEdxDataApi.ENROLLMENT) certificates = CachedEdxDataApi.get_cached_edx_data( self.user, CachedEdxDataApi.CERTIFICATE) grades = CachedEdxDataApi.get_cached_edx_data( self.user, CachedEdxDataApi.CURRENT_GRADE) assert sorted(list( enrollments.enrollments.keys())) == sorted(enrollment_keys) assert sorted(list( certificates.certificates.keys())) == sorted(certificate_keys) assert sorted(list( grades.current_grades.keys())) == sorted(grades_keys)