Add support for testkit

Author: bigmontz

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

@@ -17,11 +17,14 @@ const features = [
   'Feature:Bolt:4.4',
   'Feature:Bolt:5.0',
   'Feature:Bolt:5.1',
+  'Feature:Bolt:5.2',
   'Feature:Bolt:Patch:UTC',
   'Feature:API:ConnectionAcquisitionTimeout',
   'Feature:API:Driver.ExecuteQuery',
+  'Feature:API:Driver:NotificationsConfig',
   'Feature:API:Driver:GetServerInfo',
   'Feature:API:Driver.VerifyConnectivity',
+  'Feature:API:Session:NotificationsConfig',
   'Optimization:EagerTransactionBegin',
   'Optimization:ImplicitDefaultArguments',
   'Optimization:MinimalBookmarksSet',

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

@@ -46,14 +46,22 @@ export function NewSession (neo4j, context, data, wire) {
       return
     }
   }
+  let notificationFilter
+  if ('notificationsMinSeverity' in data || 'notificationsDisabledCategories' in data) {
+    notificationFilter = {
+      minimumSeverityLevel: data.notificationsMinSeverity,
+      disabledCategories: data.notificationsDisabledCategories
+    }
+  }
   const driver = context.getDriver(driverId)
   const session = driver.rxSession({
     defaultAccessMode: accessMode,
     bookmarks,
     database,
     fetchSize,
     impersonatedUser,
-    bookmarkManager
+    bookmarkManager,
+    notificationFilter
   })
   const id = context.addSession(session)
   wire.writeResponse(responses.Session({ id }))

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

@@ -39,6 +39,7 @@ export function NewDriver (neo4j, context, data, wire) {
         authToken.parameters
       )
   }
+
   const resolver = resolverRegistered
     ? address =>
         new Promise((resolve, reject) => {
@@ -83,6 +84,12 @@ export function NewDriver (neo4j, context, data, wire) {
   if ('maxTxRetryTimeMs' in data) {
     config.maxTransactionRetryTime = data.maxTxRetryTimeMs
   }
+  if ('notificationsMinSeverity' in data || 'notificationsDisabledCategories' in data) {
+    config.notificationFilter = {
+      minimumSeverityLevel: data.notificationsMinSeverity,
+      disabledCategories: data.notificationsDisabledCategories
+    }
+  }
   let driver
   try {
     driver = neo4j.driver(uri, parsedAuthToken, config)
@@ -126,14 +133,22 @@ export function NewSession (neo4j, context, data, wire) {
       return
     }
   }
+  let notificationFilter
+  if ('notificationsMinSeverity' in data || 'notificationsDisabledCategories' in data) {
+    notificationFilter = {
+      minimumSeverityLevel: data.notificationsMinSeverity,
+      disabledCategories: data.notificationsDisabledCategories
+    }
+  }
   const driver = context.getDriver(driverId)
   const session = driver.session({
     defaultAccessMode: accessMode,
     bookmarks,
     database,
     fetchSize,
     impersonatedUser,
-    bookmarkManager
+    bookmarkManager,
+    notificationFilter
   })
   const id = context.addSession(session)
   wire.writeResponse(responses.Session({ id }))

packages/testkit-backend/src/summary-binder.js

@@ -42,6 +42,7 @@ function mapProfile (profile, child = false, binder) {
 function mapNotification (notification) {
   return {
     ...notification,
+    rawCategory: notification.rawCategory || '',
     position: Object.keys(notification.position).length !== 0 ? notification.position : undefined
   }
 }