Files
directus/tests/setup/utils/await-connection.ts
2021-09-28 15:42:50 -04:00

45 lines
1001 B
TypeScript

import { Knex } from 'knex';
import axios from 'axios';
export async function awaitDatabaseConnection(
database: Knex,
checkSQL: string,
currentAttempt = 0
): Promise<void | null> {
try {
await database.raw(checkSQL);
} catch {
if (currentAttempt === 10) {
throw new Error(`Couldn't connect to DB`);
}
return new Promise((resolve) => {
setTimeout(async () => {
await awaitDatabaseConnection(database, checkSQL, currentAttempt + 1);
resolve(null);
}, 5000);
});
}
}
export async function awaitDirectusConnection(port = 6100, currentAttempt = 0): Promise<void | null> {
try {
await axios.get(`http://localhost:${port}/server/ping`);
} catch {
if (currentAttempt === 10) {
throw new Error(`Couldn't connect to Directus`);
}
return new Promise((resolve, reject) => {
setTimeout(async () => {
try {
await awaitDirectusConnection(port, currentAttempt + 1);
resolve(null);
} catch (err) {
reject(err);
}
}, 5000);
});
}
}