Skip to content

Commit

Permalink
Test coverage.
Browse files Browse the repository at this point in the history
  • Loading branch information
Chris Rossi committed Oct 12, 2012
1 parent 201596a commit bf902d4
Show file tree
Hide file tree
Showing 2 changed files with 88 additions and 14 deletions.
2 changes: 2 additions & 0 deletions .gitignore
View file
Original file line number Diff line number Diff line change
Expand Up @@ -5,6 +5,7 @@
*.pt.py
*.txt.py
*~
.*.swp
.coverage
.tox/
nosetests.xml
Expand All @@ -21,3 +22,4 @@ bookenv/
jyenv/
pypyenv/
env*/
venv/
100 changes: 86 additions & 14 deletions pyramid/tests/test_authentication.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -14,7 +14,7 @@ def setUp(self):

def tearDown(self):
del self.config

def debug(self, msg):
self.messages.append(msg)

Expand Down Expand Up @@ -151,7 +151,7 @@ def _getTargetClass(self):

def _makeOne(self, identifier_name='auth_tkt', callback=None):
return self._getTargetClass()(identifier_name, callback)

def test_class_implements_IAuthenticationPolicy(self):
from zope.interface.verify import verifyClass
from pyramid.interfaces import IAuthenticationPolicy
Expand Down Expand Up @@ -251,7 +251,7 @@ def test_remember(self):
result = policy.remember(request, 'fred')
self.assertEqual(result[0], request.environ)
self.assertEqual(result[1], {'repoze.who.userid':'fred'})

def test_forget_no_plugins(self):
request = DummyRequest({})
policy = self._makeOne()
Expand All @@ -276,7 +276,7 @@ def _getTargetClass(self):

def _makeOne(self, environ_key='REMOTE_USER', callback=None):
return self._getTargetClass()(environ_key, callback)

def test_class_implements_IAuthenticationPolicy(self):
from zope.interface.verify import verifyClass
from pyramid.interfaces import IAuthenticationPolicy
Expand All @@ -301,7 +301,7 @@ def test_authenticated_userid_None(self):
request = DummyRequest({})
policy = self._makeOne()
self.assertEqual(policy.authenticated_userid(request), None)

def test_authenticated_userid(self):
request = DummyRequest({'REMOTE_USER':'fred'})
policy = self._makeOne()
Expand All @@ -326,7 +326,7 @@ def test_remember(self):
policy = self._makeOne()
result = policy.remember(request, 'fred')
self.assertEqual(result, [])

def test_forget(self):
request = DummyRequest({'REMOTE_USER':'fred'})
policy = self._makeOne()
Expand Down Expand Up @@ -375,7 +375,7 @@ def test_authenticated_userid_no_cookie_identity(self):
request = DummyRequest({})
policy = self._makeOne(None, None)
self.assertEqual(policy.authenticated_userid(request), None)

def test_authenticated_userid_callback_returns_None(self):
request = DummyRequest({})
def callback(userid, request):
Expand Down Expand Up @@ -426,7 +426,7 @@ def test_remember_with_extra_kargs(self):
result = policy.remember(request, 'fred', a=1, b=2)
self.assertEqual(policy.cookie.kw, {'a':1, 'b':2})
self.assertEqual(result, [])

def test_forget(self):
request = DummyRequest({})
policy = self._makeOne(None, None)
Expand Down Expand Up @@ -482,7 +482,7 @@ def test_identify_cookie_value_is_None(self):
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')
Expand Down Expand Up @@ -605,7 +605,7 @@ def test_identify_bad_cookie(self):
request = self._makeRequest('ticket')
result = helper.identify(request)
self.assertEqual(result, None)

def test_identify_cookie_timed_out(self):
helper = self._makeOne('secret', timeout=1)
request = self._makeRequest({'HTTP_COOKIE':'auth_tkt=bogus'})
Expand Down Expand Up @@ -828,7 +828,7 @@ def test_remember_domain_has_port(self):
self.assertEqual(result[1][0], 'Set-Cookie')
self.assertTrue(result[1][1].endswith('; Path=/; Domain=example.com'))
self.assertTrue(result[1][1].startswith('auth_tkt='))

def test_remember_binary_userid(self):
import base64
helper = self._makeOne('secret')
Expand Down Expand Up @@ -1106,6 +1106,78 @@ def test_forget_no_identity(self):
self.assertEqual(request.session.get('userid'), None)
self.assertEqual(result, [])

class TestBasicAuthAuthenticationPolicy(unittest.TestCase):
def _getTargetClass(self):
from pyramid.authentication import BasicAuthAuthenticationPolicy as cls
return cls

def _makeOne(self, check):
return self._getTargetClass()(check, realm='SomeRealm')

def test_class_implements_IAuthenticationPolicy(self):
from zope.interface.verify import verifyClass
from pyramid.interfaces import IAuthenticationPolicy
verifyClass(IAuthenticationPolicy, self._getTargetClass())

def test_unauthenticated_userid(self):
import base64
request = testing.DummyRequest()
request.headers['Authorization'] = 'Basic %s' % base64.b64encode(
'chrisr:password')
policy = self._makeOne(None)
self.assertEqual(policy.unauthenticated_userid(request), 'chrisr')

def test_unauthenticated_userid_no_credentials(self):
request = testing.DummyRequest()
policy = self._makeOne(None)
self.assertEqual(policy.unauthenticated_userid(request), None)

def test_unauthenticated_bad_header(self):
request = testing.DummyRequest()
request.headers['Authorization'] = '...'
policy = self._makeOne(None)
self.assertEqual(policy.unauthenticated_userid(request), None)

def test_unauthenticated_userid_not_basic(self):
request = testing.DummyRequest()
request.headers['Authorization'] = 'Complicated things'
policy = self._makeOne(None)
self.assertEqual(policy.unauthenticated_userid(request), None)

def test_unauthenticated_userid_corrupt_base64(self):
request = testing.DummyRequest()
request.headers['Authorization'] = 'Basic chrisr:password'
policy = self._makeOne(None)
self.assertEqual(policy.unauthenticated_userid(request), None)

def test_authenticated_userid(self):
import base64
request = testing.DummyRequest()
request.headers['Authorization'] = 'Basic %s' % base64.b64encode(
'chrisr:password')
def check(username, password, request):
return []
policy = self._makeOne(check)
self.assertEqual(policy.authenticated_userid(request), 'chrisr')

def test_unauthenticated_userid_invalid_payload(self):
import base64
request = testing.DummyRequest()
request.headers['Authorization'] = 'Basic %s' % base64.b64encode(
'chrisrpassword')
policy = self._makeOne(None)
self.assertEqual(policy.unauthenticated_userid(request), None)

def test_remember(self):
policy = self._makeOne(None)
self.assertEqual(policy.remember(None, None), [])

def test_forget(self):
policy = self._makeOne(None)
self.assertEqual(policy.forget(None), [
('WWW-Authenticate', 'Basic realm="SomeRealm"')])


class DummyContext:
pass

Expand All @@ -1130,7 +1202,7 @@ def add_response_callback(self, callback):
class DummyWhoPlugin:
def remember(self, environ, identity):
return environ, identity

def forget(self, environ, identity):
return environ, identity

Expand Down Expand Up @@ -1164,7 +1236,7 @@ def parse_ticket(secret, value, remote_addr):
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
Expand All @@ -1186,4 +1258,4 @@ class BadTicket(Exception):
class DummyResponse:
def __init__(self):
self.headerlist = []

0 comments on commit bf902d4

Please sign in to comment.