pollutri
/
res_levis_API
1
0
複製 0
程式碼 問題 0 合併請求 0 版本發佈 0 Wiki 活動
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
5 提交
1 分支
203 MiB
 
 
 
 
分支: master
分支列表 標籤
${ item.name }
建立分支 ${ searchTerm }
從 'master'
${ noResults }
res_levis_API/lib/python3.6/site-packages/passlib/tests/test_apache.py

652 line
25 KiB
原始文件 永久連結 Blame 歷史記錄 

  1. """tests for passlib.apache -- (c) Assurance Technologies 2008-2011"""

  2. #=============================================================================

  3. # imports

  4. #=============================================================================

  5. from __future__ import with_statement

  6. # core

  7. from logging import getLogger

  8. import os

  9. # site

  10. # pkg

  11. from passlib import apache

  12. from passlib.exc import MissingBackendError

  13. from passlib.utils.compat import irange

  14. from passlib.tests.utils import TestCase, get_file, set_file, ensure_mtime_changed

  15. from passlib.utils.compat import u

  16. from passlib.utils import to_bytes

  17. # module

  18. log = getLogger(__name__)

  19. 

  20. def backdate_file_mtime(path, offset=10):

  21.     """backdate file's mtime by specified amount"""

  22.     # NOTE: this is used so we can test code which detects mtime changes,

  23.     #       without having to actually *pause* for that long.

  24.     atime = os.path.getatime(path)

  25.     mtime = os.path.getmtime(path)-offset

  26.     os.utime(path, (atime, mtime))

  27. 

  28. #=============================================================================

  29. # htpasswd

  30. #=============================================================================

  31. class HtpasswdFileTest(TestCase):

  32.     """test HtpasswdFile class"""

  33.     descriptionPrefix = "HtpasswdFile"

  34. 

  35.     # sample with 4 users

  36.     sample_01 = (b'user2:2CHkkwa2AtqGs\n'

  37.                  b'user3:{SHA}3ipNV1GrBtxPmHFC21fCbVCSXIo=\n'

  38.                  b'user4:pass4\n'

  39.                  b'user1:$apr1$t4tc7jTh$GPIWVUo8sQKJlUdV8V5vu0\n')

  40. 

  41.     # sample 1 with user 1, 2 deleted; 4 changed

  42.     sample_02 = b'user3:{SHA}3ipNV1GrBtxPmHFC21fCbVCSXIo=\nuser4:pass4\n'

  43. 

  44.     # sample 1 with user2 updated, user 1 first entry removed, and user 5 added

  45.     sample_03 = (b'user2:pass2x\n'

  46.                  b'user3:{SHA}3ipNV1GrBtxPmHFC21fCbVCSXIo=\n'

  47.                  b'user4:pass4\n'

  48.                  b'user1:$apr1$t4tc7jTh$GPIWVUo8sQKJlUdV8V5vu0\n'

  49.                  b'user5:pass5\n')

  50. 

  51.     # standalone sample with 8-bit username

  52.     sample_04_utf8 = b'user\xc3\xa6:2CHkkwa2AtqGs\n'

  53.     sample_04_latin1 = b'user\xe6:2CHkkwa2AtqGs\n'

  54. 

  55.     sample_dup = b'user1:pass1\nuser1:pass2\n'

  56. 

  57.     # sample with bcrypt & sha256_crypt hashes

  58.     sample_05 = (b'user2:2CHkkwa2AtqGs\n'

  59.                  b'user3:{SHA}3ipNV1GrBtxPmHFC21fCbVCSXIo=\n'

  60.                  b'user4:pass4\n'

  61.                  b'user1:$apr1$t4tc7jTh$GPIWVUo8sQKJlUdV8V5vu0\n'

  62.                  b'user5:$2a$12$yktDxraxijBZ360orOyCOePFGhuis/umyPNJoL5EbsLk.s6SWdrRO\n'

  63.                  b'user6:$5$rounds=110000$cCRp/xUUGVgwR4aP$'

  64.                      b'p0.QKFS5qLNRqw1/47lXYiAcgIjJK.WjCO8nrEKuUK.\n')

  65. 

  66.     def test_00_constructor_autoload(self):

  67.         """test constructor autoload"""

  68.         # check with existing file

  69.         path = self.mktemp()

  70.         set_file(path, self.sample_01)

  71.         ht = apache.HtpasswdFile(path)

  72.         self.assertEqual(ht.to_string(), self.sample_01)

  73.         self.assertEqual(ht.path, path)

  74.         self.assertTrue(ht.mtime)

  75. 

  76.         # check changing path

  77.         ht.path = path + "x"

  78.         self.assertEqual(ht.path, path + "x")

  79.         self.assertFalse(ht.mtime)

  80. 

  81.         # check new=True

  82.         ht = apache.HtpasswdFile(path, new=True)

  83.         self.assertEqual(ht.to_string(), b"")

  84.         self.assertEqual(ht.path, path)

  85.         self.assertFalse(ht.mtime)

  86. 

  87.         # check autoload=False (deprecated alias for new=True)

  88.         with self.assertWarningList("``autoload=False`` is deprecated"):

  89.             ht = apache.HtpasswdFile(path, autoload=False)

  90.         self.assertEqual(ht.to_string(), b"")

  91.         self.assertEqual(ht.path, path)

  92.         self.assertFalse(ht.mtime)

  93. 

  94.         # check missing file

  95.         os.remove(path)

  96.         self.assertRaises(IOError, apache.HtpasswdFile, path)

  97. 

  98.         # NOTE: "default_scheme" option checked via set_password() test, among others

  99. 

  100.     def test_00_from_path(self):

  101.         path = self.mktemp()

  102.         set_file(path, self.sample_01)

  103.         ht = apache.HtpasswdFile.from_path(path)

  104.         self.assertEqual(ht.to_string(), self.sample_01)

  105.         self.assertEqual(ht.path, None)

  106.         self.assertFalse(ht.mtime)

  107. 

  108.     def test_01_delete(self):

  109.         """test delete()"""

  110.         ht = apache.HtpasswdFile.from_string(self.sample_01)

  111.         self.assertTrue(ht.delete("user1")) # should delete both entries

  112.         self.assertTrue(ht.delete("user2"))

  113.         self.assertFalse(ht.delete("user5")) # user not present

  114.         self.assertEqual(ht.to_string(), self.sample_02)

  115. 

  116.         # invalid user

  117.         self.assertRaises(ValueError, ht.delete, "user:")

  118. 

  119.     def test_01_delete_autosave(self):

  120.         path = self.mktemp()

  121.         sample = b'user1:pass1\nuser2:pass2\n'

  122.         set_file(path, sample)

  123. 

  124.         ht = apache.HtpasswdFile(path)

  125.         ht.delete("user1")

  126.         self.assertEqual(get_file(path), sample)

  127. 

  128.         ht = apache.HtpasswdFile(path, autosave=True)

  129.         ht.delete("user1")

  130.         self.assertEqual(get_file(path), b"user2:pass2\n")

  131. 

  132.     def test_02_set_password(self):

  133.         """test set_password()"""

  134.         ht = apache.HtpasswdFile.from_string(

  135.             self.sample_01, default_scheme="plaintext")

  136.         self.assertTrue(ht.set_password("user2", "pass2x"))

  137.         self.assertFalse(ht.set_password("user5", "pass5"))

  138.         self.assertEqual(ht.to_string(), self.sample_03)

  139. 

  140.         # test legacy default kwd

  141.         with self.assertWarningList("``default`` is deprecated"):

  142.             ht = apache.HtpasswdFile.from_string(self.sample_01, default="plaintext")

  143.         self.assertTrue(ht.set_password("user2", "pass2x"))

  144.         self.assertFalse(ht.set_password("user5", "pass5"))

  145.         self.assertEqual(ht.to_string(), self.sample_03)

  146. 

  147.         # invalid user

  148.         self.assertRaises(ValueError, ht.set_password, "user:", "pass")

  149. 

  150.         # test that legacy update() still works

  151.         with self.assertWarningList("update\(\) is deprecated"):

  152.             ht.update("user2", "test")

  153.         self.assertTrue(ht.check_password("user2", "test"))

  154. 

  155.     def test_02_set_password_autosave(self):

  156.         path = self.mktemp()

  157.         sample = b'user1:pass1\n'

  158.         set_file(path, sample)

  159. 

  160.         ht = apache.HtpasswdFile(path)

  161.         ht.set_password("user1", "pass2")

  162.         self.assertEqual(get_file(path), sample)

  163. 

  164.         ht = apache.HtpasswdFile(path, default_scheme="plaintext", autosave=True)

  165.         ht.set_password("user1", "pass2")

  166.         self.assertEqual(get_file(path), b"user1:pass2\n")

  167. 

  168.     def test_02_set_password_default_scheme(self):

  169.         """test set_password() -- default_scheme"""

  170. 

  171.         def check(scheme):

  172.             ht = apache.HtpasswdFile(default_scheme=scheme)

  173.             ht.set_password("user1", "pass1")

  174.             return ht.context.identify(ht.get_hash("user1"))

  175. 

  176.         # explicit scheme

  177.         self.assertEqual(check("sha256_crypt"), "sha256_crypt")

  178.         self.assertEqual(check("des_crypt"), "des_crypt")

  179. 

  180.         # unknown scheme

  181.         self.assertRaises(KeyError, check, "xxx")

  182. 

  183.         # alias resolution

  184.         self.assertEqual(check("portable"), apache.htpasswd_defaults["portable"])

  185.         self.assertEqual(check("portable_apache_22"), apache.htpasswd_defaults["portable_apache_22"])

  186.         self.assertEqual(check("host_apache_22"), apache.htpasswd_defaults["host_apache_22"])

  187. 

  188.         # default

  189.         self.assertEqual(check(None), apache.htpasswd_defaults["portable_apache_22"])

  190. 

  191.     def test_03_users(self):

  192.         """test users()"""

  193.         ht = apache.HtpasswdFile.from_string(self.sample_01)

  194.         ht.set_password("user5", "pass5")

  195.         ht.delete("user3")

  196.         ht.set_password("user3", "pass3")

  197.         self.assertEqual(sorted(ht.users()), ["user1", "user2", "user3", "user4", "user5"])

  198. 

  199.     def test_04_check_password(self):

  200.         """test check_password()"""

  201.         ht = apache.HtpasswdFile.from_string(self.sample_05)

  202.         self.assertRaises(TypeError, ht.check_password, 1, 'pass9')

  203.         self.assertTrue(ht.check_password("user9","pass9") is None)

  204. 

  205.         # users 1..6 of sample_01 run through all the main hash formats,

  206.         # to make sure they're recognized.

  207.         for i in irange(1, 7):

  208.             i = str(i)

  209.             try:

  210.                 self.assertTrue(ht.check_password("user"+i, "pass"+i))

  211.                 self.assertTrue(ht.check_password("user"+i, "pass9") is False)

  212.             except MissingBackendError:

  213.                 if i == "5":

  214.                     # user5 uses bcrypt, which is apparently not available right now

  215.                     continue

  216.                 raise

  217. 

  218.         self.assertRaises(ValueError, ht.check_password, "user:", "pass")

  219. 

  220.         # test that legacy verify() still works

  221.         with self.assertWarningList(["verify\(\) is deprecated"]*2):

  222.             self.assertTrue(ht.verify("user1", "pass1"))

  223.             self.assertFalse(ht.verify("user1", "pass2"))

  224. 

  225.     def test_05_load(self):

  226.         """test load()"""

  227.         # setup empty file

  228.         path = self.mktemp()

  229.         set_file(path, "")

  230.         backdate_file_mtime(path, 5)

  231.         ha = apache.HtpasswdFile(path, default_scheme="plaintext")

  232.         self.assertEqual(ha.to_string(), b"")

  233. 

  234.         # make changes, check load_if_changed() does nothing

  235.         ha.set_password("user1", "pass1")

  236.         ha.load_if_changed()

  237.         self.assertEqual(ha.to_string(), b"user1:pass1\n")

  238. 

  239.         # change file

  240.         set_file(path, self.sample_01)

  241.         ha.load_if_changed()

  242.         self.assertEqual(ha.to_string(), self.sample_01)

  243. 

  244.         # make changes, check load() overwrites them

  245.         ha.set_password("user5", "pass5")

  246.         ha.load()

  247.         self.assertEqual(ha.to_string(), self.sample_01)

  248. 

  249.         # test load w/ no path

  250.         hb = apache.HtpasswdFile()

  251.         self.assertRaises(RuntimeError, hb.load)

  252.         self.assertRaises(RuntimeError, hb.load_if_changed)

  253. 

  254.         # test load w/ dups and explicit path

  255.         set_file(path, self.sample_dup)

  256.         hc = apache.HtpasswdFile()

  257.         hc.load(path)

  258.         self.assertTrue(hc.check_password('user1','pass1'))

  259. 

  260.     # NOTE: load_string() tested via from_string(), which is used all over this file

  261. 

  262.     def test_06_save(self):

  263.         """test save()"""

  264.         # load from file

  265.         path = self.mktemp()

  266.         set_file(path, self.sample_01)

  267.         ht = apache.HtpasswdFile(path)

  268. 

  269.         # make changes, check they saved

  270.         ht.delete("user1")

  271.         ht.delete("user2")

  272.         ht.save()

  273.         self.assertEqual(get_file(path), self.sample_02)

  274. 

  275.         # test save w/ no path

  276.         hb = apache.HtpasswdFile(default_scheme="plaintext")

  277.         hb.set_password("user1", "pass1")

  278.         self.assertRaises(RuntimeError, hb.save)

  279. 

  280.         # test save w/ explicit path

  281.         hb.save(path)

  282.         self.assertEqual(get_file(path), b"user1:pass1\n")

  283. 

  284.     def test_07_encodings(self):

  285.         """test 'encoding' kwd"""

  286.         # test bad encodings cause failure in constructor

  287.         self.assertRaises(ValueError, apache.HtpasswdFile, encoding="utf-16")

  288. 

  289.         # check sample utf-8

  290.         ht = apache.HtpasswdFile.from_string(self.sample_04_utf8, encoding="utf-8",

  291.                                              return_unicode=True)

  292.         self.assertEqual(ht.users(), [ u("user\u00e6") ])

  293. 

  294.         # test deprecated encoding=None

  295.         with self.assertWarningList("``encoding=None`` is deprecated"):

  296.             ht = apache.HtpasswdFile.from_string(self.sample_04_utf8, encoding=None)

  297.         self.assertEqual(ht.users(), [ b'user\xc3\xa6' ])

  298. 

  299.         # check sample latin-1

  300.         ht = apache.HtpasswdFile.from_string(self.sample_04_latin1,

  301.                                               encoding="latin-1", return_unicode=True)

  302.         self.assertEqual(ht.users(), [ u("user\u00e6") ])

  303. 

  304.     def test_08_get_hash(self):

  305.         """test get_hash()"""

  306.         ht = apache.HtpasswdFile.from_string(self.sample_01)

  307.         self.assertEqual(ht.get_hash("user3"), b"{SHA}3ipNV1GrBtxPmHFC21fCbVCSXIo=")

  308.         self.assertEqual(ht.get_hash("user4"), b"pass4")

  309.         self.assertEqual(ht.get_hash("user5"), None)

  310. 

  311.         with self.assertWarningList("find\(\) is deprecated"):

  312.             self.assertEqual(ht.find("user4"), b"pass4")

  313. 

  314.     def test_09_to_string(self):

  315.         """test to_string"""

  316. 

  317.         # check with known sample

  318.         ht = apache.HtpasswdFile.from_string(self.sample_01)

  319.         self.assertEqual(ht.to_string(), self.sample_01)

  320. 

  321.         # test blank

  322.         ht = apache.HtpasswdFile()

  323.         self.assertEqual(ht.to_string(), b"")

  324. 

  325.     def test_10_repr(self):

  326.         ht = apache.HtpasswdFile("fakepath", autosave=True, new=True, encoding="latin-1")

  327.         repr(ht)

  328. 

  329.     def test_11_malformed(self):

  330.         self.assertRaises(ValueError, apache.HtpasswdFile.from_string,

  331.             b'realm:user1:pass1\n')

  332.         self.assertRaises(ValueError, apache.HtpasswdFile.from_string,

  333.             b'pass1\n')

  334. 

  335.     def test_12_from_string(self):

  336.         # forbid path kwd

  337.         self.assertRaises(TypeError, apache.HtpasswdFile.from_string,

  338.                           b'', path=None)

  339. 

  340.     def test_13_whitespace(self):

  341.         """whitespace & comment handling"""

  342. 

  343.         # per htpasswd source (https://github.com/apache/httpd/blob/trunk/support/htpasswd.c),

  344.         # lines that match "^\s*(#.*)?$" should be ignored

  345.         source = to_bytes(

  346.             '\n'

  347.             'user2:pass2\n'

  348.             'user4:pass4\n'

  349.             'user7:pass7\r\n'

  350.             ' \t \n'

  351.             'user1:pass1\n'

  352.             ' # legacy users\n'

  353.             '#user6:pass6\n'

  354.             'user5:pass5\n\n'

  355.         )

  356. 

  357.         # loading should see all users (except user6, who was commented out)

  358.         ht = apache.HtpasswdFile.from_string(source)

  359.         self.assertEqual(sorted(ht.users()), ["user1", "user2", "user4", "user5", "user7"])

  360. 

  361.         # update existing user

  362.         ht.set_hash("user4", "althash4")

  363.         self.assertEqual(sorted(ht.users()), ["user1", "user2", "user4", "user5", "user7"])

  364. 

  365.         # add a new user

  366.         ht.set_hash("user6", "althash6")

  367.         self.assertEqual(sorted(ht.users()), ["user1", "user2", "user4", "user5", "user6", "user7"])

  368. 

  369.         # delete existing user

  370.         ht.delete("user7")

  371.         self.assertEqual(sorted(ht.users()), ["user1", "user2", "user4", "user5", "user6"])

  372. 

  373.         # re-serialization should preserve whitespace

  374.         target = to_bytes(

  375.             '\n'

  376.             'user2:pass2\n'

  377.             'user4:althash4\n'

  378.             ' \t \n'

  379.             'user1:pass1\n'

  380.             ' # legacy users\n'

  381.             '#user6:pass6\n'

  382.             'user5:pass5\n'

  383.             'user6:althash6\n'

  384.         )

  385.         self.assertEqual(ht.to_string(), target)

  386. 

  387.     #===================================================================

  388.     # eoc

  389.     #===================================================================

  390. 

  391. #=============================================================================

  392. # htdigest

  393. #=============================================================================

  394. class HtdigestFileTest(TestCase):

  395.     """test HtdigestFile class"""

  396.     descriptionPrefix = "HtdigestFile"

  397. 

  398.     # sample with 4 users

  399.     sample_01 = (b'user2:realm:549d2a5f4659ab39a80dac99e159ab19\n'

  400.                  b'user3:realm:a500bb8c02f6a9170ae46af10c898744\n'

  401.                  b'user4:realm:ab7b5d5f28ccc7666315f508c7358519\n'

  402.                  b'user1:realm:2a6cf53e7d8f8cf39d946dc880b14128\n')

  403. 

  404.     # sample 1 with user 1, 2 deleted; 4 changed

  405.     sample_02 = (b'user3:realm:a500bb8c02f6a9170ae46af10c898744\n'

  406.                  b'user4:realm:ab7b5d5f28ccc7666315f508c7358519\n')

  407. 

  408.     # sample 1 with user2 updated, user 1 first entry removed, and user 5 added

  409.     sample_03 = (b'user2:realm:5ba6d8328943c23c64b50f8b29566059\n'

  410.                  b'user3:realm:a500bb8c02f6a9170ae46af10c898744\n'

  411.                  b'user4:realm:ab7b5d5f28ccc7666315f508c7358519\n'

  412.                  b'user1:realm:2a6cf53e7d8f8cf39d946dc880b14128\n'

  413.                  b'user5:realm:03c55fdc6bf71552356ad401bdb9af19\n')

  414. 

  415.     # standalone sample with 8-bit username & realm

  416.     sample_04_utf8 = b'user\xc3\xa6:realm\xc3\xa6:549d2a5f4659ab39a80dac99e159ab19\n'

  417.     sample_04_latin1 = b'user\xe6:realm\xe6:549d2a5f4659ab39a80dac99e159ab19\n'

  418. 

  419.     def test_00_constructor_autoload(self):

  420.         """test constructor autoload"""

  421.         # check with existing file

  422.         path = self.mktemp()

  423.         set_file(path, self.sample_01)

  424.         ht = apache.HtdigestFile(path)

  425.         self.assertEqual(ht.to_string(), self.sample_01)

  426. 

  427.         # check without autoload

  428.         ht = apache.HtdigestFile(path, new=True)

  429.         self.assertEqual(ht.to_string(), b"")

  430. 

  431.         # check missing file

  432.         os.remove(path)

  433.         self.assertRaises(IOError, apache.HtdigestFile, path)

  434. 

  435.         # NOTE: default_realm option checked via other tests.

  436. 

  437.     def test_01_delete(self):

  438.         """test delete()"""

  439.         ht = apache.HtdigestFile.from_string(self.sample_01)

  440.         self.assertTrue(ht.delete("user1", "realm"))

  441.         self.assertTrue(ht.delete("user2", "realm"))

  442.         self.assertFalse(ht.delete("user5", "realm"))

  443.         self.assertFalse(ht.delete("user3", "realm5"))

  444.         self.assertEqual(ht.to_string(), self.sample_02)

  445. 

  446.         # invalid user

  447.         self.assertRaises(ValueError, ht.delete, "user:", "realm")

  448. 

  449.         # invalid realm

  450.         self.assertRaises(ValueError, ht.delete, "user", "realm:")

  451. 

  452.     def test_01_delete_autosave(self):

  453.         path = self.mktemp()

  454.         set_file(path, self.sample_01)

  455. 

  456.         ht = apache.HtdigestFile(path)

  457.         self.assertTrue(ht.delete("user1", "realm"))

  458.         self.assertFalse(ht.delete("user3", "realm5"))

  459.         self.assertFalse(ht.delete("user5", "realm"))

  460.         self.assertEqual(get_file(path), self.sample_01)

  461. 

  462.         ht.autosave = True

  463.         self.assertTrue(ht.delete("user2", "realm"))

  464.         self.assertEqual(get_file(path), self.sample_02)

  465. 

  466.     def test_02_set_password(self):

  467.         """test update()"""

  468.         ht = apache.HtdigestFile.from_string(self.sample_01)

  469.         self.assertTrue(ht.set_password("user2", "realm", "pass2x"))

  470.         self.assertFalse(ht.set_password("user5", "realm", "pass5"))

  471.         self.assertEqual(ht.to_string(), self.sample_03)

  472. 

  473.         # default realm

  474.         self.assertRaises(TypeError, ht.set_password, "user2", "pass3")

  475.         ht.default_realm = "realm2"

  476.         ht.set_password("user2", "pass3")

  477.         ht.check_password("user2", "realm2", "pass3")

  478. 

  479.         # invalid user

  480.         self.assertRaises(ValueError, ht.set_password, "user:", "realm", "pass")

  481.         self.assertRaises(ValueError, ht.set_password, "u"*256, "realm", "pass")

  482. 

  483.         # invalid realm

  484.         self.assertRaises(ValueError, ht.set_password, "user", "realm:", "pass")

  485.         self.assertRaises(ValueError, ht.set_password, "user", "r"*256, "pass")

  486. 

  487.         # test that legacy update() still works

  488.         with self.assertWarningList("update\(\) is deprecated"):

  489.             ht.update("user2", "realm2", "test")

  490.         self.assertTrue(ht.check_password("user2", "test"))

  491. 

  492.     # TODO: test set_password autosave

  493. 

  494.     def test_03_users(self):

  495.         """test users()"""

  496.         ht = apache.HtdigestFile.from_string(self.sample_01)

  497.         ht.set_password("user5", "realm", "pass5")

  498.         ht.delete("user3", "realm")

  499.         ht.set_password("user3", "realm", "pass3")

  500.         self.assertEqual(sorted(ht.users("realm")), ["user1", "user2", "user3", "user4", "user5"])

  501. 

  502.         self.assertRaises(TypeError, ht.users, 1)

  503. 

  504.     def test_04_check_password(self):

  505.         """test check_password()"""

  506.         ht = apache.HtdigestFile.from_string(self.sample_01)

  507.         self.assertRaises(TypeError, ht.check_password, 1, 'realm', 'pass5')

  508.         self.assertRaises(TypeError, ht.check_password, 'user', 1, 'pass5')

  509.         self.assertIs(ht.check_password("user5", "realm","pass5"), None)

  510.         for i in irange(1,5):

  511.             i = str(i)

  512.             self.assertTrue(ht.check_password("user"+i, "realm", "pass"+i))

  513.             self.assertIs(ht.check_password("user"+i, "realm", "pass5"), False)

  514. 

  515.         # default realm

  516.         self.assertRaises(TypeError, ht.check_password, "user5", "pass5")

  517.         ht.default_realm = "realm"

  518.         self.assertTrue(ht.check_password("user1", "pass1"))

  519.         self.assertIs(ht.check_password("user5", "pass5"), None)

  520. 

  521.         # test that legacy verify() still works

  522.         with self.assertWarningList(["verify\(\) is deprecated"]*2):

  523.             self.assertTrue(ht.verify("user1", "realm", "pass1"))

  524.             self.assertFalse(ht.verify("user1", "realm", "pass2"))

  525. 

  526.         # invalid user

  527.         self.assertRaises(ValueError, ht.check_password, "user:", "realm", "pass")

  528. 

  529.     def test_05_load(self):

  530.         """test load()"""

  531.         # setup empty file

  532.         path = self.mktemp()

  533.         set_file(path, "")

  534.         backdate_file_mtime(path, 5)

  535.         ha = apache.HtdigestFile(path)

  536.         self.assertEqual(ha.to_string(), b"")

  537. 

  538.         # make changes, check load_if_changed() does nothing

  539.         ha.set_password("user1", "realm", "pass1")

  540.         ha.load_if_changed()

  541.         self.assertEqual(ha.to_string(), b'user1:realm:2a6cf53e7d8f8cf39d946dc880b14128\n')

  542. 

  543.         # change file

  544.         set_file(path, self.sample_01)

  545.         ha.load_if_changed()

  546.         self.assertEqual(ha.to_string(), self.sample_01)

  547. 

  548.         # make changes, check load_if_changed overwrites them

  549.         ha.set_password("user5", "realm", "pass5")

  550.         ha.load()

  551.         self.assertEqual(ha.to_string(), self.sample_01)

  552. 

  553.         # test load w/ no path

  554.         hb = apache.HtdigestFile()

  555.         self.assertRaises(RuntimeError, hb.load)

  556.         self.assertRaises(RuntimeError, hb.load_if_changed)

  557. 

  558.         # test load w/ explicit path

  559.         hc = apache.HtdigestFile()

  560.         hc.load(path)

  561.         self.assertEqual(hc.to_string(), self.sample_01)

  562. 

  563.         # change file, test deprecated force=False kwd

  564.         ensure_mtime_changed(path)

  565.         set_file(path, "")

  566.         with self.assertWarningList(r"load\(force=False\) is deprecated"):

  567.             ha.load(force=False)

  568.         self.assertEqual(ha.to_string(), b"")

  569. 

  570.     def test_06_save(self):

  571.         """test save()"""

  572.         # load from file

  573.         path = self.mktemp()

  574.         set_file(path, self.sample_01)

  575.         ht = apache.HtdigestFile(path)

  576. 

  577.         # make changes, check they saved

  578.         ht.delete("user1", "realm")

  579.         ht.delete("user2", "realm")

  580.         ht.save()

  581.         self.assertEqual(get_file(path), self.sample_02)

  582. 

  583.         # test save w/ no path

  584.         hb = apache.HtdigestFile()

  585.         hb.set_password("user1", "realm", "pass1")

  586.         self.assertRaises(RuntimeError, hb.save)

  587. 

  588.         # test save w/ explicit path

  589.         hb.save(path)

  590.         self.assertEqual(get_file(path), hb.to_string())

  591. 

  592.     def test_07_realms(self):

  593.         """test realms() & delete_realm()"""

  594.         ht = apache.HtdigestFile.from_string(self.sample_01)

  595. 

  596.         self.assertEqual(ht.delete_realm("x"), 0)

  597.         self.assertEqual(ht.realms(), ['realm'])

  598. 

  599.         self.assertEqual(ht.delete_realm("realm"), 4)

  600.         self.assertEqual(ht.realms(), [])

  601.         self.assertEqual(ht.to_string(), b"")

  602. 

  603.     def test_08_get_hash(self):

  604.         """test get_hash()"""

  605.         ht = apache.HtdigestFile.from_string(self.sample_01)

  606.         self.assertEqual(ht.get_hash("user3", "realm"), "a500bb8c02f6a9170ae46af10c898744")

  607.         self.assertEqual(ht.get_hash("user4", "realm"), "ab7b5d5f28ccc7666315f508c7358519")

  608.         self.assertEqual(ht.get_hash("user5", "realm"), None)

  609. 

  610.         with self.assertWarningList("find\(\) is deprecated"):

  611.             self.assertEqual(ht.find("user4", "realm"), "ab7b5d5f28ccc7666315f508c7358519")

  612. 

  613.     def test_09_encodings(self):

  614.         """test encoding parameter"""

  615.         # test bad encodings cause failure in constructor

  616.         self.assertRaises(ValueError, apache.HtdigestFile, encoding="utf-16")

  617. 

  618.         # check sample utf-8

  619.         ht = apache.HtdigestFile.from_string(self.sample_04_utf8, encoding="utf-8", return_unicode=True)

  620.         self.assertEqual(ht.realms(), [ u("realm\u00e6") ])

  621.         self.assertEqual(ht.users(u("realm\u00e6")), [ u("user\u00e6") ])

  622. 

  623.         # check sample latin-1

  624.         ht = apache.HtdigestFile.from_string(self.sample_04_latin1, encoding="latin-1", return_unicode=True)

  625.         self.assertEqual(ht.realms(), [ u("realm\u00e6") ])

  626.         self.assertEqual(ht.users(u("realm\u00e6")), [ u("user\u00e6") ])

  627. 

  628.     def test_10_to_string(self):

  629.         """test to_string()"""

  630. 

  631.         # check sample

  632.         ht = apache.HtdigestFile.from_string(self.sample_01)

  633.         self.assertEqual(ht.to_string(), self.sample_01)

  634. 

  635.         # check blank

  636.         ht = apache.HtdigestFile()

  637.         self.assertEqual(ht.to_string(), b"")

  638. 

  639.     def test_11_malformed(self):

  640.         self.assertRaises(ValueError, apache.HtdigestFile.from_string,

  641.             b'realm:user1:pass1:other\n')

  642.         self.assertRaises(ValueError, apache.HtdigestFile.from_string,

  643.             b'user1:pass1\n')

  644. 

  645.     #===================================================================

  646.     # eoc

  647.     #===================================================================

  648. 

  649. #=============================================================================

  650. # eof

  651. #=============================================================================