Make all integration tests pass

brianc · brianc · commit 0ed7b57e37a3 · 2017-06-10T16:47:46.000-05:00
diff --git a/test/integration/domain-tests.js b/test/integration/domain-tests.js
@@ -1,6 +1,8 @@
-var helper = require('./test-helper')
 var async = require('async')
 
+var helper = require('./test-helper')
+var Query = helper.pg.Query
+
 var testWithoutDomain = function(cb) {
   test('no domain', function() {
     assert(!process.domain)
@@ -42,7 +44,7 @@ var testErrorWithDomain = function(cb) {
     })
     domain.run(function() {
       helper.pg.connect(helper.config, assert.success(function(client, done) {
-        client.query('SELECT SLDKJFLSKDJF')
+        client.query(new Query('SELECT SLDKJFLSKDJF'))
         client.on('drain', done)
       }))
     })
