Merge pull request #416 from lutovich/1.7-flaky-tests

Fix flaky tests

Author: lutovich

test/browser/karma-firefox.conf.js

@@ -34,6 +34,9 @@ module.exports = function(config) {
       FirefoxHeadless: {
         base: 'Firefox',
         flags: [ '-headless' ],
+        prefs: {
+          'network.websocket.max-connections': 256 // as in Chrome
+        }
       },
     },
   })

test/internal/connection.test.js

@@ -24,15 +24,14 @@ import {Chunker} from '../../src/v1/internal/chunking';
 import {alloc} from '../../src/v1/internal/buf';
 import {Neo4jError, newError, SERVICE_UNAVAILABLE} from '../../src/v1/error';
 import sharedNeo4j from '../internal/shared-neo4j';
-import {ServerVersion} from '../../src/v1/internal/server-version';
+import {ServerVersion, VERSION_3_5_0} from '../../src/v1/internal/server-version';
 import lolex from 'lolex';
 import Logger from '../../src/v1/internal/logger';
 import StreamObserver from '../../src/v1/internal/stream-observer';
 import ConnectionErrorHandler from '../../src/v1/internal/connection-error-handler';
 import testUtils from '../internal/test-utils';
 import Bookmark from '../../src/v1/internal/bookmark';
 import TxConfig from '../../src/v1/internal/tx-config';
-import boltStub from '../internal/bolt-stub';
 
 const ILLEGAL_MESSAGE = {signature: 42, fields: []};
 const SUCCESS_MESSAGE = {signature: 0x70, fields: [{}]};
@@ -347,27 +346,25 @@ describe('Connection', () => {
     connection._handleFatalError(newError('Hello', SERVICE_UNAVAILABLE));
   });
 
-  it('should send hello and goodbye messages', done => {
-    if (!boltStub.supported) {
-      done();
-      return;
-    }
+  it('should send INIT/HELLO and GOODBYE messages', done => {
+    const messages = [];
+    connection = createConnection('bolt://localhost');
+    recordWrittenMessages(connection, messages);
 
-    const server = boltStub.start('./test/resources/boltstub/hello_goodbye.script', 9001);
-
-    boltStub.run(() => {
-      connection = createConnection('bolt://127.0.0.1:9001', {encrypted: false});
-      connection.connect('single-connection/1.2.3', basicAuthToken())
-        .then(() => {
-          connection.close(() => {
-            server.exit(code => {
-              expect(code).toEqual(0);
-              done();
-            });
-          });
-        })
-        .catch(error => done.fail(error));
-    });
+    connection.connect('mydriver/0.0.0', basicAuthToken())
+      .then(() => {
+        expect(connection.isOpen()).toBeTruthy();
+        connection.close(() => {
+          expect(messages.length).toBeGreaterThan(0);
+          expect(messages[0].signature).toEqual(0x01); // first message is either INIT or HELLO
+
+          const serverVersion = ServerVersion.fromString(connection.server.version);
+          if (serverVersion.compareTo(VERSION_3_5_0) >= 0) {
+            expect(messages[messages.length - 1].signature).toEqual(0x02); // last message is GOODBYE in V3
+          }
+          done();
+        });
+      }).catch(done.fail);
   });
 
   function packedHandshakeMessage() {
@@ -446,4 +443,12 @@ describe('Connection', () => {
     return Connection.create(url, config || {}, new ConnectionErrorHandler(errorCode || SERVICE_UNAVAILABLE), Logger.noOp());
   }
 
+  function recordWrittenMessages(connection, messages) {
+    const originalWrite = connection.write.bind(connection);
+    connection.write = (message, observer, flush) => {
+      messages.push(message);
+      originalWrite(message, observer, flush);
+    };
+  }
+
 });