Update test to use newly introduced RecordStorageIdController to acce…

…ss id into from record storage engine.
neo4j · Jul 12, 2016 · 91ee61d · 91ee61d
1 parent 4d46a1c
commit 91ee61d
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/enterprise/kernel/src/test/java/org/neo4j/graphdb/store/id/RelationshipIdReuseStressIT.java b/enterprise/kernel/src/test/java/org/neo4j/graphdb/store/id/RelationshipIdReuseStressIT.java
@@ -51,7 +51,7 @@
 import org.neo4j.helpers.collection.Iterators;
 import org.neo4j.kernel.DeadlockDetectedException;
 import org.neo4j.kernel.impl.enterprise.configuration.EnterpriseEditionSettings;
-import org.neo4j.kernel.impl.store.NeoStores;
+import org.neo4j.kernel.impl.storageengine.impl.recordstorage.id.RecordStorageIdController;
 import org.neo4j.kernel.impl.store.id.IdType;
 import org.neo4j.test.EmbeddedDatabaseRule;
 
@@ -115,8 +115,8 @@ public void relationshipIdReused() throws Exception
 
     private long getHighestUsedIdForRelationships()
     {
-        NeoStores neoStores = embeddedDatabase.getDependencyResolver().resolveDependency( NeoStores.class );
+        RecordStorageIdController idController = embeddedDatabase.getDependencyResolver().resolveDependency( RecordStorageIdController.class );
-        return neoStores.getRelationshipStore().getHighestPossibleIdInUse();
+        return idController.getIdGeneratorFactory().get( IdType.RELATIONSHIP ).getHighestPossibleIdInUse();
     }
 
     private void completeFutures( List<Future> futures )