[dataimport] Move stores to new API.
Here is the final store API:
* ``prepare_insert_entity(<entity type>, **kwargs) -> eid``: given an entity
type, attributes and inlined relations, return the eid of the entity to be
inserted, *with no guarantee that anything has been inserted in database*,
* ``prepare_update_entity(<entity type>, eid, **kwargs) -> None``: given an
entity type and eid, promise for update given attributes and inlined
relations *with no guarantee that anything has been inserted in database*,
* ``prepare_insert_relation(eid_from, rtype, eid_to) -> None``: indicate that a
relation ``rtype`` should be added between entities with eids ``eid_from``
and ``eid_to``. Similarly to ``prepare_insert_entity()``, *there is no
guarantee that the relation will be inserted in database*,
* ``flush() -> None``: flush any temporary data to database. May be called
several times during an import,
* ``finish() -> None``: additional stuff to do after import is terminated.
**Warning:** ``prepare_update_entity()`` still needs to be implemented for
NoHookRQLObjectStore.
Related to #5040344
# copyright 2003-2015 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/>.
import os.path as osp
from datetime import datetime
from threading import Thread
from logilab.common.testlib import SkipTest
from cubicweb import ValidationError
from cubicweb.devtools import PostgresApptestConfiguration, startpgcluster, stoppgcluster
from cubicweb.devtools.testlib import CubicWebTC
from cubicweb.predicates import is_instance
from cubicweb.entities.adapters import IFTIndexableAdapter
from unittest_querier import FixedOffset
def setUpModule():
startpgcluster(__file__)
def tearDownModule():
stoppgcluster(__file__)
class PostgresFTITC(CubicWebTC):
configcls = PostgresApptestConfiguration
def test_eid_range(self):
# concurrent allocation of eid ranges
source = self.session.repo.sources_by_uri['system']
range1 = []
range2 = []
def allocate_eid_ranges(session, target):
for x in xrange(1, 10):
eid = source.create_eid(session, count=x)
target.extend(range(eid-x, eid))
t1 = Thread(target=lambda: allocate_eid_ranges(self.session, range1))
t2 = Thread(target=lambda: allocate_eid_ranges(self.session, range2))
t1.start()
t2.start()
t1.join()
t2.join()
self.assertEqual(range1, sorted(range1))
self.assertEqual(range2, sorted(range2))
self.assertEqual(set(), set(range1) & set(range2))
def test_occurence_count(self):
with self.admin_access.repo_cnx() as cnx:
c1 = cnx.create_entity('Card', title=u'c1',
content=u'cubicweb cubicweb cubicweb')
c2 = cnx.create_entity('Card', title=u'c3',
content=u'cubicweb')
c3 = cnx.create_entity('Card', title=u'c2',
content=u'cubicweb cubicweb')
cnx.commit()
self.assertEqual(cnx.execute('Card X ORDERBY FTIRANK(X) DESC '
'WHERE X has_text "cubicweb"').rows,
[[c1.eid,], [c3.eid,], [c2.eid,]])
def test_attr_weight(self):
class CardIFTIndexableAdapter(IFTIndexableAdapter):
__select__ = is_instance('Card')
attr_weight = {'title': 'A'}
with self.temporary_appobjects(CardIFTIndexableAdapter):
with self.admin_access.repo_cnx() as cnx:
c1 = cnx.create_entity('Card', title=u'c1',
content=u'cubicweb cubicweb cubicweb')
c2 = cnx.create_entity('Card', title=u'c2',
content=u'cubicweb cubicweb')
c3 = cnx.create_entity('Card', title=u'cubicweb',
content=u'autre chose')
cnx.commit()
self.assertEqual(cnx.execute('Card X ORDERBY FTIRANK(X) DESC '
'WHERE X has_text "cubicweb"').rows,
[[c3.eid,], [c1.eid,], [c2.eid,]])
def test_entity_weight(self):
class PersonneIFTIndexableAdapter(IFTIndexableAdapter):
__select__ = is_instance('Personne')
entity_weight = 2.0
with self.temporary_appobjects(PersonneIFTIndexableAdapter):
with self.admin_access.repo_cnx() as cnx:
c1 = cnx.create_entity('Personne', nom=u'c1', prenom=u'cubicweb')
c2 = cnx.create_entity('Comment', content=u'cubicweb cubicweb',
comments=c1)
c3 = cnx.create_entity('Comment', content=u'cubicweb cubicweb cubicweb',
comments=c1)
cnx.commit()
self.assertEqual(cnx.execute('Any X ORDERBY FTIRANK(X) DESC '
'WHERE X has_text "cubicweb"').rows,
[[c1.eid,], [c3.eid,], [c2.eid,]])
def test_tz_datetime(self):
with self.admin_access.repo_cnx() as cnx:
cnx.execute("INSERT Personne X: X nom 'bob', X tzdatenaiss %(date)s",
{'date': datetime(1977, 6, 7, 2, 0, tzinfo=FixedOffset(1))})
datenaiss = cnx.execute("Any XD WHERE X nom 'bob', X tzdatenaiss XD")[0][0]
self.assertEqual(datenaiss.tzinfo, None)
self.assertEqual(datenaiss.utctimetuple()[:5], (1977, 6, 7, 1, 0))
cnx.commit()
cnx.execute("INSERT Personne X: X nom 'boby', X tzdatenaiss %(date)s",
{'date': datetime(1977, 6, 7, 2, 0)})
datenaiss = cnx.execute("Any XD WHERE X nom 'boby', X tzdatenaiss XD")[0][0]
self.assertEqual(datenaiss.tzinfo, None)
self.assertEqual(datenaiss.utctimetuple()[:5], (1977, 6, 7, 2, 0))
def test_constraint_validationerror(self):
with self.admin_access.repo_cnx() as cnx:
with cnx.allow_all_hooks_but('integrity'):
with self.assertRaises(ValidationError) as cm:
cnx.execute("INSERT Note N: N type 'nogood'")
self.assertEqual(cm.exception.errors,
{'type-subject': u'invalid value %(KEY-value)s, it must be one of %(KEY-choices)s'})
self.assertEqual(cm.exception.msgargs,
{'type-subject-value': u'"nogood"',
'type-subject-choices': u'"todo", "a", "b", "T", "lalala"'})
class PostgresLimitSizeTC(CubicWebTC):
configcls = PostgresApptestConfiguration
def test(self):
with self.admin_access.repo_cnx() as cnx:
def sql(string):
return cnx.system_sql(string).fetchone()[0]
yield self.assertEqual, sql("SELECT limit_size('<p>hello</p>', 'text/html', 20)"), \
'<p>hello</p>'
yield self.assertEqual, sql("SELECT limit_size('<p>hello</p>', 'text/html', 2)"), \
'he...'
yield self.assertEqual, sql("SELECT limit_size('<br/>hello', 'text/html', 2)"), \
'he...'
yield self.assertEqual, sql("SELECT limit_size('<span class=\"1\">he</span>llo', 'text/html', 2)"), \
'he...'
yield self.assertEqual, sql("SELECT limit_size('<span>a>b</span>', 'text/html', 2)"), \
'a>...'
if __name__ == '__main__':
from logilab.common.testlib import unittest_main
unittest_main()