source: main/waeup.sirp/trunk/src/waeup/sirp/utils/tests/test_converters.py @ 6264

##
## test_converters.py
## Login : <uli@pu.smp.net>
## Started on  Tue May 31 08:34:44 2011 Uli Fouquet
## $Id$
## 
## Copyright (C) 2011 Uli Fouquet
## This program is free software; you can redistribute it and/or modify
## it under the terms of the GNU General Public License as published by
## the Free Software Foundation; either version 2 of the License, or
## (at your option) any later version.
## 
## This program is distributed in the hope that it will be useful,
## but WITHOUT ANY WARRANTY; without even the implied warranty of
## MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
## GNU General Public License for more details.
## 
## You should have received a copy of the GNU General Public License
## along with this program; if not, write to the Free Software
## Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
##
"""
Tests for converterts.
"""
import datetime
import os
import shutil
import tempfile
import unittest
from zope.app.testing.functional import FunctionalTestCase
from zope.component.hooks import setSite, clearSite
from zope.interface.verify import verifyClass, verifyObject

from waeup.sirp.app import University
from waeup.sirp.testing import FunctionalLayer, doctestsuite_for_module
from waeup.sirp.utils.converters import IObjectConverter

from zope import schema
from zope.interface import Interface, implements, invariant, Invalid
from waeup.sirp.utils.helpers import attrs_to_fields

class IContact(Interface):
    name = schema.TextLine(
        title = u'Name',
        default = u'Manfred'
        )
    age = schema.Int(
        title = u'Age',
        default = 23,
        required = True,
        )
    city = schema.TextLine(
        title = u'City',
        required = True,
        )
    vip = schema.Bool(
        title = u'Celebrity',
        default = False,
        required = True,
        )
    birthday = schema.Date(
        title = u'Birthday',
        default = None,
        )
    @invariant
    def kevinIsYoung(contact):
        if contact.age > 16 and contact.name == 'Kevin':
            raise Invalid('Kevins are age 16 or below.')

class Contact(object):
    implements(IContact)
    name = None
    age = None
Contact = attrs_to_fields(Contact)

class ConverterTests(FunctionalTestCase):

    layer = FunctionalLayer

    def setUp(self):
        super(ConverterTests, self).setUp()

        # Setup a sample site for each test
        app = University()
        self.dc_root = tempfile.mkdtemp()
        app['datacenter'].setStoragePath(self.dc_root)

        # Prepopulate the ZODB...
        self.getRootFolder()['app'] = app
        self.app = self.getRootFolder()['app']

        self.workdir = tempfile.mkdtemp()
        return

    def tearDown(self):
        super(ConverterTests, self).tearDown()
        shutil.rmtree(self.workdir)
        shutil.rmtree(self.dc_root)
        clearSite()
        return

    def test_valid_data(self):
        contact = Contact()
        contact.age = 33
        input_data = dict(name='Rudi', age='99')
        converter = IObjectConverter(IContact) # a converter to IContact
        err, inv_err, new_contact = converter.applyRowData(
            input_data, contact)
        assert new_contact is contact
        assert contact.name == 'Rudi'
        assert contact.age == 99
        return

    def test_bool(self):
        contact1 = Contact()
        contact2 = Contact()
        input_data1 = dict(vip='on')
        input_data2 = dict(vip='')
        converter = IObjectConverter(IContact) # a converter to IContact
        err1, inv_err1, new_contact1 = converter.applyRowData(
            input_data1, contact1)
        err2, inv_err2, new_contact2 = converter.applyRowData(
            input_data2, contact2)
        assert contact1.vip is True
        assert contact2.vip is False

    def test_int(self):
        contact = Contact()
        input_data = dict(age='99')
        converter = IObjectConverter(IContact) # a converter to IContact
        err, inv_err, new_contact = converter.applyRowData(
            input_data, contact)
        assert contact.age == 99
        return

    def test_int_invalid(self):
        contact = Contact()
        input_data = dict(age='sweet sixteen')
        converter = IObjectConverter(IContact) # a converter to IContact
        err, inv_err, new_contact = converter.applyRowData(
            input_data, contact)
        self.assertEqual(err, [('age', u'Invalid integer data')])
        return

    def test_textline(self):
        contact = Contact()
        input_data = dict(name='Rudi')
        converter = IObjectConverter(IContact) # a converter to IContact
        err, inv_err, new_contact = converter.applyRowData(
            input_data, contact)
        self.assertEqual(contact.name, u'Rudi')
        assert isinstance(contact.name, unicode)
        return

    def test_invariant(self):
        contact = Contact()
        input_data = dict(name='Kevin', age='22')
        converter = IObjectConverter(IContact) # a converter to IContact
        err, inv_err, new_contact = converter.applyRowData(
            input_data, contact)
        self.assertEqual(inv_err, ['Kevins are age 16 or below.'])
        return

    def test_date(self):
        contact = Contact()
        converter = IObjectConverter(IContact) # a converter to IContact
        err, inv_err, new_contact = converter.applyRowData(
            dict(birthday='1945/12/23'), contact)
        assert contact.birthday == datetime.date(1945, 12, 23)
        assert isinstance(contact.birthday, datetime.date)

        err, inv_err, new_contact = converter.applyRowData(
            dict(birthday='1945/23/12'), contact)
        assert contact.birthday == datetime.date(1945, 12, 23)

        err, inv_err, new_contact = converter.applyRowData(
            dict(birthday='23/12/1945'), contact)
        assert contact.birthday == datetime.date(1945, 12, 23)

        err, inv_err, new_contact = converter.applyRowData(
            dict(birthday='23.12.1945'), contact)
        assert contact.birthday == datetime.date(1945, 12, 23)

        err, inv_err, new_contact = converter.applyRowData(
            dict(birthday='23-12-1945'), contact)
        assert contact.birthday == datetime.date(1945, 12, 23)
        return

    def test_date_invalid(self):
        contact = Contact()
        converter = IObjectConverter(IContact) # a converter to IContact
        err, inv_err, new_contact = converter.applyRowData(
            dict(birthday='not-a-date'), contact)
        self.assertEqual(err, [('birthday', u'Invalid datetime data')])

def test_suite():
    suite = unittest.TestSuite()
    for testcase in [
        ConverterTests,
        ]:
        suite.addTest(unittest.TestLoader().loadTestsFromTestCase(
                testcase
                )
        )
    return suite