Couple unit tests for connection providers

To assert that both `DirectConnectionProvider` and `LoadBalancer`
return promises with initialized connections.

Author: lutovich

test/internal/connection-providers.test.js

@@ -44,6 +44,16 @@ describe('DirectConnectionProvider', () => {
     });
   });
 
+  it('returns an initialized connection', done => {
+    const pool = newPool();
+    const connectionProvider = newDirectConnectionProvider('localhost:123', pool);
+
+    connectionProvider.acquireConnection(READ).then(connection => {
+      expect(connection.initialized).toBeTruthy();
+      done();
+    });
+  });
+
 });
 
 describe('LoadBalancer', () => {
@@ -1049,6 +1059,26 @@ describe('LoadBalancer', () => {
     });
   });
 
+  it('returns an initialized connection', done => {
+    const pool = newPool();
+    const loadBalancer = newLoadBalancer(
+      ['server-1', 'server-2'],
+      ['server-3', 'server-4'],
+      ['server-5', 'server-6'],
+      pool
+    );
+
+    loadBalancer.acquireConnection(READ).then(connection => {
+      expect(connection.initialized).toBeTruthy();
+
+      loadBalancer.acquireConnection(WRITE).then(connection => {
+        expect(connection.initialized).toBeTruthy();
+
+        done();
+      });
+    });
+  });
+
 });
 
 function newDirectConnectionProvider(address, pool) {
@@ -1126,13 +1156,15 @@ class FakeConnection {
   constructor(address, release) {
     this.address = address;
     this.release = release;
+    this.initialized = false;
   }
 
   static create(address, release) {
     return new FakeConnection(address, release);
   }
 
   initializationCompleted() {
+    this.initialized = true;
     return Promise.resolve(this);
   }
 }