panggilan balik perintah khusus nightwatch

Saya harus membungkus koneksi database dalam perintah perform agar ini berfungsi. Saya tidak yakin apakah ini cara terbaik untuk menangani panggilan balik, tetapi berhasil. Inilah versi terbaru dari perintah kustom:

exports.command = function(sql,callback) {
  var self = this;
  var pg = require('pg');
  var cs = self.globals.testinfo.connectionString;
  self.perform(function(self,done) {
    pg.connect(cs,function(err,db,done) {
      if(err) {
        return console.error(err);
      }  
      db.query(sql, function(err,result) {
        done();
        if(err) {
          return console.error(err);
        } 
        console.log(result.rows.length);
        callback(result.rows[0]);
      });
    });
    pg.end();
    done();
  });
};

Inilah cara saya memanggil perintah khusus dalam pengujian:

browser.myCustomCommand('select * from table limit 1;', function(row) {
  browser.assert.deepEqual(row.column,'some value');
});