diff --git a/common/djangoapps/student/tests/test_views.py b/common/djangoapps/student/tests/test_views.py index 51b029b73b..de7414bfb9 100644 --- a/common/djangoapps/student/tests/test_views.py +++ b/common/djangoapps/student/tests/test_views.py @@ -12,24 +12,28 @@ from django.core.urlresolvers import reverse from django.test import RequestFactory, TestCase from django.test.utils import override_settings from django.utils.timezone import now -from edx_oauth2_provider.constants import AUTHORIZED_CLIENTS_SESSION_KEY -from edx_oauth2_provider.tests.factories import ClientFactory, TrustedClientFactory -from milestones.tests.utils import MilestonesTestCaseMixin from mock import patch from opaque_keys import InvalidKeyError -from pyquery import PyQuery as pq from bulk_email.models import BulkEmailFlag +from course_modes.models import CourseMode +from edx_oauth2_provider.constants import AUTHORIZED_CLIENTS_SESSION_KEY +from edx_oauth2_provider.tests.factories import (ClientFactory, + TrustedClientFactory) from entitlements.tests.factories import CourseEntitlementFactory +from milestones.tests.utils import MilestonesTestCaseMixin from openedx.core.djangoapps.catalog.tests.factories import ProgramFactory from openedx.core.djangoapps.content.course_overviews.models import CourseOverview from openedx.core.djangoapps.content.course_overviews.tests.factories import CourseOverviewFactory +from pyquery import PyQuery as pq from student.cookies import get_user_info_cookie_data from student.helpers import DISABLE_UNENROLL_CERT_STATES from student.models import CourseEnrollment, UserProfile from student.signals import REFUND_ORDER from student.tests.factories import CourseEnrollmentFactory, UserFactory -from util.milestones_helpers import get_course_milestones, remove_prerequisite_course, set_prerequisite_courses +from util.milestones_helpers import (get_course_milestones, + remove_prerequisite_course, + set_prerequisite_courses) from util.testing import UrlResetMixin from xmodule.modulestore import ModuleStoreEnum from xmodule.modulestore.tests.django_utils import SharedModuleStoreTestCase @@ -350,7 +354,8 @@ class StudentDashboardTests(SharedModuleStoreTestCase, MilestonesTestCaseMixin): @patch('student.views.get_visible_sessions_for_entitlement') @patch('student.views.get_pseudo_session_for_entitlement') @patch.object(CourseOverview, 'get_from_id') - def test_unfulfilled_entitlement(self, mock_course_overview, mock_pseudo_session, mock_course_runs, mock_get_programs): + def test_unfulfilled_entitlement(self, mock_course_overview, mock_pseudo_session, + mock_course_runs, mock_get_programs): """ When a learner has an unfulfilled entitlement, their course dashboard should have: - a hidden 'View Course' button @@ -359,9 +364,9 @@ class StudentDashboardTests(SharedModuleStoreTestCase, MilestonesTestCaseMixin): - a related programs message """ program = ProgramFactory() - CourseEntitlementFactory(user=self.user, course_uuid=program['courses'][0]['uuid']) + CourseEntitlementFactory.create(user=self.user, course_uuid=program['courses'][0]['uuid']) mock_get_programs.return_value = [program] - mock_course_overview.return_value = CourseOverviewFactory(start=self.TOMORROW) + mock_course_overview.return_value = CourseOverviewFactory.create(start=self.TOMORROW) mock_course_runs.return_value = [ { 'key': 'course-v1:FAKE+FA1-MA1.X+3T2017', @@ -380,6 +385,29 @@ class StudentDashboardTests(SharedModuleStoreTestCase, MilestonesTestCaseMixin): self.assertIn('