Session config

Author: bigmontz

packages/bolt-connection/src/connection-provider/connection-provider-direct.js

@@ -42,28 +42,34 @@ export default class DirectConnectionProvider extends PooledConnectionProvider {
    * See {@link ConnectionProvider} for more information about this method and
    * its arguments.
    */
-  acquireConnection ({ accessMode, database, bookmarks } = {}) {
+  async acquireConnection ({ accessMode, database, bookmarks, auth } = {}) {
     const databaseSpecificErrorHandler = ConnectionErrorHandler.create({
       errorCode: SERVICE_UNAVAILABLE,
       handleAuthorizationExpired: (error, address, conn) =>
         this._handleAuthorizationExpired(error, address, conn, database)
     })
 
-    return this._connectionPool
-      .acquire(this._address)
-      .then(
-        connection =>
-          new DelegateConnection(connection, databaseSpecificErrorHandler)
-      )
+    const connection = await this._connectionPool.acquire(this._address)
+
+    if (auth && auth !== connection.authToken) {
+      if (connection.supportsReAuth) {
+        await connection.connect(this._userAgent, auth)
+      } else {
+        await connection._release()
+        return await this._createStickyConnection({ address: this._address, auth })
+      }
+    }
+
+    return new DelegateConnection(connection, databaseSpecificErrorHandler)
   }
 
   _handleAuthorizationExpired (error, address, connection, database) {
     this._log.warn(
       `Direct driver ${this._id} will close connection to ${address} for database '${database}' because of an error ${error.code} '${error.message}'`
     )
-    
+
     this._authenticationProvider.handleError({ connection, code: error.code })
-    
+
     if (error.code === 'Neo.ClientError.Security.AuthorizationExpired') {
       this._connectionPool.apply(address, (conn) => conn.authToken === null)
     }

packages/testkit-backend/src/feature/common.js

@@ -4,6 +4,7 @@ const features = [
   'Feature:Auth:Kerberos',
   'Feature:Auth:Bearer',
   'Feature:API:BookmarkManager',
+  'Feature:API:Session:AuthConfig',
   'Feature:API:SSLConfig',
   'Feature:API:SSLSchemes',
   'Feature:API:Type.Temporal',

packages/testkit-backend/src/request-handlers.js

@@ -116,14 +116,19 @@ export function NewSession (neo4j, context, data, wire) {
       return
     }
   }
+  const auth = data.authorizationToken != null
+    ? context.binder.parseAuthToken(data.authorizationToken.data)
+    : undefined
+
   const driver = context.getDriver(driverId)
   const session = driver.session({
     defaultAccessMode: accessMode,
     bookmarks,
     database,
     fetchSize,
     impersonatedUser,
-    bookmarkManager
+    bookmarkManager,
+    auth
   })
   const id = context.addSession(session)
   wire.writeResponse(responses.Session({ id }))