Fixes postgres flaky test (#3822)

* Changes expected error code * nits
2017-05-16 12:06:17 -04:00
parent ab04641ed0
commit 9dbb89a2e4
3 changed files with 21 additions and 11 deletions
--- a/src/Adapters/Storage/Postgres/PostgresClient.js
+++ b/src/Adapters/Storage/Postgres/PostgresClient.js
@@ -23,5 +23,5 @@ export function createClient(uri, databaseOptions) {
    }
  }

-  return client;
+  return { client, pgp };
 }
--- a/src/Adapters/Storage/Postgres/PostgresStorageAdapter.js
+++ b/src/Adapters/Storage/Postgres/PostgresStorageAdapter.js
@@ -403,6 +403,7 @@ export class PostgresStorageAdapter {
  // Private
  _collectionPrefix: string;
  _client;
+  _pgp;

  constructor({
    uri,
@@ -410,7 +411,9 @@ export class PostgresStorageAdapter {
    databaseOptions
  }) {
    this._collectionPrefix = collectionPrefix;
-    this._client = createClient(uri, databaseOptions);
+    const { client, pgp } = createClient(uri, databaseOptions);
+    this._client = client;
+    this._pgp = pgp;
  }

  _ensureSchemaCollectionExists(conn) {