stress-test:Fix check if the ssl scheme is set in the url

bigmontz · bigmontz · commit ba439b3b90b9 · 2022-05-04T09:59:43.000-03:00
diff --git a/packages/neo4j-driver/test/stress-test.js b/packages/neo4j-driver/test/stress-test.js
@@ -17,6 +17,7 @@
  * limitations under the License.
  */
 
+import { internal } from 'neo4j-driver-core'
 import neo4j from '../src'
 import { READ, WRITE } from '../src/driver'
 import parallelLimit from 'async/parallelLimit'
@@ -136,19 +137,9 @@ function isRemoteCluster () {
   return fromEnvOrDefault('STRESS_TEST_DATABASE_URI') !== undefined
 }
 
-function isSslSchemeNotSet (uri) {
-  function extractScheme (scheme) {
-    if (scheme) {
-      scheme = scheme.trim()
-      if (scheme.charAt(scheme.length - 1) === ':') {
-        scheme = scheme.substring(0, scheme.length - 1)
-      }
-      return scheme
-    }
-    return null
-  }
-  const scheme = extractScheme(uri)
-  return scheme === null || scheme === 'bolt' || scheme === 'neo4j'
+function isSslSchemeNotSet (url) {
+  const parsedUri = internal.urlUtil.parseDatabaseUrl(url)
+  return parsedUri.scheme === null || parsedUri.scheme === 'bolt' || parsedUri.scheme === 'neo4j'
 }
 
 function isCluster () {
