summaryrefslogtreecommitdiff
path: root/tests
diff options
context:
space:
mode:
authorTheron Luhn <theron@luhn.com>2019-04-15 19:05:47 -0700
committerTheron Luhn <theron@luhn.com>2019-04-15 19:05:47 -0700
commitf4c6c993ded900b4e32d1dd49207ca1e18b11336 (patch)
tree6c8bc3f9148699408fcfddcd52d95e51be662faf /tests
parent47f8935bb5423718ec293bba4709307be7d2f51c (diff)
downloadpyramid-f4c6c993ded900b4e32d1dd49207ca1e18b11336.tar.gz
pyramid-f4c6c993ded900b4e32d1dd49207ca1e18b11336.tar.bz2
pyramid-f4c6c993ded900b4e32d1dd49207ca1e18b11336.zip
Revert "Migrate AuthTktCookieHelper to pyramid.security."
This reverts commit 9f267dd842c5e93336f0392f2809da75a716039a.
Diffstat (limited to 'tests')
-rw-r--r--tests/test_authentication.py1000
-rw-r--r--tests/test_security.py1008
2 files changed, 999 insertions, 1009 deletions
diff --git a/tests/test_authentication.py b/tests/test_authentication.py
index 89cf9866d..8671eba05 100644
--- a/tests/test_authentication.py
+++ b/tests/test_authentication.py
@@ -1,7 +1,8 @@
+from http.cookies import SimpleCookie
import unittest
import warnings
from pyramid import testing
-from pyramid.util import bytes_
+from pyramid.util import text_, bytes_
class TestCallbackAuthenticationPolicyDebugging(unittest.TestCase):
@@ -663,6 +664,926 @@ class TestAuthTktAuthenticationPolicy(unittest.TestCase):
verifyObject(IAuthenticationPolicy, self._makeOne(None, None))
+class TestAuthTktCookieHelper(unittest.TestCase):
+ def _getTargetClass(self):
+ from pyramid.authentication import AuthTktCookieHelper
+
+ return AuthTktCookieHelper
+
+ def _makeOne(self, *arg, **kw):
+ helper = self._getTargetClass()(*arg, **kw)
+ # laziness after moving auth_tkt classes and funcs into
+ # authentication module
+ auth_tkt = DummyAuthTktModule()
+ helper.auth_tkt = auth_tkt
+ helper.AuthTicket = auth_tkt.AuthTicket
+ helper.parse_ticket = auth_tkt.parse_ticket
+ helper.BadTicket = auth_tkt.BadTicket
+ return helper
+
+ def _makeRequest(self, cookie=None, ipv6=False):
+ environ = {'wsgi.version': (1, 0)}
+
+ if ipv6 is False:
+ environ['REMOTE_ADDR'] = '1.1.1.1'
+ else:
+ environ['REMOTE_ADDR'] = '::1'
+ environ['SERVER_NAME'] = 'localhost'
+ return DummyRequest(environ, cookie=cookie)
+
+ def _cookieValue(self, cookie):
+ items = cookie.value.split('/')
+ D = {}
+ for item in items:
+ k, v = item.split('=', 1)
+ D[k] = v
+ return D
+
+ def _parseHeaders(self, headers):
+ return [self._parseHeader(header) for header in headers]
+
+ def _parseHeader(self, header):
+ cookie = self._parseCookie(header[1])
+ return cookie
+
+ def _parseCookie(self, cookie):
+ cookies = SimpleCookie()
+ cookies.load(cookie)
+ return cookies.get('auth_tkt')
+
+ def test_init_cookie_str_reissue_invalid(self):
+ self.assertRaises(
+ ValueError, self._makeOne, 'secret', reissue_time='invalid value'
+ )
+
+ def test_init_cookie_str_timeout_invalid(self):
+ self.assertRaises(
+ ValueError, self._makeOne, 'secret', timeout='invalid value'
+ )
+
+ def test_init_cookie_str_max_age_invalid(self):
+ self.assertRaises(
+ ValueError, self._makeOne, 'secret', max_age='invalid value'
+ )
+
+ def test_identify_nocookie(self):
+ helper = self._makeOne('secret')
+ request = self._makeRequest()
+ result = helper.identify(request)
+ self.assertEqual(result, None)
+
+ def test_identify_cookie_value_is_None(self):
+ helper = self._makeOne('secret')
+ request = self._makeRequest(None)
+ result = helper.identify(request)
+ self.assertEqual(result, None)
+
+ def test_identify_good_cookie_include_ip(self):
+ helper = self._makeOne('secret', include_ip=True)
+ request = self._makeRequest('ticket')
+ result = helper.identify(request)
+ self.assertEqual(len(result), 4)
+ self.assertEqual(result['tokens'], ())
+ self.assertEqual(result['userid'], 'userid')
+ self.assertEqual(result['userdata'], '')
+ self.assertEqual(result['timestamp'], 0)
+ self.assertEqual(helper.auth_tkt.value, 'ticket')
+ self.assertEqual(helper.auth_tkt.remote_addr, '1.1.1.1')
+ self.assertEqual(helper.auth_tkt.secret, 'secret')
+ environ = request.environ
+ self.assertEqual(environ['REMOTE_USER_TOKENS'], ())
+ self.assertEqual(environ['REMOTE_USER_DATA'], '')
+ self.assertEqual(environ['AUTH_TYPE'], 'cookie')
+
+ def test_identify_good_cookie_include_ipv6(self):
+ helper = self._makeOne('secret', include_ip=True)
+ request = self._makeRequest('ticket', ipv6=True)
+ result = helper.identify(request)
+ self.assertEqual(len(result), 4)
+ self.assertEqual(result['tokens'], ())
+ self.assertEqual(result['userid'], 'userid')
+ self.assertEqual(result['userdata'], '')
+ self.assertEqual(result['timestamp'], 0)
+ self.assertEqual(helper.auth_tkt.value, 'ticket')
+ self.assertEqual(helper.auth_tkt.remote_addr, '::1')
+ self.assertEqual(helper.auth_tkt.secret, 'secret')
+ environ = request.environ
+ self.assertEqual(environ['REMOTE_USER_TOKENS'], ())
+ self.assertEqual(environ['REMOTE_USER_DATA'], '')
+ self.assertEqual(environ['AUTH_TYPE'], 'cookie')
+
+ def test_identify_good_cookie_dont_include_ip(self):
+ helper = self._makeOne('secret', include_ip=False)
+ request = self._makeRequest('ticket')
+ result = helper.identify(request)
+ self.assertEqual(len(result), 4)
+ self.assertEqual(result['tokens'], ())
+ self.assertEqual(result['userid'], 'userid')
+ self.assertEqual(result['userdata'], '')
+ self.assertEqual(result['timestamp'], 0)
+ self.assertEqual(helper.auth_tkt.value, 'ticket')
+ self.assertEqual(helper.auth_tkt.remote_addr, '0.0.0.0')
+ self.assertEqual(helper.auth_tkt.secret, 'secret')
+ environ = request.environ
+ self.assertEqual(environ['REMOTE_USER_TOKENS'], ())
+ self.assertEqual(environ['REMOTE_USER_DATA'], '')
+ self.assertEqual(environ['AUTH_TYPE'], 'cookie')
+
+ def test_identify_good_cookie_int_useridtype(self):
+ helper = self._makeOne('secret', include_ip=False)
+ helper.auth_tkt.userid = '1'
+ helper.auth_tkt.user_data = 'userid_type:int'
+ request = self._makeRequest('ticket')
+ result = helper.identify(request)
+ self.assertEqual(len(result), 4)
+ self.assertEqual(result['tokens'], ())
+ self.assertEqual(result['userid'], 1)
+ self.assertEqual(result['userdata'], 'userid_type:int')
+ self.assertEqual(result['timestamp'], 0)
+ environ = request.environ
+ self.assertEqual(environ['REMOTE_USER_TOKENS'], ())
+ self.assertEqual(environ['REMOTE_USER_DATA'], 'userid_type:int')
+ self.assertEqual(environ['AUTH_TYPE'], 'cookie')
+
+ def test_identify_nonuseridtype_user_data(self):
+ helper = self._makeOne('secret', include_ip=False)
+ helper.auth_tkt.userid = '1'
+ helper.auth_tkt.user_data = 'bogus:int'
+ request = self._makeRequest('ticket')
+ result = helper.identify(request)
+ self.assertEqual(len(result), 4)
+ self.assertEqual(result['tokens'], ())
+ self.assertEqual(result['userid'], '1')
+ self.assertEqual(result['userdata'], 'bogus:int')
+ self.assertEqual(result['timestamp'], 0)
+ environ = request.environ
+ self.assertEqual(environ['REMOTE_USER_TOKENS'], ())
+ self.assertEqual(environ['REMOTE_USER_DATA'], 'bogus:int')
+ self.assertEqual(environ['AUTH_TYPE'], 'cookie')
+
+ def test_identify_good_cookie_unknown_useridtype(self):
+ helper = self._makeOne('secret', include_ip=False)
+ helper.auth_tkt.userid = 'abc'
+ helper.auth_tkt.user_data = 'userid_type:unknown'
+ request = self._makeRequest('ticket')
+ result = helper.identify(request)
+ self.assertEqual(len(result), 4)
+ self.assertEqual(result['tokens'], ())
+ self.assertEqual(result['userid'], 'abc')
+ self.assertEqual(result['userdata'], 'userid_type:unknown')
+ self.assertEqual(result['timestamp'], 0)
+ environ = request.environ
+ self.assertEqual(environ['REMOTE_USER_TOKENS'], ())
+ self.assertEqual(environ['REMOTE_USER_DATA'], 'userid_type:unknown')
+ self.assertEqual(environ['AUTH_TYPE'], 'cookie')
+
+ def test_identify_good_cookie_b64str_useridtype(self):
+ from base64 import b64encode
+
+ helper = self._makeOne('secret', include_ip=False)
+ helper.auth_tkt.userid = b64encode(b'encoded').strip()
+ helper.auth_tkt.user_data = 'userid_type:b64str'
+ request = self._makeRequest('ticket')
+ result = helper.identify(request)
+ self.assertEqual(len(result), 4)
+ self.assertEqual(result['tokens'], ())
+ self.assertEqual(result['userid'], b'encoded')
+ self.assertEqual(result['userdata'], 'userid_type:b64str')
+ self.assertEqual(result['timestamp'], 0)
+ environ = request.environ
+ self.assertEqual(environ['REMOTE_USER_TOKENS'], ())
+ self.assertEqual(environ['REMOTE_USER_DATA'], 'userid_type:b64str')
+ self.assertEqual(environ['AUTH_TYPE'], 'cookie')
+
+ def test_identify_good_cookie_b64unicode_useridtype(self):
+ from base64 import b64encode
+
+ helper = self._makeOne('secret', include_ip=False)
+ helper.auth_tkt.userid = b64encode(b'\xc3\xa9ncoded').strip()
+ helper.auth_tkt.user_data = 'userid_type:b64unicode'
+ request = self._makeRequest('ticket')
+ result = helper.identify(request)
+ self.assertEqual(len(result), 4)
+ self.assertEqual(result['tokens'], ())
+ self.assertEqual(result['userid'], text_(b'\xc3\xa9ncoded', 'utf-8'))
+ self.assertEqual(result['userdata'], 'userid_type:b64unicode')
+ self.assertEqual(result['timestamp'], 0)
+ environ = request.environ
+ self.assertEqual(environ['REMOTE_USER_TOKENS'], ())
+ self.assertEqual(environ['REMOTE_USER_DATA'], 'userid_type:b64unicode')
+ self.assertEqual(environ['AUTH_TYPE'], 'cookie')
+
+ def test_identify_bad_cookie(self):
+ helper = self._makeOne('secret', include_ip=True)
+ helper.auth_tkt.parse_raise = True
+ request = self._makeRequest('ticket')
+ result = helper.identify(request)
+ self.assertEqual(result, None)
+
+ def test_identify_cookie_timeout(self):
+ helper = self._makeOne('secret', timeout=1)
+ self.assertEqual(helper.timeout, 1)
+
+ def test_identify_cookie_str_timeout(self):
+ helper = self._makeOne('secret', timeout='1')
+ self.assertEqual(helper.timeout, 1)
+
+ def test_identify_cookie_timeout_aged(self):
+ import time
+
+ helper = self._makeOne('secret', timeout=10)
+ now = time.time()
+ helper.auth_tkt.timestamp = now - 1
+ helper.now = now + 10
+ helper.auth_tkt.tokens = (text_('a'),)
+ request = self._makeRequest('bogus')
+ result = helper.identify(request)
+ self.assertFalse(result)
+
+ def test_identify_cookie_reissue(self):
+ import time
+
+ helper = self._makeOne('secret', timeout=10, reissue_time=0)
+ now = time.time()
+ helper.auth_tkt.timestamp = now
+ helper.now = now + 1
+ helper.auth_tkt.tokens = (text_('a'),)
+ request = self._makeRequest('bogus')
+ result = helper.identify(request)
+ self.assertTrue(result)
+ self.assertEqual(len(request.callbacks), 1)
+ response = DummyResponse()
+ request.callbacks[0](request, response)
+ self.assertEqual(len(response.headerlist), 3)
+ self.assertEqual(response.headerlist[0][0], 'Set-Cookie')
+
+ def test_identify_cookie_str_reissue(self):
+ import time
+
+ helper = self._makeOne('secret', timeout=10, reissue_time='0')
+ now = time.time()
+ helper.auth_tkt.timestamp = now
+ helper.now = now + 1
+ helper.auth_tkt.tokens = (text_('a'),)
+ request = self._makeRequest('bogus')
+ result = helper.identify(request)
+ self.assertTrue(result)
+ self.assertEqual(len(request.callbacks), 1)
+ response = DummyResponse()
+ request.callbacks[0](request, response)
+ self.assertEqual(len(response.headerlist), 3)
+ self.assertEqual(response.headerlist[0][0], 'Set-Cookie')
+
+ def test_identify_cookie_reissue_already_reissued_this_request(self):
+ import time
+
+ helper = self._makeOne('secret', timeout=10, reissue_time=0)
+ now = time.time()
+ helper.auth_tkt.timestamp = now
+ helper.now = now + 1
+ request = self._makeRequest('bogus')
+ request._authtkt_reissued = True
+ result = helper.identify(request)
+ self.assertTrue(result)
+ self.assertEqual(len(request.callbacks), 0)
+
+ def test_identify_cookie_reissue_notyet(self):
+ import time
+
+ helper = self._makeOne('secret', timeout=10, reissue_time=10)
+ now = time.time()
+ helper.auth_tkt.timestamp = now
+ helper.now = now + 1
+ request = self._makeRequest('bogus')
+ result = helper.identify(request)
+ self.assertTrue(result)
+ self.assertEqual(len(request.callbacks), 0)
+
+ def test_identify_cookie_reissue_revoked_by_forget(self):
+ import time
+
+ helper = self._makeOne('secret', timeout=10, reissue_time=0)
+ now = time.time()
+ helper.auth_tkt.timestamp = now
+ helper.now = now + 1
+ request = self._makeRequest('bogus')
+ result = helper.identify(request)
+ self.assertTrue(result)
+ self.assertEqual(len(request.callbacks), 1)
+ result = helper.forget(request)
+ self.assertTrue(result)
+ self.assertEqual(len(request.callbacks), 1)
+ response = DummyResponse()
+ request.callbacks[0](request, response)
+ self.assertEqual(len(response.headerlist), 0)
+
+ def test_identify_cookie_reissue_revoked_by_remember(self):
+ import time
+
+ helper = self._makeOne('secret', timeout=10, reissue_time=0)
+ now = time.time()
+ helper.auth_tkt.timestamp = now
+ helper.now = now + 1
+ request = self._makeRequest('bogus')
+ result = helper.identify(request)
+ self.assertTrue(result)
+ self.assertEqual(len(request.callbacks), 1)
+ result = helper.remember(request, 'bob')
+ self.assertTrue(result)
+ self.assertEqual(len(request.callbacks), 1)
+ response = DummyResponse()
+ request.callbacks[0](request, response)
+ self.assertEqual(len(response.headerlist), 0)
+
+ def test_identify_cookie_reissue_with_tokens_default(self):
+ # see https://github.com/Pylons/pyramid/issues#issue/108
+ import time
+
+ helper = self._makeOne('secret', timeout=10, reissue_time=0)
+ auth_tkt = DummyAuthTktModule(tokens=[''])
+ helper.auth_tkt = auth_tkt
+ helper.AuthTicket = auth_tkt.AuthTicket
+ helper.parse_ticket = auth_tkt.parse_ticket
+ helper.BadTicket = auth_tkt.BadTicket
+ now = time.time()
+ helper.auth_tkt.timestamp = now
+ helper.now = now + 1
+ request = self._makeRequest('bogus')
+ result = helper.identify(request)
+ self.assertTrue(result)
+ self.assertEqual(len(request.callbacks), 1)
+ response = DummyResponse()
+ request.callbacks[0](None, response)
+ self.assertEqual(len(response.headerlist), 3)
+ self.assertEqual(response.headerlist[0][0], 'Set-Cookie')
+ self.assertTrue("/tokens=/" in response.headerlist[0][1])
+
+ def test_remember(self):
+ helper = self._makeOne('secret')
+ request = self._makeRequest()
+ result = helper.remember(request, 'userid')
+ self.assertEqual(len(result), 3)
+
+ self.assertEqual(result[0][0], 'Set-Cookie')
+ self.assertTrue(result[0][1].endswith('; Path=/; SameSite=Lax'))
+ self.assertTrue(result[0][1].startswith('auth_tkt='))
+
+ self.assertEqual(result[1][0], 'Set-Cookie')
+ self.assertTrue(
+ result[1][1].endswith('; Domain=localhost; Path=/; SameSite=Lax')
+ )
+ self.assertTrue(result[1][1].startswith('auth_tkt='))
+
+ self.assertEqual(result[2][0], 'Set-Cookie')
+ self.assertTrue(
+ result[2][1].endswith('; Domain=.localhost; Path=/; SameSite=Lax')
+ )
+ self.assertTrue(result[2][1].startswith('auth_tkt='))
+
+ def test_remember_nondefault_samesite(self):
+ helper = self._makeOne('secret', samesite='Strict')
+ request = self._makeRequest()
+ result = helper.remember(request, 'userid')
+ self.assertEqual(len(result), 3)
+
+ self.assertEqual(result[0][0], 'Set-Cookie')
+ self.assertTrue(result[0][1].endswith('; Path=/; SameSite=Strict'))
+ self.assertTrue(result[0][1].startswith('auth_tkt='))
+
+ self.assertEqual(result[1][0], 'Set-Cookie')
+ self.assertTrue(
+ result[1][1].endswith(
+ '; Domain=localhost; Path=/; SameSite=Strict'
+ )
+ )
+ self.assertTrue(result[1][1].startswith('auth_tkt='))
+
+ self.assertEqual(result[2][0], 'Set-Cookie')
+ self.assertTrue(
+ result[2][1].endswith(
+ '; Domain=.localhost; Path=/; SameSite=Strict'
+ )
+ )
+ self.assertTrue(result[2][1].startswith('auth_tkt='))
+
+ def test_remember_None_samesite(self):
+ helper = self._makeOne('secret', samesite=None)
+ request = self._makeRequest()
+ result = helper.remember(request, 'userid')
+ self.assertEqual(len(result), 3)
+
+ self.assertEqual(result[0][0], 'Set-Cookie')
+ self.assertTrue(result[0][1].endswith('; Path=/')) # no samesite
+ self.assertTrue(result[0][1].startswith('auth_tkt='))
+
+ self.assertEqual(result[1][0], 'Set-Cookie')
+ self.assertTrue(result[1][1].endswith('; Domain=localhost; Path=/'))
+ self.assertTrue(result[1][1].startswith('auth_tkt='))
+
+ self.assertEqual(result[2][0], 'Set-Cookie')
+ self.assertTrue(result[2][1].endswith('; Domain=.localhost; Path=/'))
+ self.assertTrue(result[2][1].startswith('auth_tkt='))
+
+ def test_remember_include_ip(self):
+ helper = self._makeOne('secret', include_ip=True)
+ request = self._makeRequest()
+ result = helper.remember(request, 'other')
+ self.assertEqual(len(result), 3)
+
+ self.assertEqual(result[0][0], 'Set-Cookie')
+ self.assertTrue(result[0][1].endswith('; Path=/; SameSite=Lax'))
+ self.assertTrue(result[0][1].startswith('auth_tkt='))
+
+ self.assertEqual(result[1][0], 'Set-Cookie')
+ self.assertTrue(
+ result[1][1].endswith('; Domain=localhost; Path=/; SameSite=Lax')
+ )
+ self.assertTrue(result[1][1].startswith('auth_tkt='))
+
+ self.assertEqual(result[2][0], 'Set-Cookie')
+ self.assertTrue(
+ result[2][1].endswith('; Domain=.localhost; Path=/; SameSite=Lax')
+ )
+ self.assertTrue(result[2][1].startswith('auth_tkt='))
+
+ def test_remember_path(self):
+ helper = self._makeOne(
+ 'secret', include_ip=True, path="/cgi-bin/app.cgi/"
+ )
+ request = self._makeRequest()
+ result = helper.remember(request, 'other')
+ self.assertEqual(len(result), 3)
+
+ self.assertEqual(result[0][0], 'Set-Cookie')
+ self.assertTrue(
+ result[0][1].endswith('; Path=/cgi-bin/app.cgi/; SameSite=Lax')
+ )
+ self.assertTrue(result[0][1].startswith('auth_tkt='))
+
+ self.assertEqual(result[1][0], 'Set-Cookie')
+ self.assertTrue(
+ result[1][1].endswith(
+ '; Domain=localhost; Path=/cgi-bin/app.cgi/; SameSite=Lax'
+ )
+ )
+ self.assertTrue(result[1][1].startswith('auth_tkt='))
+
+ self.assertEqual(result[2][0], 'Set-Cookie')
+ self.assertTrue(
+ result[2][1].endswith(
+ '; Domain=.localhost; Path=/cgi-bin/app.cgi/; SameSite=Lax'
+ )
+ )
+ self.assertTrue(result[2][1].startswith('auth_tkt='))
+
+ def test_remember_http_only(self):
+ helper = self._makeOne('secret', include_ip=True, http_only=True)
+ request = self._makeRequest()
+ result = helper.remember(request, 'other')
+ self.assertEqual(len(result), 3)
+
+ self.assertEqual(result[0][0], 'Set-Cookie')
+ self.assertTrue(result[0][1].endswith('; HttpOnly; SameSite=Lax'))
+ self.assertTrue(result[0][1].startswith('auth_tkt='))
+
+ self.assertEqual(result[1][0], 'Set-Cookie')
+ self.assertTrue('; HttpOnly' in result[1][1])
+ self.assertTrue(result[1][1].startswith('auth_tkt='))
+
+ self.assertEqual(result[2][0], 'Set-Cookie')
+ self.assertTrue('; HttpOnly' in result[2][1])
+ self.assertTrue(result[2][1].startswith('auth_tkt='))
+
+ def test_remember_secure(self):
+ helper = self._makeOne('secret', include_ip=True, secure=True)
+ request = self._makeRequest()
+ result = helper.remember(request, 'other')
+ self.assertEqual(len(result), 3)
+
+ self.assertEqual(result[0][0], 'Set-Cookie')
+ self.assertTrue('; secure' in result[0][1])
+ self.assertTrue(result[0][1].startswith('auth_tkt='))
+
+ self.assertEqual(result[1][0], 'Set-Cookie')
+ self.assertTrue('; secure' in result[1][1])
+ self.assertTrue(result[1][1].startswith('auth_tkt='))
+
+ self.assertEqual(result[2][0], 'Set-Cookie')
+ self.assertTrue('; secure' in result[2][1])
+ self.assertTrue(result[2][1].startswith('auth_tkt='))
+
+ def test_remember_wild_domain_disabled(self):
+ helper = self._makeOne('secret', wild_domain=False)
+ request = self._makeRequest()
+ result = helper.remember(request, 'other')
+ self.assertEqual(len(result), 2)
+
+ self.assertEqual(result[0][0], 'Set-Cookie')
+ self.assertTrue(result[0][1].endswith('; Path=/; SameSite=Lax'))
+ self.assertTrue(result[0][1].startswith('auth_tkt='))
+
+ self.assertEqual(result[1][0], 'Set-Cookie')
+ self.assertTrue(
+ result[1][1].endswith('; Domain=localhost; Path=/; SameSite=Lax')
+ )
+ self.assertTrue(result[1][1].startswith('auth_tkt='))
+
+ def test_remember_parent_domain(self):
+ helper = self._makeOne('secret', parent_domain=True)
+ request = self._makeRequest()
+ request.domain = 'www.example.com'
+ result = helper.remember(request, 'other')
+ self.assertEqual(len(result), 1)
+
+ self.assertEqual(result[0][0], 'Set-Cookie')
+ self.assertTrue(
+ result[0][1].endswith(
+ '; Domain=.example.com; Path=/; SameSite=Lax'
+ )
+ )
+ self.assertTrue(result[0][1].startswith('auth_tkt='))
+
+ def test_remember_parent_domain_supercedes_wild_domain(self):
+ helper = self._makeOne('secret', parent_domain=True, wild_domain=True)
+ request = self._makeRequest()
+ request.domain = 'www.example.com'
+ result = helper.remember(request, 'other')
+ self.assertEqual(len(result), 1)
+ self.assertTrue(
+ result[0][1].endswith(
+ '; Domain=.example.com; Path=/; SameSite=Lax'
+ )
+ )
+
+ def test_remember_explicit_domain(self):
+ helper = self._makeOne('secret', domain='pyramid.bazinga')
+ request = self._makeRequest()
+ request.domain = 'www.example.com'
+ result = helper.remember(request, 'other')
+ self.assertEqual(len(result), 1)
+
+ self.assertEqual(result[0][0], 'Set-Cookie')
+ self.assertTrue(
+ result[0][1].endswith(
+ '; Domain=pyramid.bazinga; Path=/; SameSite=Lax'
+ )
+ )
+ self.assertTrue(result[0][1].startswith('auth_tkt='))
+
+ def test_remember_domain_supercedes_parent_and_wild_domain(self):
+ helper = self._makeOne(
+ 'secret',
+ domain='pyramid.bazinga',
+ parent_domain=True,
+ wild_domain=True,
+ )
+ request = self._makeRequest()
+ request.domain = 'www.example.com'
+ result = helper.remember(request, 'other')
+ self.assertEqual(len(result), 1)
+ self.assertTrue(
+ result[0][1].endswith(
+ '; Domain=pyramid.bazinga; Path=/; SameSite=Lax'
+ )
+ )
+
+ def test_remember_binary_userid(self):
+ import base64
+
+ helper = self._makeOne('secret')
+ request = self._makeRequest()
+ result = helper.remember(request, b'userid')
+ values = self._parseHeaders(result)
+ self.assertEqual(len(result), 3)
+ val = self._cookieValue(values[0])
+ self.assertEqual(
+ val['userid'], text_(base64.b64encode(b'userid').strip())
+ )
+ self.assertEqual(val['user_data'], 'userid_type:b64str')
+
+ def test_remember_int_userid(self):
+ helper = self._makeOne('secret')
+ request = self._makeRequest()
+ result = helper.remember(request, 1)
+ values = self._parseHeaders(result)
+ self.assertEqual(len(result), 3)
+ val = self._cookieValue(values[0])
+ self.assertEqual(val['userid'], '1')
+ self.assertEqual(val['user_data'], 'userid_type:int')
+
+ def test_remember_unicode_userid(self):
+ import base64
+
+ helper = self._makeOne('secret')
+ request = self._makeRequest()
+ userid = text_(b'\xc2\xa9', 'utf-8')
+ result = helper.remember(request, userid)
+ values = self._parseHeaders(result)
+ self.assertEqual(len(result), 3)
+ val = self._cookieValue(values[0])
+ self.assertEqual(
+ val['userid'], text_(base64.b64encode(userid.encode('utf-8')))
+ )
+ self.assertEqual(val['user_data'], 'userid_type:b64unicode')
+
+ def test_remember_insane_userid(self):
+ helper = self._makeOne('secret')
+ request = self._makeRequest()
+ userid = object()
+ with warnings.catch_warnings(record=True) as w:
+ warnings.simplefilter('always', RuntimeWarning)
+ result = helper.remember(request, userid)
+ self.assertTrue(str(w[-1].message).startswith('userid is of type'))
+ values = self._parseHeaders(result)
+ self.assertEqual(len(result), 3)
+ value = values[0]
+ self.assertTrue('userid' in value.value)
+
+ def test_remember_max_age(self):
+ helper = self._makeOne('secret')
+ request = self._makeRequest()
+ result = helper.remember(request, 'userid', max_age=500)
+ values = self._parseHeaders(result)
+ self.assertEqual(len(result), 3)
+
+ self.assertEqual(values[0]['max-age'], '500')
+ self.assertTrue(values[0]['expires'])
+
+ def test_remember_str_max_age(self):
+ helper = self._makeOne('secret')
+ request = self._makeRequest()
+ result = helper.remember(request, 'userid', max_age='500')
+ values = self._parseHeaders(result)
+ self.assertEqual(len(result), 3)
+
+ self.assertEqual(values[0]['max-age'], '500')
+ self.assertTrue(values[0]['expires'])
+
+ def test_remember_str_max_age_invalid(self):
+ helper = self._makeOne('secret')
+ request = self._makeRequest()
+ self.assertRaises(
+ ValueError,
+ helper.remember,
+ request,
+ 'userid',
+ max_age='invalid value',
+ )
+
+ def test_remember_tokens(self):
+ helper = self._makeOne('secret')
+ request = self._makeRequest()
+ result = helper.remember(request, 'other', tokens=('foo', 'bar'))
+ self.assertEqual(len(result), 3)
+
+ self.assertEqual(result[0][0], 'Set-Cookie')
+ self.assertTrue("/tokens=foo|bar/" in result[0][1])
+
+ self.assertEqual(result[1][0], 'Set-Cookie')
+ self.assertTrue("/tokens=foo|bar/" in result[1][1])
+
+ self.assertEqual(result[2][0], 'Set-Cookie')
+ self.assertTrue("/tokens=foo|bar/" in result[2][1])
+
+ def test_remember_samesite_nondefault(self):
+ helper = self._makeOne('secret', samesite='Strict')
+ request = self._makeRequest()
+ result = helper.remember(request, 'userid')
+ self.assertEqual(len(result), 3)
+
+ self.assertEqual(result[0][0], 'Set-Cookie')
+ cookieval = result[0][1]
+ self.assertTrue(
+ 'SameSite=Strict' in [x.strip() for x in cookieval.split(';')],
+ cookieval,
+ )
+
+ self.assertEqual(result[1][0], 'Set-Cookie')
+ cookieval = result[1][1]
+ self.assertTrue(
+ 'SameSite=Strict' in [x.strip() for x in cookieval.split(';')],
+ cookieval,
+ )
+
+ self.assertEqual(result[2][0], 'Set-Cookie')
+ cookieval = result[2][1]
+ self.assertTrue(
+ 'SameSite=Strict' in [x.strip() for x in cookieval.split(';')],
+ cookieval,
+ )
+
+ def test_remember_samesite_default(self):
+ helper = self._makeOne('secret')
+ request = self._makeRequest()
+ result = helper.remember(request, 'userid')
+ self.assertEqual(len(result), 3)
+
+ self.assertEqual(result[0][0], 'Set-Cookie')
+ cookieval = result[0][1]
+ self.assertTrue(
+ 'SameSite=Lax' in [x.strip() for x in cookieval.split(';')],
+ cookieval,
+ )
+
+ self.assertEqual(result[1][0], 'Set-Cookie')
+ cookieval = result[1][1]
+ self.assertTrue(
+ 'SameSite=Lax' in [x.strip() for x in cookieval.split(';')],
+ cookieval,
+ )
+
+ self.assertEqual(result[2][0], 'Set-Cookie')
+ cookieval = result[2][1]
+ self.assertTrue(
+ 'SameSite=Lax' in [x.strip() for x in cookieval.split(';')],
+ cookieval,
+ )
+
+ def test_remember_unicode_but_ascii_token(self):
+ helper = self._makeOne('secret')
+ request = self._makeRequest()
+ la = text_(b'foo', 'utf-8')
+ result = helper.remember(request, 'other', tokens=(la,))
+ # tokens must be str type on both Python 2 and 3
+ self.assertTrue("/tokens=foo/" in result[0][1])
+
+ def test_remember_nonascii_token(self):
+ helper = self._makeOne('secret')
+ request = self._makeRequest()
+ la = text_(b'La Pe\xc3\xb1a', 'utf-8')
+ self.assertRaises(
+ ValueError, helper.remember, request, 'other', tokens=(la,)
+ )
+
+ def test_remember_invalid_token_format(self):
+ helper = self._makeOne('secret')
+ request = self._makeRequest()
+ self.assertRaises(
+ ValueError, helper.remember, request, 'other', tokens=('foo bar',)
+ )
+ self.assertRaises(
+ ValueError, helper.remember, request, 'other', tokens=('1bar',)
+ )
+
+ def test_forget(self):
+ helper = self._makeOne('secret')
+ request = self._makeRequest()
+ headers = helper.forget(request)
+ self.assertEqual(len(headers), 3)
+ name, value = headers[0]
+ self.assertEqual(name, 'Set-Cookie')
+ self.assertEqual(
+ value,
+ 'auth_tkt=; Max-Age=0; Path=/; '
+ 'expires=Wed, 31-Dec-97 23:59:59 GMT; SameSite=Lax',
+ )
+ name, value = headers[1]
+ self.assertEqual(name, 'Set-Cookie')
+ self.assertEqual(
+ value,
+ 'auth_tkt=; Domain=localhost; Max-Age=0; Path=/; '
+ 'expires=Wed, 31-Dec-97 23:59:59 GMT; SameSite=Lax',
+ )
+ name, value = headers[2]
+ self.assertEqual(name, 'Set-Cookie')
+ self.assertEqual(
+ value,
+ 'auth_tkt=; Domain=.localhost; Max-Age=0; Path=/; '
+ 'expires=Wed, 31-Dec-97 23:59:59 GMT; SameSite=Lax',
+ )
+
+
+class TestAuthTicket(unittest.TestCase):
+ def _makeOne(self, *arg, **kw):
+ from pyramid.authentication import AuthTicket
+
+ return AuthTicket(*arg, **kw)
+
+ def test_ctor_with_tokens(self):
+ ticket = self._makeOne('secret', 'userid', 'ip', tokens=('a', 'b'))
+ self.assertEqual(ticket.tokens, 'a,b')
+
+ def test_ctor_with_time(self):
+ ticket = self._makeOne('secret', 'userid', 'ip', time='time')
+ self.assertEqual(ticket.time, 'time')
+
+ def test_digest(self):
+ ticket = self._makeOne('secret', 'userid', '0.0.0.0', time=10)
+ result = ticket.digest()
+ self.assertEqual(result, '126fd6224912187ee9ffa80e0b81420c')
+
+ def test_digest_sha512(self):
+ ticket = self._makeOne(
+ 'secret', 'userid', '0.0.0.0', time=10, hashalg='sha512'
+ )
+ result = ticket.digest()
+ self.assertEqual(
+ result,
+ '74770b2e0d5b1a54c2a466ec567a40f7d7823576aa49'
+ '3c65fc3445e9b44097f4a80410319ef8cb256a2e60b9'
+ 'c2002e48a9e33a3e8ee4379352c04ef96d2cb278',
+ )
+
+ def test_cookie_value(self):
+ ticket = self._makeOne(
+ 'secret', 'userid', '0.0.0.0', time=10, tokens=('a', 'b')
+ )
+ result = ticket.cookie_value()
+ self.assertEqual(
+ result, '66f9cc3e423dc57c91df696cf3d1f0d80000000auserid!a,b!'
+ )
+
+ def test_ipv4(self):
+ ticket = self._makeOne(
+ 'secret', 'userid', '198.51.100.1', time=10, hashalg='sha256'
+ )
+ result = ticket.cookie_value()
+ self.assertEqual(
+ result,
+ 'b3e7156db4f8abde4439c4a6499a0668f9e7ffd7fa27b'
+ '798400ecdade8d76c530000000auserid!',
+ )
+
+ def test_ipv6(self):
+ ticket = self._makeOne(
+ 'secret', 'userid', '2001:db8::1', time=10, hashalg='sha256'
+ )
+ result = ticket.cookie_value()
+ self.assertEqual(
+ result,
+ 'd025b601a0f12ca6d008aa35ff3a22b7d8f3d1c1456c8'
+ '5becf8760cd7a2fa4910000000auserid!',
+ )
+
+
+class TestBadTicket(unittest.TestCase):
+ def _makeOne(self, msg, expected=None):
+ from pyramid.authentication import BadTicket
+
+ return BadTicket(msg, expected)
+
+ def test_it(self):
+ exc = self._makeOne('msg', expected=True)
+ self.assertEqual(exc.expected, True)
+ self.assertTrue(isinstance(exc, Exception))
+
+
+class Test_parse_ticket(unittest.TestCase):
+ def _callFUT(self, secret, ticket, ip, hashalg='md5'):
+ from pyramid.authentication import parse_ticket
+
+ return parse_ticket(secret, ticket, ip, hashalg)
+
+ def _assertRaisesBadTicket(self, secret, ticket, ip, hashalg='md5'):
+ from pyramid.authentication import BadTicket
+
+ self.assertRaises(
+ BadTicket, self._callFUT, secret, ticket, ip, hashalg
+ )
+
+ def test_bad_timestamp(self):
+ ticket = 'x' * 64
+ self._assertRaisesBadTicket('secret', ticket, 'ip')
+
+ def test_bad_userid_or_data(self):
+ ticket = 'x' * 32 + '11111111' + 'x' * 10
+ self._assertRaisesBadTicket('secret', ticket, 'ip')
+
+ def test_digest_sig_incorrect(self):
+ ticket = 'x' * 32 + '11111111' + 'a!b!c'
+ self._assertRaisesBadTicket('secret', ticket, '0.0.0.0')
+
+ def test_correct_with_user_data(self):
+ ticket = text_('66f9cc3e423dc57c91df696cf3d1f0d80000000auserid!a,b!')
+ result = self._callFUT('secret', ticket, '0.0.0.0')
+ self.assertEqual(result, (10, 'userid', ['a', 'b'], ''))
+
+ def test_correct_with_user_data_sha512(self):
+ ticket = text_(
+ '7d947cdef99bad55f8e3382a8bd089bb9dd0547f7925b7d189adc1'
+ '160cab0ec0e6888faa41eba641a18522b26f19109f3ffafb769767'
+ 'ba8a26d02aaeae56599a0000000auserid!a,b!'
+ )
+ result = self._callFUT('secret', ticket, '0.0.0.0', 'sha512')
+ self.assertEqual(result, (10, 'userid', ['a', 'b'], ''))
+
+ def test_ipv4(self):
+ ticket = text_(
+ 'b3e7156db4f8abde4439c4a6499a0668f9e7ffd7fa27b798400ecd'
+ 'ade8d76c530000000auserid!'
+ )
+ result = self._callFUT('secret', ticket, '198.51.100.1', 'sha256')
+ self.assertEqual(result, (10, 'userid', [''], ''))
+
+ def test_ipv6(self):
+ ticket = text_(
+ 'd025b601a0f12ca6d008aa35ff3a22b7d8f3d1c1456c85becf8760'
+ 'cd7a2fa4910000000auserid!'
+ )
+ result = self._callFUT('secret', ticket, '2001:db8::1', 'sha256')
+ self.assertEqual(result, (10, 'userid', [''], ''))
+
+
class TestSessionAuthenticationPolicy(unittest.TestCase):
def _getTargetClass(self):
from pyramid.authentication import SessionAuthenticationPolicy
@@ -990,6 +1911,14 @@ class DummyContext:
pass
+class DummyCookies(object):
+ def __init__(self, cookie):
+ self.cookie = cookie
+
+ def get(self, name):
+ return self.cookie
+
+
class DummyRequest:
domain = 'localhost'
@@ -998,6 +1927,10 @@ class DummyRequest:
self.session = session or {}
self.registry = registry
self.callbacks = []
+ self.cookies = DummyCookies(cookie)
+
+ def add_response_callback(self, callback):
+ self.callbacks.append(callback)
class DummyWhoPlugin:
@@ -1021,3 +1954,68 @@ class DummyCookieHelper:
def forget(self, *arg):
return []
+
+
+class DummyAuthTktModule(object):
+ def __init__(
+ self,
+ timestamp=0,
+ userid='userid',
+ tokens=(),
+ user_data='',
+ parse_raise=False,
+ hashalg="md5",
+ ):
+ self.timestamp = timestamp
+ self.userid = userid
+ self.tokens = tokens
+ self.user_data = user_data
+ self.parse_raise = parse_raise
+ self.hashalg = hashalg
+
+ def parse_ticket(secret, value, remote_addr, hashalg):
+ self.secret = secret
+ self.value = value
+ self.remote_addr = remote_addr
+ if self.parse_raise:
+ raise self.BadTicket()
+ return self.timestamp, self.userid, self.tokens, self.user_data
+
+ self.parse_ticket = parse_ticket
+
+ class AuthTicket(object):
+ def __init__(self, secret, userid, remote_addr, **kw):
+ self.secret = secret
+ self.userid = userid
+ self.remote_addr = remote_addr
+ self.kw = kw
+
+ def cookie_value(self):
+ result = {
+ 'secret': self.secret,
+ 'userid': self.userid,
+ 'remote_addr': self.remote_addr,
+ }
+ result.update(self.kw)
+ tokens = result.pop('tokens', None)
+ if tokens is not None:
+ tokens = '|'.join(tokens)
+ result['tokens'] = tokens
+ items = sorted(result.items())
+ new_items = []
+ for k, v in items:
+ if isinstance(v, bytes):
+ v = text_(v)
+ new_items.append((k, v))
+ result = '/'.join(['%s=%s' % (k, v) for k, v in new_items])
+ return result
+
+ self.AuthTicket = AuthTicket
+
+ class BadTicket(Exception):
+ pass
+
+
+class DummyResponse:
+ def __init__(self):
+ self.headerlist = []
diff --git a/tests/test_security.py b/tests/test_security.py
index 97aec42c6..f14159156 100644
--- a/tests/test_security.py
+++ b/tests/test_security.py
@@ -1,9 +1,6 @@
import unittest
-import warnings
-from http.cookies import SimpleCookie
from pyramid import testing
-from pyramid.util import text_
class TestAllPermissionsList(unittest.TestCase):
@@ -940,1008 +937,3 @@ class TestSessionAuthenticationHelper(unittest.TestCase):
result = helper.forget(request)
self.assertEqual(request.session.get('userid'), None)
self.assertEqual(result, [])
-
-
-class TestAuthTicket(unittest.TestCase):
- def _makeOne(self, *arg, **kw):
- from pyramid.security import AuthTicket
-
- return AuthTicket(*arg, **kw)
-
- def test_ctor_with_tokens(self):
- ticket = self._makeOne('secret', 'userid', 'ip', tokens=('a', 'b'))
- self.assertEqual(ticket.tokens, 'a,b')
-
- def test_ctor_with_time(self):
- ticket = self._makeOne('secret', 'userid', 'ip', time='time')
- self.assertEqual(ticket.time, 'time')
-
- def test_digest(self):
- ticket = self._makeOne('secret', 'userid', '0.0.0.0', time=10)
- result = ticket.digest()
- self.assertEqual(result, '126fd6224912187ee9ffa80e0b81420c')
-
- def test_digest_sha512(self):
- ticket = self._makeOne(
- 'secret', 'userid', '0.0.0.0', time=10, hashalg='sha512'
- )
- result = ticket.digest()
- self.assertEqual(
- result,
- '74770b2e0d5b1a54c2a466ec567a40f7d7823576aa49'
- '3c65fc3445e9b44097f4a80410319ef8cb256a2e60b9'
- 'c2002e48a9e33a3e8ee4379352c04ef96d2cb278',
- )
-
- def test_cookie_value(self):
- ticket = self._makeOne(
- 'secret', 'userid', '0.0.0.0', time=10, tokens=('a', 'b')
- )
- result = ticket.cookie_value()
- self.assertEqual(
- result, '66f9cc3e423dc57c91df696cf3d1f0d80000000auserid!a,b!'
- )
-
- def test_ipv4(self):
- ticket = self._makeOne(
- 'secret', 'userid', '198.51.100.1', time=10, hashalg='sha256'
- )
- result = ticket.cookie_value()
- self.assertEqual(
- result,
- 'b3e7156db4f8abde4439c4a6499a0668f9e7ffd7fa27b'
- '798400ecdade8d76c530000000auserid!',
- )
-
- def test_ipv6(self):
- ticket = self._makeOne(
- 'secret', 'userid', '2001:db8::1', time=10, hashalg='sha256'
- )
- result = ticket.cookie_value()
- self.assertEqual(
- result,
- 'd025b601a0f12ca6d008aa35ff3a22b7d8f3d1c1456c8'
- '5becf8760cd7a2fa4910000000auserid!',
- )
-
-
-class TestBadTicket(unittest.TestCase):
- def _makeOne(self, msg, expected=None):
- from pyramid.security import BadTicket
-
- return BadTicket(msg, expected)
-
- def test_it(self):
- exc = self._makeOne('msg', expected=True)
- self.assertEqual(exc.expected, True)
- self.assertTrue(isinstance(exc, Exception))
-
-
-class Test_parse_ticket(unittest.TestCase):
- def _callFUT(self, secret, ticket, ip, hashalg='md5'):
- from pyramid.security import parse_ticket
-
- return parse_ticket(secret, ticket, ip, hashalg)
-
- def _assertRaisesBadTicket(self, secret, ticket, ip, hashalg='md5'):
- from pyramid.security import BadTicket
-
- self.assertRaises(
- BadTicket, self._callFUT, secret, ticket, ip, hashalg
- )
-
- def test_bad_timestamp(self):
- ticket = 'x' * 64
- self._assertRaisesBadTicket('secret', ticket, 'ip')
-
- def test_bad_userid_or_data(self):
- ticket = 'x' * 32 + '11111111' + 'x' * 10
- self._assertRaisesBadTicket('secret', ticket, 'ip')
-
- def test_digest_sig_incorrect(self):
- ticket = 'x' * 32 + '11111111' + 'a!b!c'
- self._assertRaisesBadTicket('secret', ticket, '0.0.0.0')
-
- def test_correct_with_user_data(self):
- ticket = text_('66f9cc3e423dc57c91df696cf3d1f0d80000000auserid!a,b!')
- result = self._callFUT('secret', ticket, '0.0.0.0')
- self.assertEqual(result, (10, 'userid', ['a', 'b'], ''))
-
- def test_correct_with_user_data_sha512(self):
- ticket = text_(
- '7d947cdef99bad55f8e3382a8bd089bb9dd0547f7925b7d189adc1'
- '160cab0ec0e6888faa41eba641a18522b26f19109f3ffafb769767'
- 'ba8a26d02aaeae56599a0000000auserid!a,b!'
- )
- result = self._callFUT('secret', ticket, '0.0.0.0', 'sha512')
- self.assertEqual(result, (10, 'userid', ['a', 'b'], ''))
-
- def test_ipv4(self):
- ticket = text_(
- 'b3e7156db4f8abde4439c4a6499a0668f9e7ffd7fa27b798400ecd'
- 'ade8d76c530000000auserid!'
- )
- result = self._callFUT('secret', ticket, '198.51.100.1', 'sha256')
- self.assertEqual(result, (10, 'userid', [''], ''))
-
- def test_ipv6(self):
- ticket = text_(
- 'd025b601a0f12ca6d008aa35ff3a22b7d8f3d1c1456c85becf8760'
- 'cd7a2fa4910000000auserid!'
- )
- result = self._callFUT('secret', ticket, '2001:db8::1', 'sha256')
- self.assertEqual(result, (10, 'userid', [''], ''))
-
-
-class TestAuthTktCookieHelper(unittest.TestCase):
- def _getTargetClass(self):
- from pyramid.security import AuthTktCookieHelper
-
- return AuthTktCookieHelper
-
- def _makeOne(self, *arg, **kw):
- helper = self._getTargetClass()(*arg, **kw)
- auth_tkt = DummyAuthTktModule()
- helper.auth_tkt = auth_tkt
- helper.AuthTicket = auth_tkt.AuthTicket
- helper.parse_ticket = auth_tkt.parse_ticket
- helper.BadTicket = auth_tkt.BadTicket
- return helper
-
- def _makeRequest(self, cookie=None, ipv6=False):
- environ = {'wsgi.version': (1, 0)}
-
- if ipv6 is False:
- environ['REMOTE_ADDR'] = '1.1.1.1'
- else:
- environ['REMOTE_ADDR'] = '::1'
- environ['SERVER_NAME'] = 'localhost'
- return DummyRequest(environ, cookie=cookie)
-
- def _cookieValue(self, cookie):
- items = cookie.value.split('/')
- D = {}
- for item in items:
- k, v = item.split('=', 1)
- D[k] = v
- return D
-
- def _parseHeaders(self, headers):
- return [self._parseHeader(header) for header in headers]
-
- def _parseHeader(self, header):
- cookie = self._parseCookie(header[1])
- return cookie
-
- def _parseCookie(self, cookie):
- cookies = SimpleCookie()
- cookies.load(cookie)
- return cookies.get('auth_tkt')
-
- def test_init_cookie_str_reissue_invalid(self):
- self.assertRaises(
- ValueError, self._makeOne, 'secret', reissue_time='invalid value'
- )
-
- def test_init_cookie_str_timeout_invalid(self):
- self.assertRaises(
- ValueError, self._makeOne, 'secret', timeout='invalid value'
- )
-
- def test_init_cookie_str_max_age_invalid(self):
- self.assertRaises(
- ValueError, self._makeOne, 'secret', max_age='invalid value'
- )
-
- def test_identify_nocookie(self):
- helper = self._makeOne('secret')
- request = self._makeRequest()
- result = helper.identify(request)
- self.assertEqual(result, None)
-
- def test_identify_cookie_value_is_None(self):
- helper = self._makeOne('secret')
- request = self._makeRequest(None)
- result = helper.identify(request)
- self.assertEqual(result, None)
-
- def test_identify_good_cookie_include_ip(self):
- helper = self._makeOne('secret', include_ip=True)
- request = self._makeRequest('ticket')
- result = helper.identify(request)
- self.assertEqual(len(result), 4)
- self.assertEqual(result['tokens'], ())
- self.assertEqual(result['userid'], 'userid')
- self.assertEqual(result['userdata'], '')
- self.assertEqual(result['timestamp'], 0)
- self.assertEqual(helper.auth_tkt.value, 'ticket')
- self.assertEqual(helper.auth_tkt.remote_addr, '1.1.1.1')
- self.assertEqual(helper.auth_tkt.secret, 'secret')
- environ = request.environ
- self.assertEqual(environ['REMOTE_USER_TOKENS'], ())
- self.assertEqual(environ['REMOTE_USER_DATA'], '')
- self.assertEqual(environ['AUTH_TYPE'], 'cookie')
-
- def test_identify_good_cookie_include_ipv6(self):
- helper = self._makeOne('secret', include_ip=True)
- request = self._makeRequest('ticket', ipv6=True)
- result = helper.identify(request)
- self.assertEqual(len(result), 4)
- self.assertEqual(result['tokens'], ())
- self.assertEqual(result['userid'], 'userid')
- self.assertEqual(result['userdata'], '')
- self.assertEqual(result['timestamp'], 0)
- self.assertEqual(helper.auth_tkt.value, 'ticket')
- self.assertEqual(helper.auth_tkt.remote_addr, '::1')
- self.assertEqual(helper.auth_tkt.secret, 'secret')
- environ = request.environ
- self.assertEqual(environ['REMOTE_USER_TOKENS'], ())
- self.assertEqual(environ['REMOTE_USER_DATA'], '')
- self.assertEqual(environ['AUTH_TYPE'], 'cookie')
-
- def test_identify_good_cookie_dont_include_ip(self):
- helper = self._makeOne('secret', include_ip=False)
- request = self._makeRequest('ticket')
- result = helper.identify(request)
- self.assertEqual(len(result), 4)
- self.assertEqual(result['tokens'], ())
- self.assertEqual(result['userid'], 'userid')
- self.assertEqual(result['userdata'], '')
- self.assertEqual(result['timestamp'], 0)
- self.assertEqual(helper.auth_tkt.value, 'ticket')
- self.assertEqual(helper.auth_tkt.remote_addr, '0.0.0.0')
- self.assertEqual(helper.auth_tkt.secret, 'secret')
- environ = request.environ
- self.assertEqual(environ['REMOTE_USER_TOKENS'], ())
- self.assertEqual(environ['REMOTE_USER_DATA'], '')
- self.assertEqual(environ['AUTH_TYPE'], 'cookie')
-
- def test_identify_good_cookie_int_useridtype(self):
- helper = self._makeOne('secret', include_ip=False)
- helper.auth_tkt.userid = '1'
- helper.auth_tkt.user_data = 'userid_type:int'
- request = self._makeRequest('ticket')
- result = helper.identify(request)
- self.assertEqual(len(result), 4)
- self.assertEqual(result['tokens'], ())
- self.assertEqual(result['userid'], 1)
- self.assertEqual(result['userdata'], 'userid_type:int')
- self.assertEqual(result['timestamp'], 0)
- environ = request.environ
- self.assertEqual(environ['REMOTE_USER_TOKENS'], ())
- self.assertEqual(environ['REMOTE_USER_DATA'], 'userid_type:int')
- self.assertEqual(environ['AUTH_TYPE'], 'cookie')
-
- def test_identify_nonuseridtype_user_data(self):
- helper = self._makeOne('secret', include_ip=False)
- helper.auth_tkt.userid = '1'
- helper.auth_tkt.user_data = 'bogus:int'
- request = self._makeRequest('ticket')
- result = helper.identify(request)
- self.assertEqual(len(result), 4)
- self.assertEqual(result['tokens'], ())
- self.assertEqual(result['userid'], '1')
- self.assertEqual(result['userdata'], 'bogus:int')
- self.assertEqual(result['timestamp'], 0)
- environ = request.environ
- self.assertEqual(environ['REMOTE_USER_TOKENS'], ())
- self.assertEqual(environ['REMOTE_USER_DATA'], 'bogus:int')
- self.assertEqual(environ['AUTH_TYPE'], 'cookie')
-
- def test_identify_good_cookie_unknown_useridtype(self):
- helper = self._makeOne('secret', include_ip=False)
- helper.auth_tkt.userid = 'abc'
- helper.auth_tkt.user_data = 'userid_type:unknown'
- request = self._makeRequest('ticket')
- result = helper.identify(request)
- self.assertEqual(len(result), 4)
- self.assertEqual(result['tokens'], ())
- self.assertEqual(result['userid'], 'abc')
- self.assertEqual(result['userdata'], 'userid_type:unknown')
- self.assertEqual(result['timestamp'], 0)
- environ = request.environ
- self.assertEqual(environ['REMOTE_USER_TOKENS'], ())
- self.assertEqual(environ['REMOTE_USER_DATA'], 'userid_type:unknown')
- self.assertEqual(environ['AUTH_TYPE'], 'cookie')
-
- def test_identify_good_cookie_b64str_useridtype(self):
- from base64 import b64encode
-
- helper = self._makeOne('secret', include_ip=False)
- helper.auth_tkt.userid = b64encode(b'encoded').strip()
- helper.auth_tkt.user_data = 'userid_type:b64str'
- request = self._makeRequest('ticket')
- result = helper.identify(request)
- self.assertEqual(len(result), 4)
- self.assertEqual(result['tokens'], ())
- self.assertEqual(result['userid'], b'encoded')
- self.assertEqual(result['userdata'], 'userid_type:b64str')
- self.assertEqual(result['timestamp'], 0)
- environ = request.environ
- self.assertEqual(environ['REMOTE_USER_TOKENS'], ())
- self.assertEqual(environ['REMOTE_USER_DATA'], 'userid_type:b64str')
- self.assertEqual(environ['AUTH_TYPE'], 'cookie')
-
- def test_identify_good_cookie_b64unicode_useridtype(self):
- from base64 import b64encode
-
- helper = self._makeOne('secret', include_ip=False)
- helper.auth_tkt.userid = b64encode(b'\xc3\xa9ncoded').strip()
- helper.auth_tkt.user_data = 'userid_type:b64unicode'
- request = self._makeRequest('ticket')
- result = helper.identify(request)
- self.assertEqual(len(result), 4)
- self.assertEqual(result['tokens'], ())
- self.assertEqual(result['userid'], text_(b'\xc3\xa9ncoded', 'utf-8'))
- self.assertEqual(result['userdata'], 'userid_type:b64unicode')
- self.assertEqual(result['timestamp'], 0)
- environ = request.environ
- self.assertEqual(environ['REMOTE_USER_TOKENS'], ())
- self.assertEqual(environ['REMOTE_USER_DATA'], 'userid_type:b64unicode')
- self.assertEqual(environ['AUTH_TYPE'], 'cookie')
-
- def test_identify_bad_cookie(self):
- helper = self._makeOne('secret', include_ip=True)
- helper.auth_tkt.parse_raise = True
- request = self._makeRequest('ticket')
- result = helper.identify(request)
- self.assertEqual(result, None)
-
- def test_identify_cookie_timeout(self):
- helper = self._makeOne('secret', timeout=1)
- self.assertEqual(helper.timeout, 1)
-
- def test_identify_cookie_str_timeout(self):
- helper = self._makeOne('secret', timeout='1')
- self.assertEqual(helper.timeout, 1)
-
- def test_identify_cookie_timeout_aged(self):
- import time
-
- helper = self._makeOne('secret', timeout=10)
- now = time.time()
- helper.auth_tkt.timestamp = now - 1
- helper.now = now + 10
- helper.auth_tkt.tokens = (text_('a'),)
- request = self._makeRequest('bogus')
- result = helper.identify(request)
- self.assertFalse(result)
-
- def test_identify_cookie_reissue(self):
- import time
-
- helper = self._makeOne('secret', timeout=10, reissue_time=0)
- now = time.time()
- helper.auth_tkt.timestamp = now
- helper.now = now + 1
- helper.auth_tkt.tokens = (text_('a'),)
- request = self._makeRequest('bogus')
- result = helper.identify(request)
- self.assertTrue(result)
- self.assertEqual(len(request.callbacks), 1)
- response = DummyResponse()
- request.callbacks[0](request, response)
- self.assertEqual(len(response.headerlist), 3)
- self.assertEqual(response.headerlist[0][0], 'Set-Cookie')
-
- def test_identify_cookie_str_reissue(self):
- import time
-
- helper = self._makeOne('secret', timeout=10, reissue_time='0')
- now = time.time()
- helper.auth_tkt.timestamp = now
- helper.now = now + 1
- helper.auth_tkt.tokens = (text_('a'),)
- request = self._makeRequest('bogus')
- result = helper.identify(request)
- self.assertTrue(result)
- self.assertEqual(len(request.callbacks), 1)
- response = DummyResponse()
- request.callbacks[0](request, response)
- self.assertEqual(len(response.headerlist), 3)
- self.assertEqual(response.headerlist[0][0], 'Set-Cookie')
-
- def test_identify_cookie_reissue_already_reissued_this_request(self):
- import time
-
- helper = self._makeOne('secret', timeout=10, reissue_time=0)
- now = time.time()
- helper.auth_tkt.timestamp = now
- helper.now = now + 1
- request = self._makeRequest('bogus')
- request._authtkt_reissued = True
- result = helper.identify(request)
- self.assertTrue(result)
- self.assertEqual(len(request.callbacks), 0)
-
- def test_identify_cookie_reissue_notyet(self):
- import time
-
- helper = self._makeOne('secret', timeout=10, reissue_time=10)
- now = time.time()
- helper.auth_tkt.timestamp = now
- helper.now = now + 1
- request = self._makeRequest('bogus')
- result = helper.identify(request)
- self.assertTrue(result)
- self.assertEqual(len(request.callbacks), 0)
-
- def test_identify_cookie_reissue_revoked_by_forget(self):
- import time
-
- helper = self._makeOne('secret', timeout=10, reissue_time=0)
- now = time.time()
- helper.auth_tkt.timestamp = now
- helper.now = now + 1
- request = self._makeRequest('bogus')
- result = helper.identify(request)
- self.assertTrue(result)
- self.assertEqual(len(request.callbacks), 1)
- result = helper.forget(request)
- self.assertTrue(result)
- self.assertEqual(len(request.callbacks), 1)
- response = DummyResponse()
- request.callbacks[0](request, response)
- self.assertEqual(len(response.headerlist), 0)
-
- def test_identify_cookie_reissue_revoked_by_remember(self):
- import time
-
- helper = self._makeOne('secret', timeout=10, reissue_time=0)
- now = time.time()
- helper.auth_tkt.timestamp = now
- helper.now = now + 1
- request = self._makeRequest('bogus')
- result = helper.identify(request)
- self.assertTrue(result)
- self.assertEqual(len(request.callbacks), 1)
- result = helper.remember(request, 'bob')
- self.assertTrue(result)
- self.assertEqual(len(request.callbacks), 1)
- response = DummyResponse()
- request.callbacks[0](request, response)
- self.assertEqual(len(response.headerlist), 0)
-
- def test_identify_cookie_reissue_with_tokens_default(self):
- # see https://github.com/Pylons/pyramid/issues#issue/108
- import time
-
- helper = self._makeOne('secret', timeout=10, reissue_time=0)
- auth_tkt = DummyAuthTktModule(tokens=[''])
- helper.auth_tkt = auth_tkt
- helper.AuthTicket = auth_tkt.AuthTicket
- helper.parse_ticket = auth_tkt.parse_ticket
- helper.BadTicket = auth_tkt.BadTicket
- now = time.time()
- helper.auth_tkt.timestamp = now
- helper.now = now + 1
- request = self._makeRequest('bogus')
- result = helper.identify(request)
- self.assertTrue(result)
- self.assertEqual(len(request.callbacks), 1)
- response = DummyResponse()
- request.callbacks[0](None, response)
- self.assertEqual(len(response.headerlist), 3)
- self.assertEqual(response.headerlist[0][0], 'Set-Cookie')
- self.assertTrue("/tokens=/" in response.headerlist[0][1])
-
- def test_remember(self):
- helper = self._makeOne('secret')
- request = self._makeRequest()
- result = helper.remember(request, 'userid')
- self.assertEqual(len(result), 3)
-
- self.assertEqual(result[0][0], 'Set-Cookie')
- self.assertTrue(result[0][1].endswith('; Path=/; SameSite=Lax'))
- self.assertTrue(result[0][1].startswith('auth_tkt='))
-
- self.assertEqual(result[1][0], 'Set-Cookie')
- self.assertTrue(
- result[1][1].endswith('; Domain=localhost; Path=/; SameSite=Lax')
- )
- self.assertTrue(result[1][1].startswith('auth_tkt='))
-
- self.assertEqual(result[2][0], 'Set-Cookie')
- self.assertTrue(
- result[2][1].endswith('; Domain=.localhost; Path=/; SameSite=Lax')
- )
- self.assertTrue(result[2][1].startswith('auth_tkt='))
-
- def test_remember_nondefault_samesite(self):
- helper = self._makeOne('secret', samesite='Strict')
- request = self._makeRequest()
- result = helper.remember(request, 'userid')
- self.assertEqual(len(result), 3)
-
- self.assertEqual(result[0][0], 'Set-Cookie')
- self.assertTrue(result[0][1].endswith('; Path=/; SameSite=Strict'))
- self.assertTrue(result[0][1].startswith('auth_tkt='))
-
- self.assertEqual(result[1][0], 'Set-Cookie')
- self.assertTrue(
- result[1][1].endswith(
- '; Domain=localhost; Path=/; SameSite=Strict'
- )
- )
- self.assertTrue(result[1][1].startswith('auth_tkt='))
-
- self.assertEqual(result[2][0], 'Set-Cookie')
- self.assertTrue(
- result[2][1].endswith(
- '; Domain=.localhost; Path=/; SameSite=Strict'
- )
- )
- self.assertTrue(result[2][1].startswith('auth_tkt='))
-
- def test_remember_None_samesite(self):
- helper = self._makeOne('secret', samesite=None)
- request = self._makeRequest()
- result = helper.remember(request, 'userid')
- self.assertEqual(len(result), 3)
-
- self.assertEqual(result[0][0], 'Set-Cookie')
- self.assertTrue(result[0][1].endswith('; Path=/')) # no samesite
- self.assertTrue(result[0][1].startswith('auth_tkt='))
-
- self.assertEqual(result[1][0], 'Set-Cookie')
- self.assertTrue(result[1][1].endswith('; Domain=localhost; Path=/'))
- self.assertTrue(result[1][1].startswith('auth_tkt='))
-
- self.assertEqual(result[2][0], 'Set-Cookie')
- self.assertTrue(result[2][1].endswith('; Domain=.localhost; Path=/'))
- self.assertTrue(result[2][1].startswith('auth_tkt='))
-
- def test_remember_include_ip(self):
- helper = self._makeOne('secret', include_ip=True)
- request = self._makeRequest()
- result = helper.remember(request, 'other')
- self.assertEqual(len(result), 3)
-
- self.assertEqual(result[0][0], 'Set-Cookie')
- self.assertTrue(result[0][1].endswith('; Path=/; SameSite=Lax'))
- self.assertTrue(result[0][1].startswith('auth_tkt='))
-
- self.assertEqual(result[1][0], 'Set-Cookie')
- self.assertTrue(
- result[1][1].endswith('; Domain=localhost; Path=/; SameSite=Lax')
- )
- self.assertTrue(result[1][1].startswith('auth_tkt='))
-
- self.assertEqual(result[2][0], 'Set-Cookie')
- self.assertTrue(
- result[2][1].endswith('; Domain=.localhost; Path=/; SameSite=Lax')
- )
- self.assertTrue(result[2][1].startswith('auth_tkt='))
-
- def test_remember_path(self):
- helper = self._makeOne(
- 'secret', include_ip=True, path="/cgi-bin/app.cgi/"
- )
- request = self._makeRequest()
- result = helper.remember(request, 'other')
- self.assertEqual(len(result), 3)
-
- self.assertEqual(result[0][0], 'Set-Cookie')
- self.assertTrue(
- result[0][1].endswith('; Path=/cgi-bin/app.cgi/; SameSite=Lax')
- )
- self.assertTrue(result[0][1].startswith('auth_tkt='))
-
- self.assertEqual(result[1][0], 'Set-Cookie')
- self.assertTrue(
- result[1][1].endswith(
- '; Domain=localhost; Path=/cgi-bin/app.cgi/; SameSite=Lax'
- )
- )
- self.assertTrue(result[1][1].startswith('auth_tkt='))
-
- self.assertEqual(result[2][0], 'Set-Cookie')
- self.assertTrue(
- result[2][1].endswith(
- '; Domain=.localhost; Path=/cgi-bin/app.cgi/; SameSite=Lax'
- )
- )
- self.assertTrue(result[2][1].startswith('auth_tkt='))
-
- def test_remember_http_only(self):
- helper = self._makeOne('secret', include_ip=True, http_only=True)
- request = self._makeRequest()
- result = helper.remember(request, 'other')
- self.assertEqual(len(result), 3)
-
- self.assertEqual(result[0][0], 'Set-Cookie')
- self.assertTrue(result[0][1].endswith('; HttpOnly; SameSite=Lax'))
- self.assertTrue(result[0][1].startswith('auth_tkt='))
-
- self.assertEqual(result[1][0], 'Set-Cookie')
- self.assertTrue('; HttpOnly' in result[1][1])
- self.assertTrue(result[1][1].startswith('auth_tkt='))
-
- self.assertEqual(result[2][0], 'Set-Cookie')
- self.assertTrue('; HttpOnly' in result[2][1])
- self.assertTrue(result[2][1].startswith('auth_tkt='))
-
- def test_remember_secure(self):
- helper = self._makeOne('secret', include_ip=True, secure=True)
- request = self._makeRequest()
- result = helper.remember(request, 'other')
- self.assertEqual(len(result), 3)
-
- self.assertEqual(result[0][0], 'Set-Cookie')
- self.assertTrue('; secure' in result[0][1])
- self.assertTrue(result[0][1].startswith('auth_tkt='))
-
- self.assertEqual(result[1][0], 'Set-Cookie')
- self.assertTrue('; secure' in result[1][1])
- self.assertTrue(result[1][1].startswith('auth_tkt='))
-
- self.assertEqual(result[2][0], 'Set-Cookie')
- self.assertTrue('; secure' in result[2][1])
- self.assertTrue(result[2][1].startswith('auth_tkt='))
-
- def test_remember_wild_domain_disabled(self):
- helper = self._makeOne('secret', wild_domain=False)
- request = self._makeRequest()
- result = helper.remember(request, 'other')
- self.assertEqual(len(result), 2)
-
- self.assertEqual(result[0][0], 'Set-Cookie')
- self.assertTrue(result[0][1].endswith('; Path=/; SameSite=Lax'))
- self.assertTrue(result[0][1].startswith('auth_tkt='))
-
- self.assertEqual(result[1][0], 'Set-Cookie')
- self.assertTrue(
- result[1][1].endswith('; Domain=localhost; Path=/; SameSite=Lax')
- )
- self.assertTrue(result[1][1].startswith('auth_tkt='))
-
- def test_remember_parent_domain(self):
- helper = self._makeOne('secret', parent_domain=True)
- request = self._makeRequest()
- request.domain = 'www.example.com'
- result = helper.remember(request, 'other')
- self.assertEqual(len(result), 1)
-
- self.assertEqual(result[0][0], 'Set-Cookie')
- self.assertTrue(
- result[0][1].endswith(
- '; Domain=.example.com; Path=/; SameSite=Lax'
- )
- )
- self.assertTrue(result[0][1].startswith('auth_tkt='))
-
- def test_remember_parent_domain_supercedes_wild_domain(self):
- helper = self._makeOne('secret', parent_domain=True, wild_domain=True)
- request = self._makeRequest()
- request.domain = 'www.example.com'
- result = helper.remember(request, 'other')
- self.assertEqual(len(result), 1)
- self.assertTrue(
- result[0][1].endswith(
- '; Domain=.example.com; Path=/; SameSite=Lax'
- )
- )
-
- def test_remember_explicit_domain(self):
- helper = self._makeOne('secret', domain='pyramid.bazinga')
- request = self._makeRequest()
- request.domain = 'www.example.com'
- result = helper.remember(request, 'other')
- self.assertEqual(len(result), 1)
-
- self.assertEqual(result[0][0], 'Set-Cookie')
- self.assertTrue(
- result[0][1].endswith(
- '; Domain=pyramid.bazinga; Path=/; SameSite=Lax'
- )
- )
- self.assertTrue(result[0][1].startswith('auth_tkt='))
-
- def test_remember_domain_supercedes_parent_and_wild_domain(self):
- helper = self._makeOne(
- 'secret',
- domain='pyramid.bazinga',
- parent_domain=True,
- wild_domain=True,
- )
- request = self._makeRequest()
- request.domain = 'www.example.com'
- result = helper.remember(request, 'other')
- self.assertEqual(len(result), 1)
- self.assertTrue(
- result[0][1].endswith(
- '; Domain=pyramid.bazinga; Path=/; SameSite=Lax'
- )
- )
-
- def test_remember_binary_userid(self):
- import base64
-
- helper = self._makeOne('secret')
- request = self._makeRequest()
- result = helper.remember(request, b'userid')
- values = self._parseHeaders(result)
- self.assertEqual(len(result), 3)
- val = self._cookieValue(values[0])
- self.assertEqual(
- val['userid'], text_(base64.b64encode(b'userid').strip())
- )
- self.assertEqual(val['user_data'], 'userid_type:b64str')
-
- def test_remember_int_userid(self):
- helper = self._makeOne('secret')
- request = self._makeRequest()
- result = helper.remember(request, 1)
- values = self._parseHeaders(result)
- self.assertEqual(len(result), 3)
- val = self._cookieValue(values[0])
- self.assertEqual(val['userid'], '1')
- self.assertEqual(val['user_data'], 'userid_type:int')
-
- def test_remember_unicode_userid(self):
- import base64
-
- helper = self._makeOne('secret')
- request = self._makeRequest()
- userid = text_(b'\xc2\xa9', 'utf-8')
- result = helper.remember(request, userid)
- values = self._parseHeaders(result)
- self.assertEqual(len(result), 3)
- val = self._cookieValue(values[0])
- self.assertEqual(
- val['userid'], text_(base64.b64encode(userid.encode('utf-8')))
- )
- self.assertEqual(val['user_data'], 'userid_type:b64unicode')
-
- def test_remember_insane_userid(self):
- helper = self._makeOne('secret')
- request = self._makeRequest()
- userid = object()
- with warnings.catch_warnings(record=True) as w:
- warnings.simplefilter('always', RuntimeWarning)
- result = helper.remember(request, userid)
- self.assertTrue(str(w[-1].message).startswith('userid is of type'))
- values = self._parseHeaders(result)
- self.assertEqual(len(result), 3)
- value = values[0]
- self.assertTrue('userid' in value.value)
-
- def test_remember_max_age(self):
- helper = self._makeOne('secret')
- request = self._makeRequest()
- result = helper.remember(request, 'userid', max_age=500)
- values = self._parseHeaders(result)
- self.assertEqual(len(result), 3)
-
- self.assertEqual(values[0]['max-age'], '500')
- self.assertTrue(values[0]['expires'])
-
- def test_remember_str_max_age(self):
- helper = self._makeOne('secret')
- request = self._makeRequest()
- result = helper.remember(request, 'userid', max_age='500')
- values = self._parseHeaders(result)
- self.assertEqual(len(result), 3)
-
- self.assertEqual(values[0]['max-age'], '500')
- self.assertTrue(values[0]['expires'])
-
- def test_remember_str_max_age_invalid(self):
- helper = self._makeOne('secret')
- request = self._makeRequest()
- self.assertRaises(
- ValueError,
- helper.remember,
- request,
- 'userid',
- max_age='invalid value',
- )
-
- def test_remember_tokens(self):
- helper = self._makeOne('secret')
- request = self._makeRequest()
- result = helper.remember(request, 'other', tokens=('foo', 'bar'))
- self.assertEqual(len(result), 3)
-
- self.assertEqual(result[0][0], 'Set-Cookie')
- self.assertTrue("/tokens=foo|bar/" in result[0][1])
-
- self.assertEqual(result[1][0], 'Set-Cookie')
- self.assertTrue("/tokens=foo|bar/" in result[1][1])
-
- self.assertEqual(result[2][0], 'Set-Cookie')
- self.assertTrue("/tokens=foo|bar/" in result[2][1])
-
- def test_remember_samesite_nondefault(self):
- helper = self._makeOne('secret', samesite='Strict')
- request = self._makeRequest()
- result = helper.remember(request, 'userid')
- self.assertEqual(len(result), 3)
-
- self.assertEqual(result[0][0], 'Set-Cookie')
- cookieval = result[0][1]
- self.assertTrue(
- 'SameSite=Strict' in [x.strip() for x in cookieval.split(';')],
- cookieval,
- )
-
- self.assertEqual(result[1][0], 'Set-Cookie')
- cookieval = result[1][1]
- self.assertTrue(
- 'SameSite=Strict' in [x.strip() for x in cookieval.split(';')],
- cookieval,
- )
-
- self.assertEqual(result[2][0], 'Set-Cookie')
- cookieval = result[2][1]
- self.assertTrue(
- 'SameSite=Strict' in [x.strip() for x in cookieval.split(';')],
- cookieval,
- )
-
- def test_remember_samesite_default(self):
- helper = self._makeOne('secret')
- request = self._makeRequest()
- result = helper.remember(request, 'userid')
- self.assertEqual(len(result), 3)
-
- self.assertEqual(result[0][0], 'Set-Cookie')
- cookieval = result[0][1]
- self.assertTrue(
- 'SameSite=Lax' in [x.strip() for x in cookieval.split(';')],
- cookieval,
- )
-
- self.assertEqual(result[1][0], 'Set-Cookie')
- cookieval = result[1][1]
- self.assertTrue(
- 'SameSite=Lax' in [x.strip() for x in cookieval.split(';')],
- cookieval,
- )
-
- self.assertEqual(result[2][0], 'Set-Cookie')
- cookieval = result[2][1]
- self.assertTrue(
- 'SameSite=Lax' in [x.strip() for x in cookieval.split(';')],
- cookieval,
- )
-
- def test_remember_unicode_but_ascii_token(self):
- helper = self._makeOne('secret')
- request = self._makeRequest()
- la = text_(b'foo', 'utf-8')
- result = helper.remember(request, 'other', tokens=(la,))
- # tokens must be str type on both Python 2 and 3
- self.assertTrue("/tokens=foo/" in result[0][1])
-
- def test_remember_nonascii_token(self):
- helper = self._makeOne('secret')
- request = self._makeRequest()
- la = text_(b'La Pe\xc3\xb1a', 'utf-8')
- self.assertRaises(
- ValueError, helper.remember, request, 'other', tokens=(la,)
- )
-
- def test_remember_invalid_token_format(self):
- helper = self._makeOne('secret')
- request = self._makeRequest()
- self.assertRaises(
- ValueError, helper.remember, request, 'other', tokens=('foo bar',)
- )
- self.assertRaises(
- ValueError, helper.remember, request, 'other', tokens=('1bar',)
- )
-
- def test_forget(self):
- helper = self._makeOne('secret')
- request = self._makeRequest()
- headers = helper.forget(request)
- self.assertEqual(len(headers), 3)
- name, value = headers[0]
- self.assertEqual(name, 'Set-Cookie')
- self.assertEqual(
- value,
- 'auth_tkt=; Max-Age=0; Path=/; '
- 'expires=Wed, 31-Dec-97 23:59:59 GMT; SameSite=Lax',
- )
- name, value = headers[1]
- self.assertEqual(name, 'Set-Cookie')
- self.assertEqual(
- value,
- 'auth_tkt=; Domain=localhost; Max-Age=0; Path=/; '
- 'expires=Wed, 31-Dec-97 23:59:59 GMT; SameSite=Lax',
- )
- name, value = headers[2]
- self.assertEqual(name, 'Set-Cookie')
- self.assertEqual(
- value,
- 'auth_tkt=; Domain=.localhost; Max-Age=0; Path=/; '
- 'expires=Wed, 31-Dec-97 23:59:59 GMT; SameSite=Lax',
- )
-
-
-class DummyAuthTktModule(object):
- def __init__(
- self,
- timestamp=0,
- userid='userid',
- tokens=(),
- user_data='',
- parse_raise=False,
- hashalg="md5",
- ):
- self.timestamp = timestamp
- self.userid = userid
- self.tokens = tokens
- self.user_data = user_data
- self.parse_raise = parse_raise
- self.hashalg = hashalg
-
- def parse_ticket(secret, value, remote_addr, hashalg):
- self.secret = secret
- self.value = value
- self.remote_addr = remote_addr
- if self.parse_raise:
- raise self.BadTicket()
- return self.timestamp, self.userid, self.tokens, self.user_data
-
- self.parse_ticket = parse_ticket
-
- class AuthTicket(object):
- def __init__(self, secret, userid, remote_addr, **kw):
- self.secret = secret
- self.userid = userid
- self.remote_addr = remote_addr
- self.kw = kw
-
- def cookie_value(self):
- result = {
- 'secret': self.secret,
- 'userid': self.userid,
- 'remote_addr': self.remote_addr,
- }
- result.update(self.kw)
- tokens = result.pop('tokens', None)
- if tokens is not None:
- tokens = '|'.join(tokens)
- result['tokens'] = tokens
- items = sorted(result.items())
- new_items = []
- for k, v in items:
- if isinstance(v, bytes):
- v = text_(v)
- new_items.append((k, v))
- result = '/'.join(['%s=%s' % (k, v) for k, v in new_items])
- return result
-
- self.AuthTicket = AuthTicket
-
- class BadTicket(Exception):
- pass
-
-
-class DummyCookies(object):
- def __init__(self, cookie):
- self.cookie = cookie
-
- def get(self, name):
- return self.cookie
-
-
-class DummyRequest:
- domain = 'localhost'
-
- def __init__(self, environ=None, session=None, registry=None, cookie=None):
- self.environ = environ or {}
- self.session = session or {}
- self.registry = registry
- self.callbacks = []
- self.cookies = DummyCookies(cookie)
-
- def add_response_callback(self, callback):
- self.callbacks.append(callback)
-
-
-class DummyResponse:
- def __init__(self):
- self.headerlist = []