admin/n8n-enterprise-unlocked
1
0
Fork 0
mirror of https://github.com/Abdulazizzn/n8n-enterprise-unlocked.git synced 2025-12-18 02:21:13 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat(core): Introduce DB health check (#10661)

Browse Source
This commit is contained in:
Iván Ovejero
2024-09-05 11:04:48 +02:00
committed by GitHub
parent 3a8078068e
commit a8e80d0c4b
6 changed files with 59 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
packages/cli/test/integration/shared/utils/test-server.ts
View File

@@ -44,9 +44,16 @@ function prefix(pathSegment: string) {
}
const browserId = 'test-browser-id';
function createAgent(app: express.Application, options?: { auth: boolean; user: User }) {
function createAgent(
app: express.Application,
options?: { auth: boolean; user?: User; noRest?: boolean },
) {
const agent = request.agent(app);
void agent.use(prefix(REST_PATH_SEGMENT));
const withRestSegment = !options?.noRest;
if (withRestSegment) void agent.use(prefix(REST_PATH_SEGMENT));
if (options?.auth && options?.user) {
const token = Container.get(AuthService).issueJWT(options.user, browserId);
agent.jar.setCookie(`${AUTH_COOKIE_NAME}=${token}`);
@@ -89,6 +96,7 @@ export const setupTestServer = ({
httpServer: app.listen(0),
authAgentFor: (user: User) => createAgent(app, { auth: true, user }),
authlessAgent: createAgent(app),
restlessAgent: createAgent(app, { auth: false, noRest: true }),
publicApiAgentFor: (user) => publicApiAgent(app, { user }),
license: new LicenseMocker(),
};
@@ -119,6 +127,13 @@ export const setupTestServer = ({
app.use(...apiRouters);
}
if (endpointGroups?.includes('health')) {
app.get('/healthz/readiness', async (_req, res) => {
testDb.isReady()
? res.status(200).send({ status: 'ok' })
: res.status(503).send({ status: 'error' });
});
}
if (endpointGroups.length) {
for (const group of endpointGroups) {
switch (group) {