diff -r 2da2dd60331c -r 49aca289134f cubicweb/dataimport/test/test_massive_store.py --- a/cubicweb/dataimport/test/test_massive_store.py Mon Jun 20 17:59:43 2016 +0200 +++ b/cubicweb/dataimport/test/test_massive_store.py Wed Jun 22 07:57:13 2016 +0200 @@ -21,6 +21,7 @@ from cubicweb.devtools import testlib, PostgresApptestConfiguration from cubicweb.devtools import startpgcluster, stoppgcluster from cubicweb.dataimport import ucsvreader, stores +from cubicweb.server.schema2sql import build_index_name from cubicweb.dataimport.massive_store import MassiveObjectStore, PGHelper import test_stores @@ -61,7 +62,7 @@ for code, gmt, dst, raw_offset in ucsvreader(open(self.datapath('timeZones.txt'), 'rb'), delimiter='\t'): cnx.create_entity('TimeZone', code=code, gmt=float(gmt), - dst=float(dst), raw_offset=float(raw_offset)) + dst=float(dst), raw_offset=float(raw_offset)) timezone_code = dict(cnx.execute('Any C, X WHERE X is TimeZone, X code C')) # Push data for ind, infos in enumerate(ucsvreader(open(dumpname, 'rb'), @@ -139,8 +140,10 @@ indexes = [r[0] for r in crs.fetchall()] self.assertIn('entities_pkey', indexes) self.assertIn('entities_extid_idx', indexes) - self.assertIn('owned_by_relation_p_key', indexes) - self.assertIn('owned_by_relation_to_idx', indexes) + self.assertIn(build_index_name('owned_by_relation', ['eid_from', 'eid_to'], 'key_'), + indexes) + self.assertIn(build_index_name('owned_by_relation', ['eid_from'], 'idx_'), + indexes) def test_eids_seq_range(self): with self.admin_access.repo_cnx() as cnx: @@ -219,8 +222,10 @@ indexes = [r[0] for r in crs.fetchall()] self.assertIn('entities_pkey', indexes) self.assertIn('entities_extid_idx', indexes) - self.assertIn('owned_by_relation_p_key', indexes) - self.assertIn('owned_by_relation_to_idx', indexes) + self.assertIn(build_index_name('owned_by_relation', ['eid_from', 'eid_to'], 'key_'), + indexes) + self.assertIn(build_index_name('owned_by_relation', ['eid_from'], 'idx_'), + indexes) def test_slave_mode_exception(self): with self.admin_access.repo_cnx() as cnx: @@ -252,8 +257,10 @@ indexes = [r[0] for r in crs.fetchall()] self.assertNotIn('entities_pkey', indexes) self.assertNotIn('entities_extid_idx', indexes) - self.assertNotIn('owned_by_relation_p_key', indexes) - self.assertNotIn('owned_by_relation_to_idx', indexes) + self.assertNotIn(build_index_name('owned_by_relation', ['eid_from', 'eid_to'], 'key_'), + indexes) + self.assertNotIn(build_index_name('owned_by_relation', ['eid_from'], 'idx_'), + indexes) # Cleanup -> index store.finish() @@ -263,8 +270,10 @@ indexes = [r[0] for r in crs.fetchall()] self.assertIn('entities_pkey', indexes) self.assertIn('entities_extid_idx', indexes) - self.assertIn('owned_by_relation_p_key', indexes) - self.assertIn('owned_by_relation_to_idx', indexes) + self.assertIn(build_index_name('owned_by_relation', ['eid_from', 'eid_to'], 'key_'), + indexes) + self.assertIn(build_index_name('owned_by_relation', ['eid_from'], 'idx_'), + indexes) def test_multiple_insert(self): with self.admin_access.repo_cnx() as cnx: