Rename TestSessions to SessionRule

community/bolt/src/test/java/org/neo4j/bolt/v1/runtime/integration/SessionAuthIT.java

@@ -37,7 +37,7 @@
 public class SessionAuthIT
 {
     @Rule
-    public TestSessions env = new TestSessions().withAuthEnabled( true );
+    public SessionRule env = new SessionRule().withAuthEnabled( true );
 
     @Test
     public void shouldGiveCredentialsExpiredStatusOnExpiredCredentials() throws Throwable

community/bolt/src/test/java/org/neo4j/bolt/v1/runtime/integration/SessionIT.java

@@ -56,7 +56,7 @@ public class SessionIT
     private static final Map<String,Object> EMPTY_PARAMS = emptyMap();
 
     @Rule
-    public TestSessions env = new TestSessions();
+    public SessionRule env = new SessionRule();
     private final RecordingCallback responses = new RecordingCallback<>();
     private final RecordingCallback<RecordStream, ?> pulling = new RecordingCallback<>();
     private final RecordingCallback<Void, ?> discarding = new RecordingCallback<>();

community/bolt/src/test/java/org/neo4j/bolt/v1/runtime/integration/TestSessions.java → community/bolt/src/test/java/org/neo4j/bolt/v1/runtime/integration/SessionRule.java

@@ -48,7 +48,7 @@
 import org.neo4j.test.TestGraphDatabaseFactory;
 import org.neo4j.udc.UsageData;
 
-class TestSessions implements TestRule, Sessions
+class SessionRule implements TestRule, Sessions
 {
     private GraphDatabaseAPI gdb;
     private Sessions actual;
@@ -109,7 +109,7 @@ public Session newSession( String connectionDescriptor, boolean isEncrypted )
         return session;
     }
 
-    TestSessions withAuthEnabled( boolean authEnabled )
+    SessionRule withAuthEnabled( boolean authEnabled )
     {
         this.authEnabled = authEnabled;
         return this;
@@ -126,6 +126,11 @@ URL putTmpFile( String prefix, String suffix, String contents ) throws IOExcepti
         return tmpFile.toURI().toURL();
     }
 
+    public GraphDatabaseAPI graph()
+    {
+        return gdb;
+    }
+
     long lastClosedTxId()
     {
         return gdb.getDependencyResolver().resolveDependency( TransactionIdStore.class ).getLastClosedTransactionId();

community/bolt/src/test/java/org/neo4j/bolt/v1/runtime/integration/TransactionIT.java

@@ -35,7 +35,7 @@ public class TransactionIT
     private static final Map<String, Object> EMPTY_PARAMS = Collections.emptyMap();
 
     @Rule
-    public TestSessions env = new TestSessions();
+    public SessionRule env = new SessionRule();
 
     @Test
     public void shouldHandleBeginCommit() throws Throwable