query-events.js

require('../test')('Query events', function(conn, t) {
  conn.query('DROP TABLE IF EXISTS streaming_test')
  conn.query('CREATE TABLE streaming_test (a int)')

  // prepare our data
  var expected = []
  for (var i = 0; i < 10; i++) {
    conn.query('INSERT INTO streaming_test (a) VALUES (' + i + ')')
    expected.push({ a: i })
  }

  t.test('events with a callback', function(t) {
    var fields, gotEnd
    t.plan(4)
    conn
      .query('SELECT a FROM streaming_test', function(err, result) {
        t.deepEqual(result.fields, fields, 'got same fields in result')
        t.deepEqual(result.rows, expected, 'got expected rows')
      })
      .on('fields', function(_fields) {
        t.pass('got "fields" event')
        fields = _fields
      })
      .on('end', function() {
        t.pass('got "end" event')
      })
  })

  t.test('events with no callback (stream)', function(t) {
    var received = []
    var fields
    t.plan(2)
    conn
      .query('SELECT a FROM streaming_test')
      .on('fields', function(_fields) {
        fields = _fields
      })
      .on('data', function(row) {
        received.push(row)
      })
      .on('end', function() {
        t.ok(fields, 'got fields event')
        t.deepEqual(expected, received)
      })
  })

  t.test('cleanup', function(t) {
    conn.query('DROP TABLE streaming_test', function() {
      t.end()
    })
  })
})