code review comments and lint

This commit is contained in:
Sharon Wang
2020-12-16 09:45:31 -05:00
parent 88b838f608
commit 8889180b59
2 changed files with 13 additions and 5 deletions

View File

@@ -16,6 +16,7 @@ from django.utils.http import http_date, parse_http_date
from edx_rest_framework_extensions.auth.jwt import cookies as jwt_cookies
from edx_rest_framework_extensions.auth.jwt.constants import JWT_DELIMITER
from oauth2_provider.models import Application
from common.djangoapps.student.models import UserProfile
from openedx.core.djangoapps.oauth_dispatch.adapters import DOTAdapter
from openedx.core.djangoapps.oauth_dispatch.api import create_dot_access_token
@@ -268,7 +269,7 @@ def _get_user_info_cookie_data(request, user):
'version': settings.EDXMKTG_USER_INFO_COOKIE_VERSION,
'username': user.username,
'header_urls': header_urls,
'image_urls': image_urls,
'user_image_urls': image_urls,
}
return user_info

View File

@@ -3,6 +3,7 @@
import json
import six
from datetime import date
from django.conf import settings
from django.http import HttpResponse
from django.test import RequestFactory, TestCase
@@ -10,7 +11,6 @@ from django.urls import reverse
from edx_rest_framework_extensions.auth.jwt.decoder import jwt_decode_handler
from edx_rest_framework_extensions.auth.jwt.middleware import JwtAuthCookieMiddleware
from mock import MagicMock, patch
from datetime import date
from openedx.core.djangoapps.user_api.accounts.utils import retrieve_last_sitewide_block_completed
from openedx.core.djangoapps.user_authn import cookies as cookies_api
@@ -36,6 +36,13 @@ class CookieTests(TestCase):
get_expiry_age=lambda: max_age,
)
def _convert_to_absolute_uris(self, request, urls_obj):
""" Convert relative URL paths to absolute URIs """
for url_name, url_path in six.iteritems(urls_obj):
urls_obj[url_name] = request.build_absolute_uri(url_path)
return urls_obj
def _get_expected_image_urls(self):
expected_image_urls = {
'full': '/static/default_500.png',
@@ -44,7 +51,7 @@ class CookieTests(TestCase):
'small': '/static/default_30.png'
}
expected_image_urls = cookies_api._convert_to_absolute_uris(self.request, expected_image_urls)
expected_image_urls = self._convert_to_absolute_uris(self.request, expected_image_urls)
return expected_image_urls
@@ -56,7 +63,7 @@ class CookieTests(TestCase):
'learner_profile': reverse('learner_profile', kwargs={'username': self.user.username}),
}
expected_header_urls = cookies_api._convert_to_absolute_uris(self.request, expected_header_urls)
expected_header_urls = self._convert_to_absolute_uris(self.request, expected_header_urls)
return expected_header_urls
@@ -111,7 +118,7 @@ class CookieTests(TestCase):
'version': settings.EDXMKTG_USER_INFO_COOKIE_VERSION,
'username': self.user.username,
'header_urls': self._get_expected_header_urls(),
'image_urls': self._get_expected_image_urls(),
'user_image_urls': self._get_expected_image_urls(),
}
self.assertDictEqual(actual, expected)