test for retry ✅

retry.js

@@ -2,7 +2,7 @@
 
 const retry = require('p-retry')
 
-const request = require('./lib/request')
+const _request = require('./lib/request')
 const createClient = require('.')
 
 const defaultRetryOpts = {
@@ -11,7 +11,7 @@ const defaultRetryOpts = {
 	minTimeout: 5 * 1000
 }
 
-const createClientWithRetry = (profile, userAgent, retryOpts = defaultRetryOpts) => {
+const createClientWithRetry = (profile, userAgent, retryOpts = defaultRetryOpts, request = _request) => {
 	const requestWithRetry = (profile, userAgent, opt, data) => {
 		const attempt = () => {
 			return request(profile, userAgent, opt, data)

test/index.js

@@ -11,3 +11,4 @@ require('./cmta')
 require('./sbahn-muenchen')
 require('./saarfahrplan')
 require('./throttle')
+require('./retry')

test/retry.js

@@ -0,0 +1,47 @@
+'use strict'
+
+const test = require('tape')
+
+const createClientWithRetry = require('../retry')
+const vbbProfile = require('../p/vbb')
+const _request = require('../lib/request')
+
+const userAgent = 'public-transport/hafas-client:test'
+const spichernstr = '900000042101'
+
+test('retry works', (t) => {
+	// for the first 3 calls, return different kinds of errors
+	let calls = 0
+	const failingRequest = (profile, userAgent, opt, data) => {
+		calls++
+		if (calls === 1) {
+			const err = new Error('HAFAS error')
+			err.isHafasError = true
+			return Promise.reject(err)
+		}
+		if (calls === 2) {
+			const err = new Error('fetch error')
+			err.code = 'EFOO'
+			return Promise.reject(err)
+		}
+		if (calls < 4) return Promise.reject(new Error('generic error'))
+		return Promise.resolve([])
+	}
+
+	const client = createClientWithRetry(vbbProfile, userAgent, {
+		retries: 3,
+		minTimeout: 100,
+		factor: 2,
+		randomize: false
+	}, failingRequest)
+
+	t.plan(1 + 4)
+	client.departures(spichernstr, {duration: 1})
+	.then(deps => t.deepEqual(deps, [], 'resolved with invalid value'))
+	.catch(t.ifError)
+
+	setTimeout(() => t.equal(calls, 1), 50) // buffer
+	setTimeout(() => t.equal(calls, 2), 150) // 100 + buffer
+	setTimeout(() => t.equal(calls, 3), 350) // 100 + 200 + buffer
+	setTimeout(() => t.equal(calls, 4), 750) // 100 + 200 + 400 + buffer
+})