node-postgres/test/index.js

var assert = require('assert')
var pgCursor = require('../')
var gonna = require('gonna')

var text = 'SELECT generate_series as num FROM generate_series(0, 5)'
var values = []

var test = function(name, fn, timeout) {
  timeout = timeout || 1000
  var done = gonna(name, timeout, function(err) {
    console.log(name)
    assert.ifError(err)
  })
  fn(done)
}

test('fetch 6 when asking for 10', function(done) {
  var cursor = pgCursor(text)
  cursor.read(10, function(err, res) {
    assert.ifError(err)
    assert.equal(res.length, 6)
    done()
  })
})

test('end before reading to end', function(done) {
  var cursor = pgCursor(text)
  cursor.read(3, function(err, res) {
    assert.equal(res.length, 3)
    cursor.end(done)
  })
})

test('callback with error', function(done) {
  var cursor = pgCursor('select asdfasdf')
  cursor.read(1, function(err) {
    assert(err)
    done()
  })
})


test('read a partial chunk of data', function(done) {
  var cursor = pgCursor(text)
  cursor.read(2, function(err, res) {
    assert.equal(res.length, 2)
    cursor.read(3, function(err, res) {
      assert.equal(res.length, 3)
      cursor.read(1, function(err, res) {
        assert.equal(res.length, 1)
        cursor.read(1, function(err, res) {
          assert.ifError(err)
          assert.strictEqual(res.length, 0)
          done()
        })
      })
    })
  })
})

test('read return length 0 past the end', function(done) {
  var cursor = pgCursor(text)
  cursor.read(2, function(err, res) {
    cursor.read(100, function(err, res) {
      assert.equal(res.length, 4)
      cursor.read(100, function(err, res) {
        assert.equal(res.length, 0)
        done()
      })
    })
  })
})

test('read huge result', function(done) {
  var text = 'SELECT generate_series as num FROM generate_series(0, 1000000)'
  var values = []
  cursor = pgCursor(text, values);
  var count = 0;
  var read = function() {
    cursor.read(1000, function(err, rows) {
      if(err) return done(err);
      if(!rows.length) {
        assert.equal(count, 1000001)
        return done()
      }
      count += rows.length;
      if(count%100000 == 0) {
        console.log(count)
      }
      setImmediate(read)
    })
  }
  read()
}, 100000)