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.ok(err);
		t.is(trys, 1);
	}
});

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

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

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

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