got/test/retry.js

import test from 'ava';
import got from '../';
import {createServer} from './helpers/server';

let s;
let trys = 0;
let knocks = 0;
let fifth = 0;

test.before('setup', async () => {
	s = await createServer();

	s.on('/long', () => {});

	s.on('/knock-twice', (req, res) => {
		if (knocks++ === 1) {
			res.end('who`s there?');
		}
	});

	s.on('/try-me', () => {
		trys++;
	});

	s.on('/fifth', (req, res) => {
		if (fifth++ === 5) {
			res.end('who`s there?');
		}
	});

	await s.listen(s.port);
});

test('works on timeout error', async t => {
	t.is((await got(`${s.url}/knock-twice`, {timeout: 100})).body, 'who`s there?');
});

test('can be disabled with option', async t => {
	try {
		await got(`${s.url}/try-me`, {
			timeout: 500,
			retries: 0
		});
		t.fail();
	} catch (err) {
		t.truthy(err);
		t.is(trys, 1);
	}
});

test('function gets iter count', async t => {
	await got(`${s.url}/fifth`, {
		timeout: 100,
		retries: iter => iter < 10
	});
	t.is(fifth, 6);
});

test('falsy value prevents retries', async t => {
	try {
		await got(`${s.url}/long`, {
			timeout: 100,
			retries: () => 0
		});
	} catch (err) {
		t.truthy(err);
	}
});

test('falsy value prevents retries #2', async t => {
	try {
		await got(`${s.url}/long`, {
			timeout: 100,
			retries: (iter, err) => {
				t.truthy(err);
				return false;
			}
		});
	} catch (err) {
		t.truthy(err);
	}
});

test.after('cleanup', async () => {
	await s.close();
});