Changeset 2077 in products

Timestamp:

Apr 7, 2010 3:38:19 PM (14 years ago)

Author:

mylan

Message:

#185: Add tests for captcha validator

File:

• quintagroup.z3cform.captcha/branches/captchas-refactoring/quintagroup/z3cform/captcha/tests.py (modified) (4 diffs)

quintagroup.z3cform.captcha/branches/captchas-refactoring/quintagroup/z3cform/captcha/tests.py

@@ -19 +19 @@
 from Products.PloneTestCase import PloneTestCase as ptc
 
+from quintagroup.captcha.core.utils import *
+from quintagroup.captcha.core.tests.base import testPatch
+from quintagroup.captcha.core.tests.testWidget import addTestLayer
+
 from quintagroup.z3cform.captcha import Captcha
 from quintagroup.z3cform.captcha import CaptchaWidget
-from quintagroup.z3cform.captcha.widget import CaptchaWidgetFactory
+from quintagroup.z3cform.captcha import CaptchaWidgetFactory
+from quintagroup.z3cform.captcha.validator import CaptchaValidator
 
 @onsetup
@@ -43 +48 @@
 
     def afterSetUp(self):
-        super(TestRegistrations, self).afterSetUp()
         self.request = self.app.REQUEST
         alsoProvides(self.request, IFormLayer)
@@ -77 +81 @@
 
     def afterSetUp(self):
-        super(TestCaptchaWidget, self).afterSetUp()
         self.request = self.app.REQUEST
         alsoProvides(self.request, IFormLayer)
@@ -109 +112 @@
         
 
+class TestCaptchaValidator(ptc.PloneTestCase):
+
+    def afterSetUp(self):
+        self.request = self.app.REQUEST
+        alsoProvides(self.request, IFormLayer)
+        # prepare context
+        self.loginAsPortalOwner()
+        testPatch()
+        addTestLayer(self)
+        self.captcha_key = self.portal.captcha_key
+        # prepare captcha data
+        self.hashkey = self.portal.getCaptcha()
+        self.request.form['hashkey'] = self.hashkey
+        # prepare validator
+        cform = form.BaseForm(self.portal, self.request)
+        cform.prefix = ""
+        cwidget = CaptchaWidget(self.request)
+        cwidget.form = cform
+        self.validator = CaptchaValidator(self.portal, self.request, None, None, cwidget)
+
+    def testSubmitRightCaptcha(self):
+        decrypted = decrypt(self.captcha_key, self.hashkey)
+        key = getWord(int(parseKey(decrypted)['key'])-1 )
+        try:
+            res = self.validator.validate(key)
+        except ConversionError, e:
+            self.fail("Rised unexpected %s error on right captcha submit" % e)
+
+    def testSubmitWrongCaptcha(self):
+        try:
+            res = self.validator.validate("wrong key")
+        except ValueError, e:
+            self.assertEqual(str(e), u'Please re-enter validation code.')
+        else:
+            self.fail("No ValueError rised on wrong captcha key submit")
+
+    def testSubmitRightCaptchaTwice(self):
+        decrypted = decrypt(self.captcha_key, self.hashkey)
+        key = getWord(int(parseKey(decrypted)['key'])-1 )
+        self.validator.validate(key)
+        try:
+            res = self.validator.validate(key)
+        except ValueError, e:
+            self.assertEqual(str(e), u'Please re-enter validation code.')
+        else:
+            self.fail("No ValueError rised on right captcha key " \
+                      "submitting twice")
+
+
 def test_suite():
     suite = unittest.TestSuite()
     suite.addTest(unittest.makeSuite(TestRegistrations))
     suite.addTest(unittest.makeSuite(TestCaptchaWidget))
+    suite.addTest(unittest.makeSuite(TestCaptchaValidator))
     return suite