orca/cloud/dev/scripts/relay-asia-topology-workflow.test.mjs

import assert from 'node:assert/strict'
import { spawnSync } from 'node:child_process'
import { mkdtempSync, readFileSync, rmSync } from 'node:fs'
import { tmpdir } from 'node:os'
import { join } from 'node:path'
import { test } from 'node:test'
import { relayWorkflowUrl } from './relay-repository.mjs'

const workflow = readFileSync(
  relayWorkflowUrl('deploy-relay-asia-topology.yml'),
  'utf8'
)
const iam = readFileSync(
  new URL('../../infra/terraform/relay-asia-topology-iam.tf', import.meta.url),
  'utf8'
)
const cells = readFileSync(
  new URL('../../infra/terraform/relay-gce-cells.tf', import.meta.url),
  'utf8'
)
const variables = readFileSync(
  new URL('../../infra/terraform/variables.tf', import.meta.url),
  'utf8'
)

test('uses only its exact workflow-bound topology identity', () => {
  assert.match(workflow, /production-cloud-sql-rollout/)
  assert.match(workflow, /relay-staging-mutation/)
  assert.match(workflow, /RELAY_ASIA_TOPOLOGY_WORKLOAD_IDENTITY_PROVIDER/)
  assert.match(workflow, /RELAY_ASIA_TOPOLOGY_SERVICE_ACCOUNT/)
  assert.doesNotMatch(workflow, /GCP_DEPLOY_SERVICE_ACCOUNT/)
  assert.match(
    iam,
    /assertion\.workflow_ref == '\$\{prefix\}\$\{local\.github_relay_asia_topology_workflow_file\}@refs\/heads\/main'/
  )
  assert.match(iam, /assertion\.ref == 'refs\/heads\/main'/)
  assert.match(iam, /assertion\.event_name == 'workflow_dispatch'/)
  assert.match(iam, /assertion\.environment == '\$\{var\.environment\}'/)
})

test('accepts only the reviewed Asia topology waves', () => {
  const cases = /case "\$\{TARGET_ENVIRONMENT\}:\$\{TARGET_CELL_IDS\}" in\n([\s\S]*?)\n\s*esac/
    .exec(workflow)?.[1]
  assert.ok(cases)
  assert.deepEqual(
    [...cases.matchAll(/^\s*([a-z]+:[a-z0-9,-]+)\)(?: target_region=[a-z0-9-]+)? ;;$/gm)]
      .map((match) => match[1]),
    [
      'staging:staging-gce-c4',
      'production:production-gce-c27,production-gce-c28,production-gce-c29',
      'production:production-gce-c30',
      'production:production-gce-c31',
      'production:production-gce-c32,production-gce-c33',
      'production:production-gce-c34'
    ]
  )
})

// Runs the workflow's own region and target blocks, so an Asia wave's targets stay byte-identical.
function waveTargets(environment, cellIds) {
  const block = (first, last) => {
    const start = workflow.indexOf(first)
    const end = workflow.indexOf(last, start)
    assert.ok(start !== -1 && end !== -1, first)
    return workflow.slice(start, end + last.length).replace(/^ {10}/gm, '')
  }
  const region = block('target_region=asia-east2\n', '\n          esac')
  const targets = block('file="${RUNNER_TEMP}/relay-asia-targets"', '\n          done')
  const temp = mkdtempSync(join(tmpdir(), 'relay-topology-targets-'))
  try {
    const result = spawnSync('bash', ['-euo', 'pipefail', '-c',
      `${region}\nTARGET_REGION="\${target_region}"\n${targets}\n` +
      'echo "${TARGET_REGION}"; cat "${file}"'], {
      env: { ...process.env, TARGET_ENVIRONMENT: environment, TARGET_CELL_IDS: cellIds, RUNNER_TEMP: temp },
      encoding: 'utf8'
    })
    assert.equal(result.status, 0, result.stderr)
    const [waveRegion, ...lines] = result.stdout.trim().split('\n')
    return { region: waveRegion, targets: lines }
  } finally {
    rmSync(temp, { recursive: true, force: true })
  }
}

const cellTargets = (cellId) => ['instance_template', 'instance_group_manager', 'backend_service']
  .map((kind) => `-target=google_compute_${kind}.relay_gce_cell["${cellId}"]`)

test('targets the asia-east2 network for Asia waves and only the cell and URL map for US waves', () => {
  const asiaNetwork = ['subnetwork', 'router', 'router_nat']
    .map((kind) => `-target=google_compute_${kind}.relay_gce_additional["asia-east2"]`)
  const urlMap = '-target=google_compute_url_map.relay_gce[0]'
  for (const [environment, cellIds] of [
    ['staging', 'staging-gce-c4'],
    ['production', 'production-gce-c27,production-gce-c28,production-gce-c29'],
    ['production', 'production-gce-c30'],
    ['production', 'production-gce-c31']
  ]) {
    assert.deepEqual(waveTargets(environment, cellIds), {
      region: 'asia-east2',
      targets: [...asiaNetwork, urlMap, ...cellIds.split(',').flatMap(cellTargets)]
    }, cellIds)
  }
  const usWave = 'production-gce-c32,production-gce-c33'
  assert.deepEqual(waveTargets('production', usWave), {
    region: 'us-central1',
    targets: [urlMap, ...usWave.split(',').flatMap(cellTargets)]
  })
})

// The live-image overlay refuses a declared non-target cell with no template, so the two US cells
// declared together must also plan together; a lone C32 or C33 plan is not a reviewed wave.
test('plans the two declared US cells as one wave', () => {
  const cases = /case "\$\{TARGET_ENVIRONMENT\}:\$\{TARGET_CELL_IDS\}" in\n([\s\S]*?)\n\s*esac/
    .exec(workflow)?.[1]
  assert.doesNotMatch(cases, /production:production-gce-c3[23]\)/)
})

test('plans only additive Asia topology and applies the saved plan', () => {
  assert.doesNotMatch(workflow, /manage_artifact_dns/)
  for (const target of [
    'relay_gce_additional',
    'google_compute_instance_template.relay_gce_cell',
    'google_compute_instance_group_manager.relay_gce_cell',
    'google_compute_backend_service.relay_gce_cell',
    'google_compute_url_map.relay_gce'
  ]) assert.match(workflow, new RegExp(target.replaceAll('.', '\\.')))
  assert.match(workflow, /apply -input=false -auto-approve "\$\{\{ steps\.plan\.outputs\.plan \}\}"/)
  assert.match(workflow, /prepare-relay-asia-topology-input\.mjs/)
  assert.doesNotMatch(workflow, /steps\.variables\.outputs\.file/)
  assert.match(workflow, /\.variables\.relay_gce_cells\.value/)
  assert.match(
    workflow,
    /\.variables\.relay_gce_additional_region_subnetwork_cidrs\.value/
  )
  // Console evaluates every output against state and fails while a declared cell has no MIG.
  assert.doesNotMatch(workflow, /terraform[^\n]*console/)
  assert.equal(
    (workflow.match(/-var-file="\$\{TF_VARS\}" -var-file="\$\{\{ steps\.live-images\.outputs\.file \}\}"/g) ?? []).length,
    2
  )
  assert.equal((workflow.match(/-var-file=/g) ?? []).length, 5)
  assert.doesNotMatch(workflow, /terraform[^\n]*apply[^\n]*-target/)
  assert.doesNotMatch(workflow, /google_(?:sql|cloudflare|dns|certificate_manager)/)
})

test('validates before apply and proves convergence afterward', () => {
  assert.equal((workflow.match(/validate-relay-asia-topology-plan\.mjs/g) ?? []).length, 2)
  assert.match(workflow, /APPLY_RELAY_ASIA_TOPOLOGY/)
  assert.match(workflow, /test "\$\(jq -er '\.changes'/)
  assert.match(workflow, /Register the exact new cells atomically as migration-only/)
})

test('checks the connection budget and production live ceiling before planning', () => {
  assert.match(workflow, /relay-cloud-sql-connection-budget\.mjs/)
  assert.match(workflow, /gcloud sql instances describe "\$\{CLOUD_SQL_INSTANCE\}"/)
  assert.match(workflow, /select\(\.name == "max_connections"\)/)
  assert.match(workflow, /VERIFIED_DEFAULT_MAX_CONNECTIONS_TIER: db-custom-4-15360/)
  assert.match(workflow, /VERIFIED_DEFAULT_MAX_CONNECTIONS_DATABASE_VERSION: POSTGRES_17/)
  assert.match(workflow, /VERIFIED_DEFAULT_MAX_CONNECTIONS: '500'/)
  assert.match(workflow, /live_max="\$\{VERIFIED_DEFAULT_MAX_CONNECTIONS\}"/)
  assert.doesNotMatch(workflow, /live_max=\d/)
  assert.match(workflow, /live_source=verified-shape-default/)
  assert.match(workflow, /test "\$\(jq -er '\.settings\.tier'/)
  assert.match(workflow, /test "\$\(jq -er '\.databaseVersion'/)
  assert.match(workflow, /test "\$\{live_max\}" = "\$\{checked_max\}"/)
  assert.ok(
    workflow.indexOf('relay-cloud-sql-connection-budget.mjs') <
      workflow.indexOf('terraform -chdir=infra/terraform plan')
  )
})

test('binds computed Asia references to the matching Terraform cell resources', () => {
  assert.match(cells, /instance_template = google_compute_instance_template\.relay_gce_cell\[each\.key\]\.self_link/)
  assert.match(cells, /group\s+= google_compute_instance_group_manager\.relay_gce_cell\[each\.key\]\.instance_group/)
  assert.match(cells, /default_service = google_compute_backend_service\.relay_gce_cell\[cell\.key\]\.id/)
  assert.match(cells, /subnetwork = local\.relay_gce_subnetworks\[each\.value\.region\]/)
})

test('keeps cross-variable region constraints in Terraform 1.5 check blocks', () => {
  assert.doesNotMatch(variables, /region != var\.region/)
  assert.doesNotMatch(variables, /cell\.region == var\.region/)
  assert.match(cells, /check "relay_gce_fixed_one_topology"[\s\S]*?region != var\.region/)
  assert.match(cells, /cell\.region == var\.region[\s\S]*?configured subnetwork/)
})

test('the custom role cannot delete topology or mutate SQL and DNS', () => {
  assert.doesNotMatch(iam, /compute\.[A-Za-z]+\.delete/)
  assert.doesNotMatch(
    iam,
    /roles\/viewer|cloudsql\.instances\.(?:update|delete)|dns\.|certificatemanager|cloudflare/i
  )
  assert.match(iam, /resource "google_project_iam_custom_role" "github_relay_asia_topology_read"/)
  assert.match(iam, /"cloudsql\.instances\.get"/)
  assert.match(iam, /"run\.revisions\.get"/)
  assert.match(iam, /"run\.services\.get"/)
  assert.match(iam, /"serviceusage\.services\.list"/)
  assert.match(iam, /"compute\.networks\.updatePolicy"/)
  assert.match(iam, /"compute\.healthChecks\.useReadOnly"/)
  assert.match(iam, /"compute\.instanceGroups\.create"/)
  assert.match(iam, /"compute\.instances\.use"/)
  assert.match(iam, /roles\/storage\.objectAdmin/)
  assert.match(iam, /default\.tfstate/)
  assert.match(iam, /default\.tflock/)
  assert.match(
    iam,
    /resource "google_project_iam_custom_role" "github_relay_asia_topology_state_list"[\s\S]*?permissions = \["storage\.objects\.list"\]/
  )
  assert.match(
    iam,
    /resource "google_storage_bucket_iam_member" "github_relay_asia_topology_state_list"[\s\S]*?role\s+= google_project_iam_custom_role\.github_relay_asia_topology_state_list\[0\]\.id/
  )
})

test('plans every non-target cell at its served image, read from state templates only', () => {
  const step = /- id: live-images\n[\s\S]*?\n\n/.exec(workflow)?.[0]
  assert.ok(step)
  assert.match(step, /terraform -chdir=infra\/terraform show -json \| jq -ce '\[/)
  assert.match(step, /\.type == "google_compute_instance_template" and \.name == "relay_gce_cell"/)
  assert.match(step, /\{ index, metadata_startup_script: \.values\.metadata_startup_script \}/)
  assert.match(step, /relay-live-cell-image-overlay\.mjs/)
  assert.match(step, /--cell-ids "\$\{TARGET_CELL_IDS\}"/)
  // The committed map comes from a read-only plan over the same targets, never from console.
  assert.match(step, /mapfile -t targets < "\$\{\{ steps\.targets\.outputs\.file \}\}"/)
  assert.match(
    step,
    /terraform -chdir=infra\/terraform plan -input=false -refresh=false -lock=false \\\n\s+-var-file="\$\{TF_VARS\}" "\$\{targets\[@\]\}" -out="\$\{committed_plan\}" > \/dev\/null/
  )
  assert.match(step, /show -json "\$\{committed_plan\}" \\\n\s+\| jq -ce '\.variables\.relay_gce_cells\.value \| objects' > "\$\{cells\}"/)
  assert.equal((step.match(/terraform -chdir=infra\/terraform (?:plan|apply)/g) ?? []).length, 1)
  assert.ok(workflow.indexOf('- id: targets') < workflow.indexOf('- id: live-images'))
  assert.ok(workflow.indexOf('- id: live-images') < workflow.indexOf('- name: Create and validate the saved topology plan'))
})

test('the deployments output tolerates a cell declared before its topology apply', () => {
  const outputs = readFileSync(new URL('../../infra/terraform/outputs.tf', import.meta.url), 'utf8')
  const start = outputs.indexOf('output "relay_gce_cell_deployments" {')
  const block = outputs.slice(start, outputs.indexOf('\n}\n', start))
  const lookups = [...block.matchAll(/^\s+\w+\s+=\s+(.*\.relay_gce_cell\[cell_id\].*)$/gm)].map((match) => match[1])
  assert.equal(lookups.length, 6)
  for (const lookup of lookups) {
    assert.match(lookup, /^try\(google_compute_\w+\.relay_gce_cell\[cell_id\]\.\w+, null\)$/)
  }
})