db-vendo-client/test/throttle.js

'use strict'

const tapePromise = require('tape-promise').default
const tape = require('tape')

const withThrottling = require('../throttle')
const createClient = require('..')
const vbbProfile = require('../p/vbb')
const depsRes = require('./fixtures/vbb-departures.json')

const ua = 'public-transport/hafas-client:test'
const spichernstr = '900000042101'

const test = tapePromise(tape)

test('withThrottling works', async (t) => {
	const ctx = {profile: vbbProfile, opt: {}}

	let calls = 0
	const mockedRequest = async (ctx, _, reqData) => {
		calls++
		return {
			res: depsRes,
			common: ctx.profile.parseCommon({...ctx, res: depsRes})
		}
	}

	const createThrottledClient = withThrottling(createClient, 2, 1000)
	const client = createThrottledClient(vbbProfile, ua, mockedRequest)

	t.plan(3)
	for (let i = 0; i < 10; i++) {
		const p = client.departures(spichernstr, {duration: 1})
		p.catch(() => {})
	}

	setTimeout(() => t.equal(calls, 2), 500)
	setTimeout(() => t.equal(calls, 4), 1500)
	setTimeout(() => t.equal(calls, 6), 2500)
})