Add temporal token tests

Author: bigmontz

packages/testkit-backend/src/context.js

@@ -15,6 +15,7 @@ export default class Context {
     this._authTokenManagers = {}
     this._authTokenManagerGetAuthRequests = {}
     this._authTokenManagerOnAuthExpiredRequests = {}
+    this._temporalAuthTokenProviderRequests = {}
     this._binder = binder
     this._environmentLogLevel = environmentLogLevel
   }
@@ -200,6 +201,18 @@ export default class Context {
     delete this._authTokenManagerOnAuthExpiredRequests[id]
   }
 
+  addTemporalAuthTokenProviderRequest (resolve, reject) {
+    return this._add(this._temporalAuthTokenProviderRequests, { resolve, reject })
+  }
+
+  getTemporalAuthTokenProviderRequest (id) {
+    return this._temporalAuthTokenProviderRequests[id]
+  }
+
+  removeTemporalAuthTokenProviderRequest (id) {
+    delete this._temporalAuthTokenProviderRequests[id]
+  }
+
   _add (map, object) {
     this._id++
     map[this._id] = object

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

@@ -26,9 +26,11 @@ export {
   StartSubTest,
   ExecuteQuery,
   NewAuthTokenManager,
+  NewTemporalAuthTokenManager,
   AuthTokenManagerClose,
   AuthTokenManagerGetAuthCompleted,
   AuthTokenManagerOnAuthExpiredCompleted,
+  TemporalAuthTokenProviderCompleted,
   FakeTimeInstall,
   FakeTimeTick,
   FakeTimeUninstall

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

@@ -435,7 +435,7 @@ export function ResolverResolutionCompleted (
 }
 
 export function NewBookmarkManager (
-  neo4j,
+  { neo4j },
   context,
   {
     initialBookmarks,
@@ -526,6 +526,19 @@ export function NewAuthTokenManager (_, context, _data, wire) {
   wire.writeResponse(responses.AuthTokenManager({ id }))
 }
 
+export function NewTemporalAuthTokenManager ({ neo4j }, context, _, wire) {
+  const id = context.addAuthTokenManager((temporalAuthTokenManagerId) => {
+    return neo4j.temporalAuthDataManager({
+      getAuthData: () => new Promise((resolve, reject) => {
+        const id = context.addTemporalAuthTokenProviderRequest(resolve, reject)
+        wire.writeResponse(responses.TemporalAuthTokenProviderRequest({ id, temporalAuthTokenManagerId }))
+      })
+    })
+  })
+
+  wire.writeResponse(responses.TemporalAuthTokenManager({ id }))
+}
+
 export function AuthTokenManagerClose (_, context, { id }, wire) {
   context.removeAuthTokenManager(id)
   wire.writeResponse(responses.AuthTokenManager({ id }))
@@ -537,10 +550,21 @@ export function AuthTokenManagerGetAuthCompleted (_, context, { requestId, auth
   context.removeAuthTokenManagerGetAuthRequest(requestId)
 }
 
-export function AuthTokenManagerOnAuthExpiredCompleted (_, context, { requestId, auth }) {
+export function AuthTokenManagerOnAuthExpiredCompleted (_, context, { requestId }) {
   context.removeAuthTokenManagerOnAuthExpiredRequest(requestId)
 }
 
+export function TemporalAuthTokenProviderCompleted (_, context, { requestId, auth }) {
+  const request = context.getTemporalAuthTokenProviderRequest(requestId)
+  request.resolve({
+    expiry: auth.data.expiresInMs != null
+      ? new Date(new Date().getTime() + auth.data.expiresInMs)
+      : undefined,
+    token: context.binder.parseAuthToken(auth.data.auth.data)
+  })
+  context.removeTemporalAuthTokenProviderRequest(requestId)
+}
+
 export function GetRoutingTable (_, context, { driverId, database }, wire) {
   const driver = context.getDriver(driverId)
   const routingTable =
@@ -584,7 +608,7 @@ export function ForcedRoutingTableUpdate (_, context, { driverId, database, book
   }
 }
 
-export function ExecuteQuery (neo4j, context, { driverId, cypher, params, config }, wire) {
+export function ExecuteQuery ({ neo4j }, context, { driverId, cypher, params, config }, wire) {
   const driver = context.getDriver(driverId)
   if (params) {
     for (const [key, value] of Object.entries(params)) {

packages/testkit-backend/src/responses.js

@@ -119,6 +119,14 @@ export function AuthTokenManagerOnAuthExpiredRequest ({ id, authTokenManagerId,
   return response('AuthTokenManagerOnAuthExpiredRequest', { id, authTokenManagerId, auth: AuthorizationToken(auth) })
 }
 
+export function TemporalAuthTokenManager ({ id }) {
+  return response('TemporalAuthTokenManager', { id })
+}
+
+export function TemporalAuthTokenProviderRequest ({ id, temporalAuthTokenManagerId }) {
+  return response('TemporalAuthTokenProviderRequest', { id, temporalAuthTokenManagerId })
+}
+
 export function DriverIsAuthenticated ({ id, authenticated }) {
   return response('DriverIsAuthenticated', { id, authenticated })
 }