Fixing mock.

4 years ago · 5b1ae1f6fc
1 changed files with 3 additions and 1 deletions
--- a/packages/server/mocks/pg.ts
+++ b/packages/server/mocks/pg.ts
@ -14,7 +14,9 @@ module PgMock {
  function Client() {}

  Client.prototype.query = query
-  Client.prototype.end = jest.fn()
+  Client.prototype.end = jest.fn(cb => {
+    if (cb) cb()
+  })
  Client.prototype.connect = jest.fn()
  Client.prototype.release = jest.fn()