class Test(unittest.TestCase):
def setUp(self):
setUpPasswordManagers()
self.reg = PrincipalRegistry()
self.reg.definePrincipal('1', 'Tim Peters', 'Sir Tim Peters',
'tim', '123')
self.reg.definePrincipal('2', 'Jim Fulton', 'Sir Jim Fulton',
'jim', '456')
def testRegistered(self):
p = self.reg.getPrincipal('1')
self.assertEqual(p.id, '1')
self.assertEqual(p.title, 'Tim Peters')
self.assertEqual(p.description, 'Sir Tim Peters')
p = self.reg.getPrincipal('2')
self.assertEqual(p.id, '2')
self.assertEqual(p.title, 'Jim Fulton')
self.assertEqual(p.description, 'Sir Jim Fulton')
self.assertEqual(len(self.reg.getPrincipals('')), 2)
def testUnRegistered(self):
self.assertRaises(PrincipalLookupError, self.reg.getPrincipal, '3')
def testDup(self):
self.assertRaises(DuplicateId,
self.reg.definePrincipal,
'1', 'Tim Peters', 'Sir Tim Peters',
'tim2', '123')
self.assertRaises(DuplicateLogin,
self.reg.definePrincipal,
'3', 'Tim Peters', 'Sir Tim Peters',
'tim', '123')
self.assertRaises(PrincipalLookupError, self.reg.getPrincipal, '3')
self.assertEqual(len(self.reg.getPrincipals('')), 2)
def testSearch(self):
r = self.reg.getPrincipals('J')
self.assertEquals(len(r), 1)
self.failUnless(r[0] is self.reg.getPrincipal('2'))
def testByLogin(self):
tim = self.reg.getPrincipalByLogin('tim')
self.assertEquals(tim.getLogin(), 'tim')
jim = self.reg.getPrincipalByLogin('jim')
self.assertEquals(jim.getLogin(), 'jim')
self.assertRaises(KeyError,
self.reg.getPrincipalByLogin, 'kim')
def testValidation(self):
tim = self.reg.getPrincipalByLogin('tim')
self.assert_(tim.validate('123'))
self.failIf(tim.validate('456'))
self.failIf(tim.validate(''))
self.failIf(tim.validate('1234'))
self.failIf(tim.validate('12'))
def testAuthenticate(self):
req = Request(('tim', '123'))
pid = self.reg.authenticate(req).id
self.assertEquals(pid, '1')
req = Request(('tim', '1234'))
p = self.reg.authenticate(req)
self.assertEquals(p, None)
req = Request(('kim', '123'))
p = self.reg.authenticate(req)
self.assertEquals(p, None)
def testUnauthorized(self):
request = Request(None)
self.reg.unauthorized(self.reg.unauthenticatedPrincipal(), request)
self.assertEquals(request.challenge, 'basic realm="Zope"')
request = Request(None)
self.reg.unauthorized(None, request)
self.assertEquals(request.challenge, 'basic realm="Zope"')
request = Request(None)
self.reg.unauthorized("1", request)
self.assertEquals(request.challenge, None)
def testDefaultPrincipal(self):
self.assertEquals(self.reg.unauthenticatedPrincipal(), None)
self.assertRaises(DuplicateId, self.reg.defineDefaultPrincipal,
"1", "tim")
self.reg.defineDefaultPrincipal("everybody", "Default Principal")
self.assertEquals(self.reg.unauthenticatedPrincipal().id, "everybody")
self.reg.defineDefaultPrincipal("anybody", "Default Principal",
"This is the default headmaster")
self.assertEquals(self.reg.unauthenticatedPrincipal().id, "anybody")
self.assertRaises(PrincipalLookupError,
self.reg.getPrincipal, "everybody")
p = self.reg.getPrincipal("anybody")
self.assertEquals(p.id, "anybody")
self.assertEquals(p.title, "Default Principal")
self.assertRaises(DuplicateId, self.reg.definePrincipal,
"anybody", "title")
class Test(unittest.TestCase):
def setUp(self):
setUpPasswordManagers()
self.reg = PrincipalRegistry()
# Passwords are provided in encoded form, which means they must be
# bytes.
self.reg.definePrincipal('1', 'Tim Peters', 'Sir Tim Peters',
'tim', b'123')
self.reg.definePrincipal('2', 'Jim Fulton', 'Sir Jim Fulton',
'jim', b'456')
def testRegistered(self):
p = self.reg.getPrincipal('1')
self.assertEqual(p.id, '1')
self.assertEqual(p.title, 'Tim Peters')
self.assertEqual(p.description, 'Sir Tim Peters')
p = self.reg.getPrincipal('2')
self.assertEqual(p.id, '2')
self.assertEqual(p.title, 'Jim Fulton')
self.assertEqual(p.description, 'Sir Jim Fulton')
self.assertEqual(len(self.reg.getPrincipals('')), 2)
def testUnRegistered(self):
self.assertRaises(PrincipalLookupError, self.reg.getPrincipal, '3')
def testDupPrincipal(self):
self.assertRaises(DuplicateId,
self.reg.definePrincipal,
'1', 'Tim Peters', 'Sir Tim Peters',
'tim2', b'123')
self.assertRaises(DuplicateLogin,
self.reg.definePrincipal,
'3', 'Tim Peters', 'Sir Tim Peters',
'tim', b'123')
self.assertRaises(PrincipalLookupError, self.reg.getPrincipal, '3')
self.assertEqual(len(self.reg.getPrincipals('')), 2)
def testSearch(self):
r = self.reg.getPrincipals('J')
self.assertEqual(len(r), 1)
self.assertTrue(r[0] is self.reg.getPrincipal('2'))
def testByLogin(self):
tim = self.reg.getPrincipalByLogin('tim')
self.assertEqual(tim.getLogin(), 'tim')
jim = self.reg.getPrincipalByLogin('jim')
self.assertEqual(jim.getLogin(), 'jim')
self.assertRaises(KeyError,
self.reg.getPrincipalByLogin, 'kim')
def testValidation(self):
tim = self.reg.getPrincipalByLogin('tim')
self.assertTrue(tim.validate('123'))
self.assertFalse(tim.validate('456'))
self.assertFalse(tim.validate(''))
self.assertFalse(tim.validate('1234'))
self.assertFalse(tim.validate('12'))
def _validatePrincipal(self, principal):
from zope.security.interfaces import IPrincipal
from zope.interface.verify import verifyObject
from zope.schema import getValidationErrors
self.assertTrue(verifyObject(IPrincipal, principal))
errors = getValidationErrors(IPrincipal, principal)
self.assertEqual([], errors)
def testSchemaValidation(self):
# Registered objects comply with IPrincipal
tim = self.reg.getPrincipalByLogin('tim')
self._validatePrincipal(tim)
def testAuthenticate(self):
req = Request((b'tim', b'123'))
pid = self.reg.authenticate(req).id
self.assertEqual(pid, '1')
req = Request((b'tim', b'1234'))
p = self.reg.authenticate(req)
self.assertEqual(p, None)
req = Request((b'kim', b'123'))
p = self.reg.authenticate(req)
self.assertEqual(p, None)
def testUnauthorized(self):
request = Request(None)
self.reg.unauthorized(self.reg.unauthenticatedPrincipal(), request)
self.assertEqual(request.challenge, 'basic realm="Zope"')
request = Request(None)
self.reg.unauthorized(None, request)
self.assertEqual(request.challenge, 'basic realm="Zope"')
request = Request(None)
self.reg.unauthorized("1", request)
self.assertEqual(request.challenge, None)
def testDefaultPrincipal(self):
self.assertEqual(self.reg.unauthenticatedPrincipal(), None)
self.assertRaises(DuplicateId, self.reg.defineDefaultPrincipal,
"1", "tim")
self.reg.defineDefaultPrincipal("everybody", "Default Principal")
self.assertEqual(self.reg.unauthenticatedPrincipal().id, "everybody")
anybody = self.reg.defineDefaultPrincipal("anybody",
"Default Principal",
"This is the default headmaster")
self.assertIs(anybody, self.reg.unauthenticatedPrincipal())
self.assertEqual(anybody.id, "anybody")
self.assertRaises(PrincipalLookupError,
self.reg.getPrincipal, "everybody")
p = self.reg.getPrincipal("anybody")
self.assertIs(p, anybody)
self.assertEqual(p.id, "anybody")
self.assertEqual(p.title, "Default Principal")
self.assertRaises(DuplicateId, self.reg.definePrincipal,
"anybody", "title")
# It complies with IPrincipal
self._validatePrincipal(anybody)
def test_logout(self):
self.assertIsNone(self.reg.logout(None))
def test_duplicate_group(self):
class Group(object):
id = "id"
self.reg.registerGroup(Group)
with self.assertRaises(DuplicateId):
self.reg.registerGroup(Group)