def test_set_enterprise_customer_cookie(self):
"""
Verify that enterprise cookies are set properly.
"""
enterprise_customer_uuid = uuid.uuid4()
response = HttpResponse()
result = set_enterprise_customer_cookie(self.site, response, enterprise_customer_uuid)
cookie = result.cookies[settings.ENTERPRISE_CUSTOMER_COOKIE_NAME]
self.assertEqual(str(enterprise_customer_uuid), cookie.value)
def _decorated(request, *args, **kwargs):
code, enterprise_customer_uuid = request.GET.get('code'), None
if code:
enterprise_customer_uuid = utils.get_enterprise_customer_uuid(code)
response = func(request, *args, **kwargs)
# Set enterprise customer cookie if enterprise customer uuid is available.
if enterprise_customer_uuid:
response = utils.set_enterprise_customer_cookie(request.site, response, enterprise_customer_uuid)
return response
def test_set_enterprise_customer_cookie_empty_cookie_domain(self):
"""
Verify that enterprise cookie is not set if base_cookie_domain is empty
in site configuration.
"""
self.site.siteconfiguration.base_cookie_domain = ''
self.site.siteconfiguration.save()
enterprise_customer_uuid = uuid.uuid4()
response = HttpResponse()
result = set_enterprise_customer_cookie(self.site, response, enterprise_customer_uuid)
self.assertNotIn(settings.ENTERPRISE_CUSTOMER_COOKIE_NAME, result.cookies)
def _decorated(request, *args, **kwargs):
response = func(request, *args, **kwargs)
# Set enterprise customer cookie if enterprise customer uuid is available.
# The max_age of the cookie is set to a relatively short 60 seconds to
# maintain the Enterprise context across jumps over to other Open edX
# services, e.g. LMS, but also to ensure that the Enterprise context
# does not linger longer than it should.
code, enterprise_customer_uuid = request.GET.get('code'), None
if code:
enterprise_customer_uuid = utils.get_enterprise_customer_uuid(code)
if enterprise_customer_uuid:
response = utils.set_enterprise_customer_cookie(
request.site,
response,
enterprise_customer_uuid,
max_age=60)
return response
