From 4b481c179c96440b3785ecc5a069af4d696f75a3 Mon Sep 17 00:00:00 2001
From: bwarsaw
Date: Thu, 22 Mar 2007 04:29:31 +0000
Subject: Restore the use of passwords.Schemes enum for selection of password
 hashing scheme.  Fix mmsitepass and test cases accordingly.  Details:

- set_global_password(): Instead of taking a string for 'scheme' argument,
  take None and then coerce that into passwords.Schemes.ssha

- Add a base PasswordError and a BadPasswordSchemeError error that derives
  from that.  For consistency, multiply inherit MMBadPasswordError and
  MMPasswordsMustMatch from PasswordError.

- Add a passwords.lookup_scheme() method which turns scheme_names into scheme
  enum constants.  It returns None if the lookup fails.

- passwords.py: change the internal representation of _SCHEMES_BY_TAG
  dictionary to map scheme names to scheme enum values.  Change internal uses
  of this dictionary to then turn those enum values into hash classes, or
  whatever else we need.

- make_secret(): Raise BadPasswordSchemeErrorif the given schema (which should
  be an enum value) is invalid.

- TestBase.tearDown(): Clear out any <site> locks that might hang around after
  a test case runs.
---
 Mailman/testing/test_passwords.py | 48 ++++++++++++++++++++++++++++++++-------
 1 file changed, 40 insertions(+), 8 deletions(-)

(limited to 'Mailman/testing/test_passwords.py')

diff --git a/Mailman/testing/test_passwords.py b/Mailman/testing/test_passwords.py
index b957b10ae..8298c22f6 100644
--- a/Mailman/testing/test_passwords.py
+++ b/Mailman/testing/test_passwords.py
@@ -19,6 +19,7 @@
 
 import unittest
 
+from Mailman import Errors
 from Mailman import passwords
 
 
@@ -71,6 +72,27 @@ class TestPasswordsBase(unittest.TestCase):
 class TestBogusPasswords(TestPasswordsBase):
     scheme = -1
 
+    def test_passwords(self):
+        self.assertRaises(Errors.BadPasswordSchemeError,
+                          passwords.make_secret, self.pw8a, self.scheme)
+
+    def test_unicode_passwords(self):
+        self.assertRaises(Errors.BadPasswordSchemeError,
+                          passwords.make_secret, self.pwua, self.scheme)
+
+    def test_passwords_with_funky_chars(self):
+        self.assertRaises(Errors.BadPasswordSchemeError,
+                          passwords.make_secret, self.pw8b, self.scheme)
+
+    def test_unicode_passwords_with_funky_chars(self):
+        self.assertRaises(Errors.BadPasswordSchemeError,
+                          passwords.make_secret, self.pwub, self.scheme)
+
+
+
+class TestNonePasswords(TestPasswordsBase):
+    scheme = passwords.Schemes.no_scheme
+
     def test_passwords(self):
         failif = self.failIf
         secret = passwords.make_secret(self.pw8a, self.scheme)
@@ -97,24 +119,33 @@ class TestBogusPasswords(TestPasswordsBase):
 
 
 
-class TestNonePasswords(TestBogusPasswords):
-    scheme = 'no_scheme'
-
-
 class TestCleartextPasswords(TestPasswordsBase):
-    scheme = 'cleartext'
+    scheme = passwords.Schemes.cleartext
 
 
 class TestSHAPasswords(TestPasswordsBase):
-    scheme = 'sha'
+    scheme = passwords.Schemes.sha
 
 
 class TestSSHAPasswords(TestPasswordsBase):
-    scheme = 'ssha'
+    scheme = passwords.Schemes.ssha
 
 
 class TestPBKDF2Passwords(TestPasswordsBase):
-    scheme = 'pbkdf2'
+    scheme = passwords.Schemes.pbkdf2
+
+
+
+class TestSchemeLookup(unittest.TestCase):
+    def test_scheme_name_lookup(self):
+        unless = self.failUnless
+        unless(passwords.lookup_scheme('NONE') is passwords.Schemes.no_scheme)
+        unless(passwords.lookup_scheme('CLEARTEXT') is
+               passwords.Schemes.cleartext)
+        unless(passwords.lookup_scheme('SHA') is passwords.Schemes.sha)
+        unless(passwords.lookup_scheme('SSHA') is passwords.Schemes.ssha)
+        unless(passwords.lookup_scheme('PBKDF2') is passwords.Schemes.pbkdf2)
+        unless(passwords.lookup_scheme(' -bogus- ') is None)
 
 
 
@@ -126,4 +157,5 @@ def test_suite():
     suite.addTest(unittest.makeSuite(TestSHAPasswords))
     suite.addTest(unittest.makeSuite(TestSSHAPasswords))
     suite.addTest(unittest.makeSuite(TestPBKDF2Passwords))
+    suite.addTest(unittest.makeSuite(TestSchemeLookup))
     return suite
-- 
cgit v1.2.3-70-g09d2