moving

packages/worker/tests/entrypoint.test.ts

@@ -0,0 +1,72 @@
+import { describe, expect, it, vi } from 'vitest';
+import { readFileSync } from 'node:fs';
+import { resolve } from 'node:path';
+import { fileURLToPath, pathToFileURL } from 'node:url';
+import { installShutdownHandlers, isDirectModule } from '../src/index.js';
+
+const REPO_ROOT = fileURLToPath(new URL('../../../', import.meta.url));
+
+describe('cloud worker production entrypoint', () => {
+  it('builds a standalone compiled worker after its server dependency', () => {
+    const rootPackage = JSON.parse(readFileSync(resolve(REPO_ROOT, 'package.json'), 'utf8')) as {
+      scripts: Record<string, string>;
+    };
+    const workerPackage = JSON.parse(
+      readFileSync(resolve(REPO_ROOT, 'packages/worker/package.json'), 'utf8'),
+    ) as {
+      main?: string;
+      types?: string;
+      exports?: Record<string, { import?: string; types?: string }>;
+      scripts?: Record<string, string>;
+    };
+    const build = rootPackage.scripts['build:packages'];
+
+    expect(build.indexOf('../worker')).toBeGreaterThan(build.indexOf('../server'));
+    expect(build).toContain('cd ../worker && npm run build');
+    expect(workerPackage.scripts?.build).toContain('tsc --build --force');
+    expect(workerPackage.scripts?.build).toContain('--bundle');
+    expect(workerPackage.main).toBe('dist/index.js');
+    expect(workerPackage.types).toBe('dist/src/index.d.ts');
+    expect(workerPackage.exports?.['.']).toEqual({
+      types: './dist/src/index.d.ts',
+      import: './dist/index.js',
+    });
+  });
+
+  it('runs the compiled worker as a production Compose service', () => {
+    const compose = readFileSync(resolve(REPO_ROOT, 'docker-compose.production.yml'), 'utf8');
+
+    expect(compose).toMatch(/\n {2}worker:\r?\n/);
+    expect(compose).toContain("command: ['node', 'packages/worker/dist/index.js']");
+    expect(compose).toContain('DATABASE_URL=postgres://waggle:${POSTGRES_PASSWORD:?set POSTGRES_PASSWORD}@postgres:5432/waggle');
+    expect(compose).toContain('REDIS_URL=redis://redis:6379');
+  });
+
+  it('recognizes the compiled entrypoint by URL identity', () => {
+    const distPath = resolve(REPO_ROOT, 'packages/worker/dist/index.js');
+
+    expect(isDirectModule(distPath, pathToFileURL(distPath).href)).toBe(true);
+    expect(isDirectModule(distPath, pathToFileURL(resolve(REPO_ROOT, 'consumer.js')).href)).toBe(false);
+    expect(isDirectModule(undefined, pathToFileURL(distPath).href)).toBe(false);
+  });
+
+  it('drains the worker once across repeated shutdown signals', async () => {
+    const listeners = new Map<string, () => void>();
+    let releaseClose!: () => void;
+    const close = vi.fn(() => new Promise<void>(resolve => { releaseClose = resolve; }));
+    const shutdown = installShutdownHandlers(
+      { close },
+      { once: (signal, listener) => { listeners.set(signal, listener); } },
+    );
+
+    expect([...listeners.keys()]).toEqual(['SIGTERM', 'SIGINT']);
+    listeners.get('SIGTERM')!();
+    listeners.get('SIGINT')!();
+    const draining = shutdown();
+
+    expect(shutdown()).toBe(draining);
+    expect(close).toHaveBeenCalledTimes(1);
+    releaseClose();
+    await draining;
+  });
+});