qinglong/test/back/cron-scheduler-routing.test.cjs

const test = require('node:test');
const assert = require('node:assert/strict');
const fs = require('node:fs');
const ts = require('typescript');
const { ScheduleType } = require('../../back/interface/schedule');

test('system and node modes route extended cron once and keep portable rules in crontab', async () => {
  const source = fs.readFileSync('back/services/cron.ts', 'utf8');
  const begin = source.indexOf('  private isNodeCron(');
  const end = source.indexOf('  private async getLogName', begin);
  const saveBegin = source.indexOf('  private async setCrontab(');
  const saveEnd = source.indexOf('  public importCrontab(', saveBegin);
  const js = ts.transpileModule(
    `class Fixture {${source.slice(begin, end)}${source.slice(
      saveBegin,
      saveEnd,
    )}}; module.exports = Fixture;`,
    {
      compilerOptions: { target: ts.ScriptTarget.ES2020 },
    },
  ).outputText;
  for (const mode of ['system', 'node']) {
    const module = { exports: {} };
    let file = '',
      installs = 0;
    new Function(
      'module',
      'process',
      'execSync',
      'ScheduleType',
      'config',
      'writeFileWithLock',
      'CrontabModel',
      js,
    )(
      module,
      { env: { QL_SCHEDULER: mode } },
      () => {
        installs++;
      },
      ScheduleType,
      { crontabFile: '/unused' },
      async (_, data) => {
        file = data;
      },
      { nativeUpdate: async () => {} },
    );
    const service = new module.exports();
    service.makeCommand = (row) => `task ${row.id}.js`;
    const rows = [
      { id: 'plain', schedule: '*/5 0-23 * * 1,3' },
      { id: 'spaces', schedule: '  0\t0 * * *  ' },
      { id: 'numeric-step', schedule: '0/5 * * * *' },
      { id: 'sunday-seven', schedule: '0 0 * * 7' },
      { id: 'day-step', schedule: '0 0 */2 * *' },
      { id: 'restricted-days', schedule: '0 0 1-31 * 1' },
      { id: 'restricted-week', schedule: '0 0 1 * 0-6' },
      { id: 'annually', schedule: '@annually' },
      { id: 'midnight', schedule: '@midnight' },
      { id: 'minutely', schedule: '@minutely' },
      { id: 'seconds', schedule: '0 0 * * * *' },
      { id: 'question', schedule: '0 0 * * ?' },
      { id: 'last', schedule: '0 0 L * *' },
      { id: 'nth', schedule: '0 0 * * 1#2' },
      { id: 'named', schedule: '0 0 * JAN MON' },
      { id: 'macro', schedule: '@daily' },
      { id: 'short', schedule: '*' },
      {
        id: 'extra',
        schedule: '* * * * *',
        extra_schedules: [{ schedule: '0 * * * *' }],
      },
      { id: 'once', schedule: '@once' },
      { id: 'boot', schedule: '@boot' },
    ];
    for (const row of rows) {
      const special = ['once', 'boot'].includes(row.id);
      const portable = ['plain', 'spaces'].includes(row.id);
      assert.equal(
        service.shouldUseCronClient(row),
        !special && (mode === 'node' || !portable),
        `${mode}: ${row.id}`,
      );
    }
    await service.setCrontab({ data: rows, total: rows.length });
    assert.equal(installs, mode === 'system' ? 1 : 0);
    for (const id of [
      'numeric-step',
      'sunday-seven',
      'day-step',
      'restricted-days',
      'restricted-week',
      'annually',
      'midnight',
      'minutely',
      'seconds',
      'question',
      'last',
      'nth',
      'named',
      'macro',
      'short',
      'extra',
      'once',
      'boot',
    ]) {
      const line = file
        .split('\n')
        .find((line) => line.endsWith(`task ${id}.js`));
      assert.ok(
        line.startsWith('# '),
        `${mode}: ${id} cannot also run from system crontab`,
      );
    }
  }
});