def cleanup_current_request_and_theme(): # pylint: disable=invalid-name
"""
Remove current request and current theme.
"""
# Since, theme is saved on the request object,
# clearing request object will also clear the theme.
set_current_request(request=None)
def setUp(self):
"""
Save thread locals reference for use in the tests.
"""
self.thread_locals = thread_locals.__thread_locals__
self.request = RequestFactory().get('/')
set_current_request(self.request)
def test_get_current_request(self):
"""
Verify get_current_request works as expected.
"""
set_current_request(self.request)
self.assertEqual(
get_current_request(),
self.request,
)
def setup_current_theme(theme, request=None):
"""
Save current request and theme in thread locals to simulate an ongoing Http Request.
Arguments:
theme (Theme): Theme instance to set.
request (HttpRequest): Django request object for the current request.
"""
request = request or get_current_request() or RequestFactory().get('/')
set_current_request(request)
set_current_theme(theme)
def test_set_current_request(self):
"""
Verify set_current_request works as expected.
"""
set_current_request(self.request)
self.assert_thread_variable('request', self.request)
def process_request(self, request): # pylint: disable=no-self-use
"""
Save request object in local thread data.
"""
set_current_request(request)