def test_cookie_handler_retrieves_cookie_if_exists(self):
"""Test that a CookieHandler retrieves a cookie form the request if that
exists"""
mock_signer = MagicMock()
mock_signer.loads = lambda x: x
mock_request = MagicMock(cookies={'my_projectpswd': ['a_user']})
project = MagicMock(short_name='my_project')
cookie_handler = CookieHandler(request=mock_request, signer=mock_signer)
cookie = cookie_handler.get_cookie_from(project)
assert cookie == ['a_user'], cookie
def test_cookie_handler_empty_list_if_no_cookie(self):
"""Test that a CookieHandler retrieves an empty list if the cookie for
a project does not exist"""
mock_signer = MagicMock()
mock_signer.loads = lambda x: x
mock_request = MagicMock(cookies={})
project = MagicMock(short_name='my_project')
cookie_handler = CookieHandler(request=mock_request, signer=mock_signer)
cookie = cookie_handler.get_cookie_from(project)
assert cookie == [], cookie
def test_cookie_handler_empty_list_if_no_cookie(self):
"""Test that a CookieHandler retrieves an empty list if the cookie for
a project does not exist"""
mock_signer = MagicMock()
mock_signer.loads = lambda x: x
mock_request = MagicMock(cookies={})
project = MagicMock(short_name='my_project')
cookie_handler = CookieHandler(request=mock_request,
signer=mock_signer)
cookie = cookie_handler.get_cookie_from(project)
assert cookie == [], cookie
def test_cookie_handler_retrieves_cookie_if_exists(self):
"""Test that a CookieHandler retrieves a cookie form the request if that
exists"""
mock_signer = MagicMock()
mock_signer.loads = lambda x: x
mock_request = MagicMock(cookies={'my_projectpswd': ['a_user']})
project = MagicMock(short_name='my_project')
cookie_handler = CookieHandler(request=mock_request,
signer=mock_signer)
cookie = cookie_handler.get_cookie_from(project)
assert cookie == ['a_user'], cookie
def test_cookie_handler_updates_cookie(self):
"""Test that a CookieHandler updates the cookie with another user's info"""
mock_signer = MagicMock()
mock_signer.loads = lambda x: x
mock_signer.dumps = lambda x: x
mock_request = MagicMock(cookies={'my_projectpswd': ['first_user']})
mock_response = MagicMock()
project = MagicMock(short_name='my_project')
user = '******'
cookie_handler = CookieHandler(request=mock_request, signer=mock_signer)
cookie_handler.add_cookie_to(mock_response, project, user)
mock_response.set_cookie.assert_called_with('my_projectpswd',
['first_user', user],
max_age=1200)
def test_cookie_handler_adds_cookie_to_response(self):
"""Test that a CookieHandler object can successfully add a cookie to a
response object"""
mock_signer = MagicMock()
mock_signer.loads = lambda x: x
mock_signer.dumps = lambda x: x
mock_request = MagicMock(cookies={})
mock_response = MagicMock()
project = MagicMock(short_name='my_project')
user = '******'
cookie_handler = CookieHandler(request=mock_request, signer=mock_signer)
cookie_handler.add_cookie_to(mock_response, project, user)
mock_response.set_cookie.assert_called_with('my_projectpswd',
[user], max_age=1200)
def test_cookie_handler_updates_cookie(self):
"""Test that a CookieHandler updates the cookie with another user's info"""
mock_signer = MagicMock()
mock_signer.loads = lambda x: x
mock_signer.dumps = lambda x: x
mock_request = MagicMock(cookies={'my_projectpswd': ['first_user']})
mock_response = MagicMock()
project = MagicMock(short_name='my_project')
user = '******'
cookie_handler = CookieHandler(request=mock_request,
signer=mock_signer)
cookie_handler.add_cookie_to(mock_response, project, user)
mock_response.set_cookie.assert_called_with('my_projectpswd',
['first_user', user],
max_age=1200)
def test_cookie_handler_adds_cookie_to_response(self):
"""Test that a CookieHandler object can successfully add a cookie to a
response object"""
mock_signer = MagicMock()
mock_signer.loads = lambda x: x
mock_signer.dumps = lambda x: x
mock_request = MagicMock(cookies={})
mock_response = MagicMock()
project = MagicMock(short_name='my_project')
user = '******'
cookie_handler = CookieHandler(request=mock_request,
signer=mock_signer)
cookie_handler.add_cookie_to(mock_response, project, user)
mock_response.set_cookie.assert_called_with('my_projectpswd', [user],
max_age=1200)
def get_pwd_manager(project):
timeout = project.info.get('timeout')
cookie_timeout = max(timeout, ContributionsGuard.STAMP_TTL)
cookie_handler = CookieHandler(request, signer, cookie_timeout)
pwd_manager = ProjectPasswdManager(cookie_handler)
return pwd_manager
