workq/test/async-await.js

A super tiny work queue

'use strict'

const sleep = require('then-sleep')

function asyncAwait (Queue, test) {
  test('The order should be guaranteed', t => {
    t.plan(5)

    const q = Queue()
    const order = [1, 2, 3, 4, 5]

    q.add(async q => {
      t.is(order.shift(), 1)
    })

    q.add(async q => {
      t.is(order.shift(), 2)

      q.add(async q => {
        t.is(order.shift(), 3)
      })
    })

    q.add(async q => {
      t.is(order.shift(), 4)
    })

    q.add(async q => {
      t.is(order.shift(), 5)
    })
  })

  test('Nested child queue, order should be guaranteed', t => {
    t.plan(7)

    const q = Queue()
    const order = [1, 2, 3, 4, 5, 6, 7]

    q.add(async q => {
      t.is(order.shift(), 1)
    })

    q.add(async q => {
      t.is(order.shift(), 2)

      q.add(async q => {
        q.add(async q => {
          t.is(order.shift(), 4)
        })

        await sleep(100)
        t.is(order.shift(), 3)

        q.add(async q => {
          t.is(order.shift(), 5)
        })
      })

      q.add(async q => {
        t.is(order.shift(), 6)
      })
    })

    q.add(async q => {
      t.is(order.shift(), 7)
    })
  })

  test('Drain hook', t => {
    t.plan(7)

    const q = Queue()
    const order = [1, 2, 3, 4, 5]

    q.drain(async () => {
      await sleep(100)
      t.ok('called')
    })

    q.add(async q => {
      t.is(order.shift(), 1)
    })

    q.add(async q => {
      t.is(order.shift(), 2)

      q.add(async q => {
        t.is(order.shift(), 3)
      })

      q.drain(async () => {
        await sleep(100)
        t.ok('called')
      })
    })

    q.add(async q => {
      t.is(order.shift(), 4)
    })

    q.add(async q => {
      t.is(order.shift(), 5)
    })
  })

  test('Should handle parameters', t => {
    t.plan(2)

    const q = Queue()

    async function worker (q, a, b) {
      t.strictEqual(a, 'a')
      t.strictEqual(b, 'b')
    }

    q.add(worker, 'a', 'b')
  })
}

module.exports = asyncAwait