1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33
|
import os
import unittest
# Switch off processing .ldaprc or ldap.conf before importing _ldap
os.environ['LDAPNOINIT'] = '1'
from ldap.controls import ppolicy
PP_GRACEAUTH = b'0\x84\x00\x00\x00\t\xa0\x84\x00\x00\x00\x03\x81\x01\x02'
PP_TIMEBEFORE = b'0\x84\x00\x00\x00\t\xa0\x84\x00\x00\x00\x03\x80\x012'
class TestControlsPPolicy(unittest.TestCase):
def assertPPolicy(self, pp, timeBeforeExpiration=None,
graceAuthNsRemaining=None, error=None):
self.assertEqual(pp.timeBeforeExpiration, timeBeforeExpiration)
self.assertEqual(pp.graceAuthNsRemaining, graceAuthNsRemaining)
self.assertEqual(pp.error, error)
def test_ppolicy_graceauth(self):
pp = ppolicy.PasswordPolicyControl()
pp.decodeControlValue(PP_GRACEAUTH)
self.assertPPolicy(pp, graceAuthNsRemaining=2)
def test_ppolicy_timebefore(self):
pp = ppolicy.PasswordPolicyControl()
pp.decodeControlValue(PP_TIMEBEFORE)
self.assertPPolicy(pp, timeBeforeExpiration=50)
if __name__ == '__main__':
unittest.main()
|