class ThreadRequestMiddlewareTests(SimpleTestCase):
def setUp(self):
if hasattr(_local, 'username'):
delattr(_local, 'username')
if hasattr(_local, 'remote_addr'):
delattr(_local, 'remote_addr')
self.middleware = ThreadRequestMiddleware()
def test_get_remote_addr(self):
req = RequestFactory().get('/')
req.META['REMOTE_ADDR'] = '63.245.217.194'
eq_(get_remote_addr(), None)
self.middleware.process_request(req)
eq_(get_remote_addr(), '63.245.217.194')
def test_get_username_no_username_field(self):
req = RequestFactory().get('/')
req.user = mock.Mock()
del req.user.USERNAME_FIELD
req.user.username = '******'
eq_(get_username(), '<anon>')
self.middleware.process_request(req)
eq_(get_username(), 'my-username')
def test_get_username_with_username_field(self):
req = RequestFactory().get('/')
req.user = mock.Mock()
req.user.USERNAME_FIELD = 'myfield'
req.user.myfield = 'my-new-username'
eq_(get_username(), '<anon>')
self.middleware.process_request(req)
eq_(get_username(), 'my-new-username')
def setUp(self):
if hasattr(_local, 'username'):
delattr(_local, 'username')
if hasattr(_local, 'remote_addr'):
delattr(_local, 'remote_addr')
self.middleware = ThreadRequestMiddleware()
