From c8ca33984dd59eea9cb2ac9808b7fc5af26cc3bb Mon Sep 17 00:00:00 2001
From: Kristian Aurlien <kristian.aurlien@cognite.com>
Date: Fri, 22 Jan 2021 13:49:32 +0100
Subject: [PATCH] add parameter to configure session in assertSchema()

---
 src/index.js | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/src/index.js b/src/index.js
index 099fae19..196b7be3 100644
--- a/src/index.js
+++ b/src/index.js
@@ -318,11 +318,12 @@ export const assertSchema = ({
   driver,
   schema,
   dropExisting = true,
-  debug = false
+  debug = false,
+  sessionParams = {}
 }) => {
   const statement = schemaAssert({ schema, dropExisting });
   const executeQuery = driver => {
-    const session = driver.session();
+    const session = driver.session(sessionParams);
     return session
       .writeTransaction(tx =>
         tx.run(statement).then(result => {