Test the database manager shutdown hook

Signed-off-by: Eric Peterson <ericpeterson@spotify.com>
2024-10-01 16:43:03 +02:00
parent ffd1f4ab74
commit 8374f0ca17
1 changed files with 31 additions and 0 deletions
@@ -176,4 +176,35 @@ describe('DatabaseManagerImpl', () => {
      skip: true,
    });
  });
+
+  it('registers a shutdown hook if root lifecycle service is provided', async () => {
+    // Given a database manager that is provided a rootLifecycle service
+    const rootLifecycle = { addShutdownHook: jest.fn() } as unknown as any;
+    const destroy = jest.fn();
+    const connector1 = {
+      getClient: jest.fn().mockResolvedValue({ destroy }),
+    } satisfies Connector;
+    const impl = new DatabaseManagerImpl(
+      new ConfigReader({
+        client: 'pg',
+      }),
+      {
+        pg: connector1,
+      },
+      { rootLifecycle },
+    );
+
+    // Then a shutdown hook should have been added
+    expect(rootLifecycle.addShutdownHook).toHaveBeenCalled();
+    const shutdownHook = rootLifecycle.addShutdownHook.mock.calls[0][0];
+
+    // When a database client for a plugin is retrieved
+    await impl.forPlugin('plugin1', deps).getClient();
+
+    // And the shutdownhook is called
+    await shutdownHook();
+
+    // Then the destroy method should have been called on the resolved client
+    expect(destroy).toHaveBeenCalled();
+  });
 });