stress-test: always wait for bm when isCluster

Author: bigmontz

packages/neo4j-driver/test/internal/shared-neo4j.js

@@ -308,18 +308,23 @@ function restart (configOverride) {
   startNeo4j()
 }
 
-async function cleanupAndGetProtocolVersion (driver) {
+async function cleanupAndGetProtocolVersionAndBookmark (driver) {
   const session = driver.session({ defaultAccessMode: neo4j.session.WRITE })
   try {
     const result = await session.writeTransaction(tx =>
       tx.run('MATCH (n) DETACH DELETE n')
     )
-    return result.summary.server.protocolVersion
+    return [result.summary.server.protocolVersion, session.lastBookmark()]
   } finally {
     await session.close()
   }
 }
 
+async function cleanupAndGetProtocolVersion (driver) {
+  const [protocolVersion] = await cleanupAndGetProtocolVersion(driver)
+  return protocolVersion
+}
+
 async function getEdition (driver) {
   const session = driver.session({ defaultAccessMode: neo4j.session.READ })
   try {
@@ -375,6 +380,7 @@ export default {
   authToken: authToken,
   logging: debugLogging,
   cleanupAndGetProtocolVersion: cleanupAndGetProtocolVersion,
+  cleanupAndGetProtocolVersionAndBookmark,
   tlsConfig: tlsConfig,
   getEdition: getEdition,
   hostname: hostname,

packages/neo4j-driver/test/stress-test.js

@@ -74,7 +74,10 @@ export default async function execute () {
     neo4j.auth.basic(USERNAME, PASSWORD),
     config
   )
-  const protocolVersion = await sharedNeo4j.cleanupAndGetProtocolVersion(driver)
+  const [
+    protocolVersion,
+    bookmarks
+  ] = await sharedNeo4j.cleanupAndGetProtocolVersionAndBookmark(driver)
   console.time('Basic-stress-test')
   const printStats = () => {
     console.timeEnd('Basic-stress-test')
@@ -83,7 +86,12 @@ export default async function execute () {
     console.log('Write statistics: ', context.writeServersWithQueryCount)
   }
 
-  const context = new Context(driver, LOGGING_ENABLED, protocolVersion)
+  const context = new Context(
+    driver,
+    LOGGING_ENABLED,
+    protocolVersion,
+    bookmarks
+  )
 
   try {
     await runWhileNotTimeout(async () => {
@@ -525,10 +533,10 @@ function noParams () {
 }
 
 function newSession (context, accessMode, useBookmark) {
-  if (useBookmark) {
+  if (useBookmark || isCluster()) {
     return context.driver.session({
       defaultAccessMode: accessMode,
-      bookmarks: [context.bookmark]
+      bookmarks: context.bookmark
     })
   }
   return context.driver.session({ defaultAccessMode: accessMode })
@@ -556,9 +564,9 @@ function arraysEqual (array1, array2) {
 }
 
 class Context {
-  constructor (driver, loggingEnabled, protocolVersion) {
+  constructor (driver, loggingEnabled, protocolVersion, bookmark) {
     this.driver = driver
-    this.bookmark = null
+    this.bookmark = bookmark
     this.createdNodesCount = 0
     this._commandIdCouter = 0
     this._loggingEnabled = loggingEnabled