source: main/waeup.kofa/trunk/src/waeup/kofa/widgets/tests/test_datetimewidget.py @ 8201

# Tests for datetimewidgets
import datetime
import pytz
from zope import schema
from zope.formlib import form
from zope.formlib.interfaces import IInputWidget, IDisplayWidget
from zope.formlib.tests.test_functional_textwidget import(
    FunctionalWidgetTestCase,)
from zope.formlib.textwidgets import TextWidget
from zope.interface import Interface, implements
from zope.interface.verify import verifyClass, verifyObject
from zope.publisher.browser import TestRequest
from waeup.kofa.widgets.datetimewidget import PytzDatetimeWidget

class IContent(Interface):
    my_dt = schema.Datetime(
        title = u'A datetime.',
        )

class Content(object):
    implements(IContent)
    my_dt = None

class SampleForm(form.EditForm):
    form_fields = form.fields(IContent)
    form_fields['my_dt'].custom_widget = PytzDatetimeWidget

class PytzDatetimeWidgetTests(FunctionalWidgetTestCase):

    widgets = [
        (schema.interfaces.IDatetime, TextWidget),
        ]

    def test_iface(self):
        # make sure we fullfill interface contracts
        obj = PytzDatetimeWidget(IContent['my_dt'], TestRequest())
        verifyClass(IInputWidget, PytzDatetimeWidget)
        verifyObject(IInputWidget, obj)
        return

    def test_to_value(self):
        # we get always UTC-based values
        content = Content()
        request = TestRequest()

        request.form['form.my_dt'] = u'2012-01-02 12:10:19 GMT+1'
        request.form['form.actions.apply'] = u''
        SampleForm(content, request)()

        self.assertEqual( content.my_dt, datetime.datetime(
            2012, 1, 2, 11, 10, 19, tzinfo=pytz.utc))
        self.assertTrue(content.my_dt.tzinfo is pytz.utc)
        return

    def test_export_values(self):
        # make sure we understand values as exported
        content = Content()
        request = TestRequest()

        request.form['form.my_dt'] = u'2012-01-02 12:10:19+00:00'
        request.form['form.actions.apply'] = u''
        SampleForm(content, request)()

        self.assertEqual( content.my_dt, datetime.datetime(
            2012, 1, 2, 12, 10, 19, tzinfo=pytz.utc))
        self.assertTrue(content.my_dt.tzinfo is pytz.utc)
        return