Drop stale (older than an hour) test databases automatically on test close

Author: CarsonF

src/core/database/database.module.ts

@@ -40,6 +40,9 @@ export class DatabaseModule implements OnApplicationShutdown {
 
   async onApplicationShutdown() {
     if (this.config.neo4j.ephemeral) {
+      if (this.config.jest) {
+        await this.dbService.dropStaleTestDbs();
+      }
       await this.dbService.dropDb();
     }
     await this.db.close();

src/core/database/database.service.ts

@@ -3,8 +3,14 @@ import { entries, mapKeys } from '@seedcompany/common';
 import { Connection, node, type Query, relation } from 'cypher-query-builder';
 import { LazyGetter } from 'lazy-get-decorator';
 import { pickBy, startCase } from 'lodash';
-import { Duration } from 'luxon';
-import { defer, firstValueFrom, shareReplay, takeUntil } from 'rxjs';
+import { DateTime, Duration } from 'luxon';
+import {
+  defer,
+  EmptyError,
+  firstValueFrom,
+  shareReplay,
+  takeUntil,
+} from 'rxjs';
 import {
   DuplicateException,
   type ID,
@@ -209,6 +215,24 @@ export class DatabaseService {
     await this.runAdminCommand('DROP', dbName);
   }
 
+  async dropStaleTestDbs() {
+    const info = await this.getServerInfo().catch((e) => {
+      if (e instanceof EmptyError) return null;
+      throw e;
+    });
+    const staleTestDbs = (info?.databases ?? []).filter(({ name }) => {
+      const match = /test\.([\d-]+\.[\d-]+)\..+/.exec(name);
+      if (!match) {
+        return false;
+      }
+      const dt = DateTime.fromFormat(match[1], 'y-MM-dd.HH-mm-ss');
+      return dt.diffNow().as('hours') < -1;
+    });
+    for (const staleTestDb of staleTestDbs) {
+      await this.runAdminCommand('DROP', staleTestDb.name);
+    }
+  }
+
   private async runAdminCommand(action: 'CREATE' | 'DROP', dbName: string) {
     // @ts-expect-error Yes this is private, but we have a special use case.
     // We need to run this query with a session that's not configured to use the