dataimport/test/test_massive_store.py
changeset 10872 ff4f94cfa2fb
parent 10869 575982c948a9
child 10875 75d1b2d66f18
--- a/dataimport/test/test_massive_store.py	Mon Nov 09 15:29:07 2015 +0100
+++ b/dataimport/test/test_massive_store.py	Mon Nov 09 15:51:02 2015 +0100
@@ -153,8 +153,12 @@
         self.assertIn('owned_by_relation_to_idx', indexes)
 
     def test_eids_seq_range(self):
+        class MyMassiveObjectStore(MassiveObjectStore):
+            eids_seq_range = 1000
+            eids_seq_start = 50000
+
         with self.admin_access.repo_cnx() as cnx:
-            store = MassiveObjectStore(cnx, eids_seq_range=1000, eids_seq_start=50000)
+            store = MyMassiveObjectStore(cnx)
             store.prepare_insert_entity('Location', name=u'toto')
             store.flush()
             cnx.commit()
@@ -163,15 +167,23 @@
             self.assertGreater(crs.fetchone()[0], 50000)
 
     def test_eid_entity(self):
+        class MyMassiveObjectStore(MassiveObjectStore):
+            eids_seq_range = 1000
+            eids_seq_start = 50000
+
         with self.admin_access.repo_cnx() as cnx:
-            store = MassiveObjectStore(cnx, eids_seq_range=1000, eids_seq_start=50000)
+            store = MyMassiveObjectStore(cnx)
             eid = store.prepare_insert_entity('Location', name=u'toto')
             store.flush()
             self.assertGreater(eid, 50000)
 
     def test_eid_entity_2(self):
+        class MyMassiveObjectStore(MassiveObjectStore):
+            eids_seq_range = 1000
+            eids_seq_start = 50000
+
         with self.admin_access.repo_cnx() as cnx:
-            store = MassiveObjectStore(cnx, eids_seq_range=1000, eids_seq_start=50000)
+            store = MyMassiveObjectStore(cnx)
             eid = store.prepare_insert_entity('Location', name=u'toto', eid=10000)
             store.flush()
         self.assertEqual(eid, 10000)