From cff8e16de380e73ccae6b003f7cd78cb35809972 Mon Sep 17 00:00:00 2001 From: Greg Price Date: Fri, 3 Oct 2014 14:40:37 -0400 Subject: [PATCH] fixup! Make notifier API resilient to bad enrollments Add explicit course key to enrollment for non-existent course in test case --- lms/djangoapps/notifier_api/tests.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/lms/djangoapps/notifier_api/tests.py b/lms/djangoapps/notifier_api/tests.py index 9bdf8990ae..69733ae7ad 100644 --- a/lms/djangoapps/notifier_api/tests.py +++ b/lms/djangoapps/notifier_api/tests.py @@ -10,6 +10,7 @@ from django_comment_common.models import Role, Permission from lang_pref import LANGUAGE_KEY from notification_prefs import NOTIFICATION_PREF_KEY from notifier_api.views import NotifierUsersViewSet +from opaque_keys.edx.locator import CourseLocator from student.models import CourseEnrollment from student.tests.factories import UserFactory, CourseEnrollmentFactory from user_api.models import UserPreference @@ -121,7 +122,10 @@ class NotifierUsersViewSetTest(UrlResetMixin, ModuleStoreTestCase): self.assertEqual(result["course_info"], {}) def test_course_info_non_existent_course_enrollment(self): - CourseEnrollmentFactory(user=self.user) + CourseEnrollmentFactory( + user=self.user, + course_id=CourseLocator(org="dummy", course="dummy", run="non_existent") + ) result = self._get_detail() self.assertEqual(result["course_info"], {})