+
−# -*- coding: iso-8859-1 -*-
+
−# copyright 2003-2010 LOGILAB S.A. (Paris, FRANCE), all rights reserved.
+
−# contact http://www.logilab.fr/ -- mailto:contact@logilab.fr
+
−#
+
−# This file is part of CubicWeb.
+
−#
+
−# CubicWeb is free software: you can redistribute it and/or modify it under the
+
−# terms of the GNU Lesser General Public License as published by the Free
+
−# Software Foundation, either version 2.1 of the License, or (at your option)
+
−# any later version.
+
−#
+
−# CubicWeb 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 Lesser General Public License for more
+
−# details.
+
−#
+
−# You should have received a copy of the GNU Lesser General Public License along
+
−# with CubicWeb.  If not, see <http://www.gnu.org/licenses/>.
+
−"""unit tests for database value generator"""
+
−
+
−import os.path as osp
+
−import re
+
−import datetime
+
−
+
−from logilab.common.testlib import TestCase, unittest_main
+
−
+
−from cubicweb.schema import Schema, EntitySchema
+
−from cubicweb.devtools.fill import ValueGenerator, make_tel
+
−from cubicweb.devtools import ApptestConfiguration
+
−
+
−DATADIR = osp.join(osp.abspath(osp.dirname(__file__)), 'data')
+
−ISODATE_SRE = re.compile('(?P<year>\d{4})-(?P<month>\d{2})-(?P<day>\d{2})$')
+
−
+
−
+
−class MyValueGenerator(ValueGenerator):
+
−
+
−    def generate_Bug_severity(self, entity, index):
+
−        return u'dangerous'
+
−
+
−    def generate_Any_description(self, entity, index, format=None):
+
−        return u'yo'
+
−
+
−
+
−class ValueGeneratorTC(TestCase):
+
−    """test case for ValueGenerator"""
+
−
+
−    def _choice_func(self, etype, attrname):
+
−        try:
+
−            return getattr(self, '_available_%s_%s' % (etype, attrname))(etype, attrname)
+
−        except AttributeError:
+
−            return None
+
−
+
−    def _available_Person_firstname(self, etype, attrname):
+
−        return [f.strip() for f in file(osp.join(DATADIR, 'firstnames.txt'))]
+
−
+
−    def setUp(self):
+
−        config = ApptestConfiguration('data', apphome=DATADIR)
+
−        config.bootstrap_cubes()
+
−        schema = config.load_schema()
+
−        e_schema = schema.eschema('Person')
+
−        self.person_valgen = ValueGenerator(e_schema, self._choice_func)
+
−        e_schema = schema.eschema('Bug')
+
−        self.bug_valgen = MyValueGenerator(e_schema)
+
−        self.config = config
+
−
+
−    def test_string(self):
+
−        """test string generation"""
+
−        surname = self.person_valgen.generate_attribute_value({}, 'surname', 12)
+
−        self.assertEqual(surname, u'�&surname12')
+
−
+
−    def test_domain_value(self):
+
−        """test value generation from a given domain value"""
+
−        firstname = self.person_valgen.generate_attribute_value({}, 'firstname', 12)
+
−        possible_choices = self._choice_func('Person', 'firstname')
+
−        self.assertTrue(firstname in possible_choices,
+
−                        '%s not in %s' % (firstname, possible_choices))
+
−
+
−    def test_choice(self):
+
−        """test choice generation"""
+
−        # Test for random index
+
−        for index in range(5):
+
−            sx_value = self.person_valgen.generate_attribute_value({}, 'civility', index)
+
−            self.assertTrue(sx_value in ('Mr', 'Mrs', 'Ms'))
+
−
+
−    def test_integer(self):
+
−        """test integer generation"""
+
−        # Test for random index
+
−        for index in range(5):
+
−            cost_value = self.bug_valgen.generate_attribute_value({}, 'cost', index)
+
−            self.assertTrue(cost_value in range(index+1))
+
−
+
−    def test_date(self):
+
−        """test date generation"""
+
−        # Test for random index
+
−        for index in range(10):
+
−            date_value = self.person_valgen.generate_attribute_value({}, 'birthday', index)
+
−            self.assertTrue(isinstance(date_value, datetime.date))
+
−
+
−    def test_phone(self):
+
−        """tests make_tel utility"""
+
−        self.assertEqual(make_tel(22030405), '22 03 04 05')
+
−
+
−    def test_customized_generation(self):
+
−        self.assertEqual(self.bug_valgen.generate_attribute_value({}, 'severity', 12),
+
−                          u'dangerous')
+
−        self.assertEqual(self.bug_valgen.generate_attribute_value({}, 'description', 12),
+
−                          u'yo')
+
−        self.assertEqual(self.person_valgen.generate_attribute_value({}, 'description', 12),
+
−                          u'yo')
+
−
+
−
+
−class ConstraintInsertionTC(TestCase):
+
−
+
−    def test_writeme(self):
+
−        self.skipTest('Test automatic insertion / Schema Constraints')
+
−
+
−
+
−if __name__ == '__main__':
+
−    unittest_main()