diff -r 7357b1485795 -r 8e1f6de61300 dataimport/test/test_massive_store.py --- a/dataimport/test/test_massive_store.py Wed Oct 21 16:31:05 2015 +0200 +++ b/dataimport/test/test_massive_store.py Wed Oct 21 16:32:11 2015 +0200 @@ -79,7 +79,7 @@ 'cwuri': u'http://sws.geonames.org/%s/' % int(infos[0]), 'geonameid': int(infos[0]), } - store.create_entity('Location', **entity) + store.prepare_insert_entity('Location', **entity) def test_autoflush_metadata(self): with self.admin_access.repo_cnx() as cnx: @@ -87,7 +87,7 @@ {'t': 'Location'}) self.assertEqual(len(crs.fetchall()), 0) store = MassiveObjectStore(cnx, autoflush_metadata=True) - store.create_entity('Location', name=u'toto') + store.prepare_insert_entity('Location', name=u'toto') store.flush() store.commit() store.cleanup() @@ -104,7 +104,7 @@ # self.assertEqual(len(crs.fetchall()), 0) # with self.admin_access.repo_cnx() as cnx: # store = MassiveObjectStore(cnx, autoflush_metadata=False) -# store.create_entity('Location', name=u'toto') +# store.prepare_insert_entity('Location', name=u'toto') # store.flush() # store.commit() # crs = cnx.system_sql('SELECT * FROM entities WHERE type=%(t)s', @@ -119,8 +119,8 @@ def test_massimport_etype_metadata(self): with self.admin_access.repo_cnx() as cnx: store = MassiveObjectStore(cnx) - timezone = store.create_entity('TimeZone') - store.create_entity('Location', timezone=timezone.eid) + timezone_eid = store.prepare_insert_entity('TimeZone') + store.prepare_insert_entity('Location', timezone=timezone_eid) store.flush() store.commit() eid, etname = cnx.execute('Any X, TN WHERE X timezone TZ, X is T, ' @@ -167,7 +167,7 @@ def test_eids_seq_range(self): with self.admin_access.repo_cnx() as cnx: store = MassiveObjectStore(cnx, eids_seq_range=1000, eids_seq_start=50000) - store.create_entity('Location', name=u'toto') + store.prepare_insert_entity('Location', name=u'toto') store.flush() cnx.commit() with self.admin_access.repo_cnx() as cnx: @@ -177,23 +177,22 @@ def test_eid_entity(self): with self.admin_access.repo_cnx() as cnx: store = MassiveObjectStore(cnx, eids_seq_range=1000, eids_seq_start=50000) - entity = store.create_entity('Location', name=u'toto') + eid = store.prepare_insert_entity('Location', name=u'toto') store.flush() - self.assertGreater(entity.eid, 50000) + self.assertGreater(eid, 50000) def test_eid_entity_2(self): with self.admin_access.repo_cnx() as cnx: store = MassiveObjectStore(cnx, eids_seq_range=1000, eids_seq_start=50000) - entity = store.create_entity('Location', name=u'toto', eid=10000) + eid = store.prepare_insert_entity('Location', name=u'toto', eid=10000) store.flush() - with self.admin_access.repo_cnx() as cnx: - self.assertEqual(entity.eid, 10000) + self.assertEqual(eid, 10000) def test_on_commit_callback(self): counter = itertools.count() with self.admin_access.repo_cnx() as cnx: store = MassiveObjectStore(cnx, on_commit_callback=lambda:next(counter)) - store.create_entity('Location', name=u'toto') + store.prepare_insert_entity('Location', name=u'toto') store.flush() store.commit() self.assertGreaterEqual(next(counter), 1) @@ -202,7 +201,7 @@ counter = itertools.count() with self.admin_access.repo_cnx() as cnx: store = MassiveObjectStore(cnx, on_rollback_callback=lambda *_: next(counter)) - store.create_entity('Location', nm='toto') + store.prepare_insert_entity('Location', nm='toto') store.flush() store.commit() self.assertGreaterEqual(next(counter), 1)