LibreChat/api/server/experimental.spec.js

const fs = require('fs');
const path = require('path');

describe('Experimental server configuration', () => {
  const source = fs.readFileSync(path.join(__dirname, 'experimental.js'), 'utf8');

  it('configures HTTP timeouts for each cluster worker server', () => {
    const listenIndex = source.indexOf('const server = app.listen');
    const timeoutConfigIndex = source.indexOf('configureServerTimeouts(server);');

    expect(listenIndex).toBeGreaterThan(-1);
    expect(timeoutConfigIndex).toBeGreaterThan(-1);
    expect(listenIndex).toBeLessThan(timeoutConfigIndex);
  });

  it('lets each worker drain registered services before cluster shutdown', () => {
    const listenIndex = source.indexOf('const server = app.listen');
    const gracefulShutdownIndex = source.indexOf('setupGracefulShutdown(server);');

    expect(gracefulShutdownIndex).toBeGreaterThan(-1);
    expect(listenIndex).toBeLessThan(gracefulShutdownIndex);
    expect(source).toContain('if (shuttingDown) {');
    expect(source).toMatch(/if \(shuttingDown\) \{[\s\S]*?return;[\s\S]*?Starting a new worker/);
  });

  it('starts approval expiry after installing the scheduled-run callback', () => {
    const handlerIndex = source.indexOf(
      'GenerationJobManager.setApprovalExpiredHandler(recordExpiredScheduleApproval);',
    );
    const initializeIndex = source.indexOf('GenerationJobManager.initialize();');

    expect(handlerIndex).toBeGreaterThan(-1);
    expect(initializeIndex).toBeGreaterThan(handlerIndex);
  });

  it('starts erasure-only schedule maintenance after connecting to Mongo, once per worker', () => {
    const connectIndex = source.indexOf('await connectDb();');
    const sweepIndex = source.indexOf('initializeScheduleErasureSweep();');

    expect(connectIndex).toBeGreaterThan(-1);
    expect(sweepIndex).toBeGreaterThan(-1);
    // Mongo must be up before the sweep reads soft-deleted rows.
    expect(sweepIndex).toBeGreaterThan(connectIndex);
    // Idempotent guard lives in the service; started exactly once from this entrypoint.
    expect(source.match(/initializeScheduleErasureSweep\(\);/g)).toHaveLength(1);
  });

  it('never arms the full schedule engine in a clustered worker', () => {
    // The clustered entrypoint runs erasure-only maintenance: arming the engine here
    // would claim/fire/absence-reconcile runs whose peer generations it cannot see.
    expect(source).not.toContain('initializeScheduleEngine(');
  });

  it('runs cross-tenant startup work in the system context', () => {
    expect(source).toContain('await runAsSystem(seedDatabase);');
    expect(source).toMatch(
      /await runAsSystem\(async \(\) => \{\s+await performStartupChecks\(appConfig\);\s+await updateInterfacePerms/,
    );
  });

  it('configures routed subagent controls before a worker accepts requests', () => {
    const redisReadyIndex = source.indexOf('await waitForKeyvRedisClient();');
    const routingIndex = source.indexOf('await configureSubagentTaskRouting();');
    const listenIndex = source.indexOf('const server = app.listen');

    expect(redisReadyIndex).toBeGreaterThan(-1);
    expect(routingIndex).toBeGreaterThan(redisReadyIndex);
    expect(listenIndex).toBeGreaterThan(routingIndex);
  });

  it('matches the standard server pre-authentication tenant routes', () => {
    expect(source).toContain("app.use('/oauth', preAuthTenantMiddleware, routes.oauth);");
    expect(source).toContain("app.use('/api/auth', preAuthTenantMiddleware, routes.auth);");
    expect(source).toContain(
      "app.use('/api/config', preAuthTenantMiddleware, optionalJwtAuth, routes.config);",
    );
    expect(source).toContain("app.use('/api/share', preAuthTenantMiddleware, routes.share);");
  });
});