Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- diff --git a/tests/test_views.py b/tests/test_views.py
- index d4db87f..1216517 100644
- --- a/tests/test_views.py
- +++ b/tests/test_views.py
- @@ -976,6 +976,12 @@ class TestCourseEnrollmentView(TestCase):
- """
- assert self.client.login(username=self.user.username, password="QWERTY")
- + def _setup_course_catalog(self, course_catalog_client_mock):
- + catalog_instance = course_catalog_client_mock.return_value
- + catalog_instance.get_course_run.return_value = {
- + 'level_type': 'Type 1',
- + }
- +
- @mock.patch('enterprise.views.render', side_effect=fake_render)
- @mock.patch('enterprise.views.configuration_helpers')
- @mock.patch('enterprise.tpa_pipeline.get_enterprise_customer_for_request')
- @@ -999,10 +1005,7 @@ class TestCourseEnrollmentView(TestCase):
- configuration_helpers_mock,
- render_to_response_mock, # pylint: disable=unused-argument
- ):
- - catalog_instance = course_catalog_client_mock.return_value
- - catalog_instance.get_course_run.return_value = {
- - 'level_type': 'Type 1',
- - }
- + self._setup_course_catalog(course_catalog_client_mock)
- logo_mock = mock.MagicMock()
- logo_mock.url = 'logo.png'
- organizations_helpers_mock.get_organization.return_value = {
- @@ -1097,10 +1100,7 @@ class TestCourseEnrollmentView(TestCase):
- configuration_helpers_mock,
- render_to_response_mock, # pylint: disable=unused-argument
- ):
- - catalog_instance = course_catalog_client_mock.return_value
- - catalog_instance.get_course_run.return_value = {
- - 'level_type': 'Type 1',
- - }
- + self._setup_course_catalog(course_catalog_client_mock)
- organizations_helpers_mock.get_organization.return_value = mock.MagicMock()
- course_id = self.demo_course_id
- configuration_helpers_mock.get_value.return_value = 'edX'
- @@ -1193,10 +1193,7 @@ class TestCourseEnrollmentView(TestCase):
- Verify that the context of the enterprise course enrollment page has
- empty course start date if course details has no start date.
- """
- - catalog_instance = course_catalog_client_mock.return_value
- - catalog_instance.get_course_run.return_value = {
- - 'level_type': 'Type 1',
- - }
- + self._setup_course_catalog(course_catalog_client_mock)
- organizations_helpers_mock.get_organization.return_value = mock.MagicMock()
- course_id = self.demo_course_id
- configuration_helpers_mock.get_value.return_value = 'edX'
- @@ -1543,10 +1540,7 @@ class TestCourseEnrollmentView(TestCase):
- configuration_helpers_mock,
- render_to_response_mock, # pylint: disable=unused-argument
- ):
- - catalog_instance = course_catalog_client_mock.return_value
- - catalog_instance.get_course_run.return_value = {
- - 'level_type': 'Type 1',
- - }
- + self._setup_course_catalog(course_catalog_client_mock)
- organizations_helpers_mock.get_organization.return_value = mock.MagicMock()
- course_id = self.demo_course_id
- configuration_helpers_mock.get_value.return_value = 'edX'
- @@ -1684,10 +1678,7 @@ class TestCourseEnrollmentView(TestCase):
- _unused_render_to_response_mock,
- ):
- # Set up course catalog API client
- - catalog_instance = course_catalog_client_mock.return_value
- - catalog_instance.get_course_run.return_value = {
- - 'level_type': 'Type 1',
- - }
- + self._setup_course_catalog(course_catalog_client_mock)
- # Set up organizations API client
- logo_mock = mock.MagicMock()
- @@ -1797,10 +1788,7 @@ class TestCourseEnrollmentView(TestCase):
- _unused_render_to_response_mock,
- ):
- # Set up course catalog API client
- - catalog_instance = course_catalog_client_mock.return_value
- - catalog_instance.get_course_run.return_value = {
- - 'level_type': 'Type 1',
- - }
- + self._setup_course_catalog(course_catalog_client_mock)
- # Set up organizations API client
- organizations_helpers_mock.get_organization.return_value = None
- @@ -1905,10 +1893,7 @@ class TestCourseEnrollmentView(TestCase):
- _unused_render_to_response_mock,
- ):
- # Set up course catalog API client
- - catalog_instance = course_catalog_client_mock.return_value
- - catalog_instance.get_course_run.return_value = {
- - 'level_type': 'Type 1',
- - }
- + self._setup_course_catalog(course_catalog_client_mock)
- # Set up organizations API client
- organizations_helpers_mock.get_organization.side_effect = ValueError
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement