qinglong/packages/ql3-runtime-core/test/clusterExecutionRevision.test.cjs

const assert = require('node:assert/strict');
const { test } = require('node:test');
const {
  InvalidClusterExecutionRevisionError,
  compileClusterCommandTaskDefinition,
  normalizeClusterTaskExecutionRevision,
} = require('../dist/task-definition/clusterExecutionRevision');
const {
  createTaskDefinitionRecord,
  normalizeAppendTaskDefinitionRevisionCommand,
} = require('../dist/task-definition/taskDefinition');
const {
  createBuiltInTaskSpecSemanticRegistry,
} = require('../dist/task-definition/taskSpecSemantic');
const { createSecretRef } = require('../dist/secret/secretReference');

function definition() {
  const registry = createBuiltInTaskSpecSemanticRegistry();
  const command = normalizeAppendTaskDefinitionRevisionCommand({
    projectId: 'default',
    taskId: 'task-1',
    expectedRevision: null,
    mutationId: '019f7600-0000-7000-8000-000000000001',
    name: 'Cluster command',
    kind: 'command',
    spec: {
      schema: 'qinglong/command@v1',
      config: {
        command: { kind: 'argv', file: '/bin/echo', args: ['ready'] },
        environment: [
          { kind: 'public', name: 'MODE', value: 'cluster' },
          {
            kind: 'secret',
            name: 'TOKEN',
            secretRef: createSecretRef({ projectId: 'default', name: 'TOKEN' }),
          },
        ],
        timeoutMs: 5000,
      },
    },
    labels: {},
    enabled: true,
    occurredAtMs: 100,
  });
  return {
    registry,
    record: createTaskDefinitionRecord(
      {
        ...command,
        spec: registry.normalize({
          projectId: command.projectId,
          taskId: command.taskId,
          kind: command.kind,
          spec: command.spec,
        }),
      },
      90,
    ),
  };
}

function definitionWithBundle() {
  const input = definition();
  const environmentBundleRef = createSecretRef({
    projectId: 'default',
    name: 'legacy-env-bundle',
    version: 4,
  });
  const spec = input.registry.normalize({
    projectId: input.record.projectId,
    taskId: input.record.taskId,
    kind: input.record.kind,
    spec: {
      ...input.record.spec,
      config: { ...input.record.spec.config, environmentBundleRef },
    },
  });
  return {
    registry: input.registry,
    environmentBundleRef,
    record: createTaskDefinitionRecord(
      {
        projectId: input.record.projectId,
        taskId: input.record.taskId,
        expectedRevision: null,
        mutationId: input.record.mutationId,
        name: input.record.name,
        kind: input.record.kind,
        spec,
        labels: input.record.labels,
        enabled: input.record.enabled,
        occurredAtMs: input.record.updatedAtMs,
      },
      input.record.createdAtMs,
    ),
  };
}

test('compiles one digest-bound remote Worker execution revision', () => {
  const input = definition();
  const revision = compileClusterCommandTaskDefinition(
    input.record,
    input.registry,
  );
  assert.equal(revision.executorType, 'remote_worker');
  assert.equal(revision.planSchema, 'qinglong/command-execution@v1');
  assert.equal(revision.sourceRevision, input.record.revision);
  assert.equal(revision.sourceContentDigest, input.record.contentDigest);
  assert.match(revision.contentDigest, /^[0-9a-f]{64}$/);
  assert.deepEqual(normalizeClusterTaskExecutionRevision(revision), revision);
});

test('carries only the pinned environment bundle reference into Cluster plans', () => {
  const input = definitionWithBundle();
  const revision = compileClusterCommandTaskDefinition(
    input.record,
    input.registry,
  );
  assert.equal(revision.environmentBundleRef, input.environmentBundleRef);
  assert.equal(JSON.stringify(revision).includes('legacy env value'), false);
  assert.deepEqual(normalizeClusterTaskExecutionRevision(revision), revision);
});

test('rejects digest drift and cross-Project Secret references', () => {
  const input = definition();
  const revision = compileClusterCommandTaskDefinition(
    input.record,
    input.registry,
  );
  assert.throws(
    () =>
      normalizeClusterTaskExecutionRevision({
        ...revision,
        contentDigest: '0'.repeat(64),
      }),
    InvalidClusterExecutionRevisionError,
  );
  assert.throws(
    () =>
      normalizeClusterTaskExecutionRevision({
        ...revision,
        environment: [
          {
            kind: 'secret',
            name: 'TOKEN',
            secretRef: createSecretRef({ projectId: 'another', name: 'TOKEN' }),
          },
        ],
      }),
    InvalidClusterExecutionRevisionError,
  );
});