db-vendo-client/test/throttle.js

// todo: use import assertions once they're supported by Node.js & ESLint
// https://github.com/tc39/proposal-import-assertions
import {createRequire} from 'module'
const require = createRequire(import.meta.url)

import tap from 'tap'

import {createClient} from '../index.js'
import {withThrottling} from '../throttle.js'
import {profile as vbbProfile} from '../p/vbb/index.js'
const depsRes = require('./fixtures/vbb-departures.json')

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

tap.test('withThrottling works', {timeout: 3000}, (t) => {
	let calls = 0
	const mockedRequest = async (ctx, userAgent, reqData) => {
		calls++
		return {
			res: depsRes,
			common: ctx.profile.parseCommon({...ctx, res: depsRes})
		}
	}

	const profile = withThrottling({
		...vbbProfile,
		request: mockedRequest
	}, 2, 1000)
	const client = createClient(profile, ua)

	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)
})