source: products/quintagroup.plonecaptchas/trunk/quintagroup/plonecaptchas/tests/testForms.py @ 3162

import re
import unittest

from urllib import urlencode
from StringIO import StringIO
from DateTime import DateTime

from quintagroup.plonecaptchas.tests.base import FunctionalTestCase
from quintagroup.plonecaptchas.config import PRODUCT_NAME

from quintagroup.captcha.core.tests.testWidget import NOT_VALID
from quintagroup.captcha.core.tests.testWidget import IMAGE_PATT
from quintagroup.captcha.core.tests.testWidget import addTestLayer
from quintagroup.captcha.core.tests.base import testPatch
from quintagroup.captcha.core.utils import getWord, decrypt, parseKey

from Products.PloneTestCase.PloneTestCase import portal_owner
from Products.PloneTestCase.PloneTestCase import default_password

from plone.app.controlpanel.security import ISecuritySchema


# BBB for plone v<3.1, where plone.protect not used yet
PROTECT_SUPPORT = True
try:
    from plone import protect
    # fix for pyflakes test
    protect
except ImportError:
    PROTECT_SUPPORT = False

# USE PATCH FROM quintagroup.captcha.core
# patch to use test images and dictionary
testPatch()


class TestFormMixin(FunctionalTestCase):

    def afterSetUp(self):
        self.loginAsPortalOwner()
        self.addProduct(PRODUCT_NAME)
        # Add test_captcha layer from quintagroup.captcah.core
        addTestLayer(self)
        # Prepare form data
        self.basic_auth = ':'.join((portal_owner, default_password))
        self.form_url = ''
        self.form_method = "POST"
        self.hasAuthenticator = False
        self.form_data = self.getFormData()
        # Prepare captcha related test data
        self.captcha_key = self.portal.captcha_key
        self.hashkey = self.portal.getCaptcha()
        self.form_data['hashkey'] = self.hashkey
        self.form_data['key'] = ''

    def getFormData(self):
        raise NotImplementedError(
            "getFormData not implemented")

    def publishForm(self):
        stdin_data = None
        form_url = self.portal.absolute_url(1) + self.form_url
        # Prepare form data
        if PROTECT_SUPPORT and self.hasAuthenticator:
            self.form_data['_authenticator'] = self._getauth()
        form_data = urlencode(self.form_data)
        if self.form_method.upper() == 'GET':
            form_url += "?%s" % form_data
        else:
            stdin_data = StringIO(form_data)
        # Publish form and get response
        response = self.publish(form_url, self.basic_auth,
            request_method=self.form_method, stdin=stdin_data)
        return response

    def _getauth(self):
        # Fix authenticator for the form
        authenticator = self.portal.restrictedTraverse("@@authenticator")
        html = authenticator.authenticator()
        handle = re.search('value="(.*)"', html).groups()[0]
        return handle

    def testImage(self):
        self.form_data = {}
        self.form_method = "GET"
        response = self.publishForm().getBody()
        patt = re.compile(IMAGE_PATT % self.portal.absolute_url())
        match_obj = patt.search(response)
        img_url = match_obj.group(1)

        content_type = self.publish('/plone' + img_url).getHeader(
                                                            'content-type')
        self.assertTrue(content_type.startswith('image'),
            "Wrong captcha image content type")

    def testSubmitRightCaptcha(self):
        key = getWord(int(parseKey(decrypt(
                self.captcha_key, self.hashkey))['key']) - 1)
        self.form_data['key'] = key

        response = self.publishForm().getBody()
        self.assertFalse(NOT_VALID.search(response))

    def testSubmitWrongCaptcha(self):
        self.form_data['key'] = 'wrong word'
        response = self.publishForm().getBody()
        self.assertTrue(NOT_VALID.search(response))

    def testSubmitRightCaptchaTwice(self):
        key = getWord(int(parseKey(decrypt(
                self.captcha_key, self.hashkey))['key']) - 1)
        self.form_data['key'] = key

        self.publishForm()
        response = self.publishForm().getBody()
        self.assertTrue(NOT_VALID.search(response))


class TestDiscussionForm(TestFormMixin):

    def afterSetUp(self):
        TestFormMixin.afterSetUp(self)
        self.portal.invokeFactory('Document', 'index_html')
        self.portal['index_html'].allowDiscussion(True)
        self.form_url = '/index_html/discussion_reply_form'

    def getFormData(self):
        return {'form.submitted': '1',
                'subject': 'testing',
                'Creator': portal_owner,
                'body_text': 'Text in Comment',
                'discussion_reply:method': 'Save',
                'form.button.form_submit': 'Save'}


class TestJoinForm(TestFormMixin):

    def afterSetUp(self):
        TestFormMixin.afterSetUp(self)
        ISecuritySchema(self.portal).enable_self_reg = True
        self.hasAuthenticator = True
        self.form_url = '/join_form'
        self.basic_auth = ":"
        self.logout()

    def getFormData(self):
        return {"last_visit:date": str(DateTime()),
                "prev_visit:date": str(DateTime() - 1),
                "came_from_prefs": "",
                "fullname": "Tester",
                "username": "tester",
                "email": "tester@test.com",
                'form.button.Register': 'Register',
                'form.submitted': '1'}


class TestSendtoForm(TestFormMixin):

    def afterSetUp(self):
        TestFormMixin.afterSetUp(self)
        self.portal.invokeFactory('Document', 'index_html')
        self.portal['index_html'].allowDiscussion(True)
        self.form_url = '/index_html/sendto_form'

    def getFormData(self):
        return {'form.submitted': '1',
                "send_to_address": "recipient@test.com",
                "send_from_address": "sender@test.com",
                'comment': 'Text in Comment',
                'form.button.Send': 'Save'}


class TestContactInfo(TestFormMixin):

    def afterSetUp(self):
        TestFormMixin.afterSetUp(self)
        # preparation to form correct working
        self.portal._updateProperty('email_from_address', 'manager@test.com')
        self.logout()
        self.form_url = '/contact-info'

    def getFormData(self):
        return {'form.submitted': '1',
                "sender_fullname": "tester",
                "sender_from_address": "sender@test.com",
                'subject': 'Subject',
                'message': 'Message',
                'form.button.Send': 'Save'}


def test_suite():
    suite = unittest.TestSuite()
    suite.addTest(unittest.makeSuite(TestDiscussionForm))
    suite.addTest(unittest.makeSuite(TestJoinForm))
    suite.addTest(unittest.makeSuite(TestSendtoForm))
    suite.addTest(unittest.makeSuite(TestContactInfo))
    return suite