iconify/packages/api-redundancy/tests/basic-query-test.ts

import type { RedundancyConfig } from '../src/config';
import { sendQuery } from '../src/query';

describe('Basic queries', () => {
	it('Empty query', () => {
		return new Promise((fulfill, reject) => {
			const payload = {};
			const config: RedundancyConfig = {
				resources: [],
				index: 0,
				timeout: 200,
				rotate: 100,
				random: false,
				dataAfterTimeout: false,
			};

			// Tracking
			let isSync = true;
			const startTime = Date.now();

			// Send query
			const getStatus = sendQuery(
				config,
				payload,
				() => {
					reject(
						'Query should not be called when resources list is empty'
					);
				},
				(data, error) => {
					expect(isSync).toEqual(false);
					expect(data).toBeUndefined();
					expect(error).toBeUndefined();

					// Check status
					const status = getStatus();
					expect(status.status).toEqual('failed');
					expect(status.queriesSent).toEqual(0);
					expect(status.queriesPending).toEqual(0);

					// Should be almost instant: no items in queue
					const diff = Date.now() - startTime;
					expect(diff).toBeLessThan(50);

					fulfill(true);
				}
			);

			// Check status
			const status = getStatus();
			expect(status.status).toEqual('pending');
			expect(status.queriesSent).toEqual(0);
			expect(status.queriesPending).toEqual(0);

			isSync = false;
		});
	});

	it('Simple query', () => {
		return new Promise((fulfill) => {
			const payload = {};
			const resources = ['test1'];
			const result = {};
			const config: RedundancyConfig = {
				resources,
				index: 0,
				timeout: 200,
				rotate: 100,
				random: false,
				dataAfterTimeout: false,
			};

			// Tracking
			let isSync = true;
			const startTime = Date.now();
			let sentQuery = false;

			// Send query
			const getStatus = sendQuery(
				config,
				payload,
				(resource, queryPayload, callback) => {
					expect(isSync).toEqual(false);
					expect(resource).toEqual(resources[0]);
					expect(queryPayload).toEqual(payload);

					// Make sure query was executed only once
					expect(sentQuery).toEqual(false);
					sentQuery = true;

					// Check status
					const status = getStatus();
					expect(status.status).toEqual('pending');
					expect(status.payload).toEqual(payload);
					expect(status.queriesSent).toEqual(1);
					expect(status.queriesPending).toEqual(1);

					// Complete
					callback('success', result);
				},
				(data, error) => {
					// Make sure query was sent
					expect(sentQuery).toEqual(true);

					// Validate data
					expect(data).toEqual(result);
					expect(error).toBeUndefined();

					// Check status
					const status = getStatus();
					expect(status.status).toEqual('completed');
					expect(status.queriesSent).toEqual(1);
					expect(status.queriesPending).toEqual(0);

					// Should be almost instant
					const diff = Date.now() - startTime;
					expect(diff).toBeLessThan(50);

					fulfill(true);
				}
			);

			// Check status
			const status = getStatus();
			expect(status.status).toEqual('pending');
			expect(status.queriesSent).toEqual(0);
			expect(status.queriesPending).toEqual(0);

			isSync = false;
		});
	});

	it('Failing query', () => {
		return new Promise((fulfill) => {
			const payload = {};
			const resources = ['api1'];
			const result = {};
			const config: RedundancyConfig = {
				resources,
				index: 0,
				timeout: 200,
				rotate: 100,
				random: false,
				dataAfterTimeout: false,
			};

			// Tracking
			let isSync = true;
			const startTime = Date.now();
			let sentQuery = false;

			// Send query
			const getStatus = sendQuery(
				config,
				payload,
				(resource, queryPayload, callback) => {
					expect(isSync).toEqual(false);
					expect(resource).toEqual(resources[0]);
					expect(queryPayload).toEqual(payload);

					// Make sure query was executed only once
					expect(sentQuery).toEqual(false);
					sentQuery = true;

					// Fail
					callback('next', result);
				},
				(data, error) => {
					// Make sure query was sent
					expect(sentQuery).toEqual(true);

					// Validate data
					expect(data).toBeUndefined();
					expect(error).toEqual(result);

					// Check status
					const status = getStatus();
					expect(status.status).toEqual('failed');
					expect(status.queriesSent).toEqual(1);
					expect(status.queriesPending).toEqual(0);

					// Should be almost instant
					const diff = Date.now() - startTime;
					expect(diff).toBeLessThan(40);

					fulfill(true);
				}
			);

			// Check status
			const status = getStatus();
			expect(status.status).toEqual('pending');
			expect(status.queriesSent).toEqual(0);
			expect(status.queriesPending).toEqual(0);

			isSync = false;
		});
	});

	it('Timed out query (~300ms)', () => {
		return new Promise((fulfill) => {
			const payload = {};
			const resources = ['api1'];
			const config: RedundancyConfig = {
				resources,
				index: 0,
				timeout: 200,
				rotate: 100,
				random: false,
				dataAfterTimeout: false,
			};

			// Tracking
			let isSync = true;
			const startTime = Date.now();
			let sentQuery = false;

			// Send query
			const getStatus = sendQuery(
				config,
				payload,
				(resource, queryPayload) => {
					expect(isSync).toEqual(false);
					expect(resource).toEqual(resources[0]);
					expect(queryPayload).toEqual(payload);

					// Make sure query was executed only once
					expect(sentQuery).toEqual(false);
					sentQuery = true;

					// Do not do anything
				},
				(data, error) => {
					// Make sure query was sent
					expect(sentQuery).toEqual(true);

					// Validate data
					expect(data).toBeUndefined();
					expect(error).toBeUndefined();

					// Check status
					const status = getStatus();
					expect(status.status).toEqual('failed');
					expect(status.queriesSent).toEqual(1);
					expect(status.queriesPending).toEqual(0);

					// Should have been config.rotate + config.timeout
					const diff = Date.now() - startTime;
					expect(diff).toBeGreaterThan(250);

					fulfill(true);
				}
			);

			// Check status
			const status = getStatus();
			expect(status.status).toEqual('pending');
			expect(status.queriesSent).toEqual(0);
			expect(status.queriesPending).toEqual(0);

			isSync = false;
		});
	});

	it('Abort query', () => {
		return new Promise((fulfill) => {
			const payload = {};
			const resources = ['api1', 'api2', 'api3'];
			const config: RedundancyConfig = {
				resources,
				index: 0,
				timeout: 200,
				rotate: 100,
				random: false,
				dataAfterTimeout: false,
			};

			// Tracking
			let isSync = true;
			let sentQuery = false;

			// Send query
			const getStatus = sendQuery(
				config,
				payload,
				(resource, queryPayload, callback) => {
					expect(isSync).toEqual(false);
					expect(resource).toEqual(resources[0]);
					expect(queryPayload).toEqual(payload);

					// Make sure query was executed only once
					expect(sentQuery).toEqual(false);
					sentQuery = true;

					// Abort
					callback('abort', 404);
				},
				(data, error) => {
					// Make sure query was sent
					expect(sentQuery).toEqual(true);

					// Validate data
					expect(data).toBeUndefined();
					expect(error).toBe(404);

					// Check status
					const status = getStatus();
					expect(status.status).toEqual('failed');
					expect(status.queriesSent).toEqual(1);
					expect(status.queriesPending).toEqual(0);

					fulfill(true);
				}
			);

			// Check status
			const status = getStatus();
			expect(status.status).toEqual('pending');
			expect(status.queriesSent).toEqual(0);
			expect(status.queriesPending).toEqual(0);

			isSync = false;
		});
	});
});