feat: add platform automatic billing worker

scripts/platform-billing-worker-integration-test.js

@@ -0,0 +1,202 @@
+import assert from 'node:assert/strict';
+import pg from 'pg';
+import { spawn } from 'node:child_process';
+
+const databaseUrl = process.env.DATABASE_URL || 'postgresql://postgres:postgres@127.0.0.1:54322/postgres';
+
+const ids = {
+  tenant: '00000000-0000-0000-0000-00000000b901',
+  subscription: '00000000-0000-0000-0000-00000000b902',
+  noExpirySubscription: '00000000-0000-0000-0000-00000000b903',
+};
+
+function runWorkerOnce() {
+  const child = spawn(process.execPath, ['apps/worker/dist/apps/worker/src/index.js', '--once', '--job', 'platform-billing'], {
+    cwd: process.cwd(),
+    env: {
+      ...process.env,
+      DATABASE_URL: databaseUrl,
+      WORKER_PLATFORM_BILLING_BATCH_SIZE: '10',
+      WORKER_PLATFORM_BILLING_DAYS_AHEAD: '90',
+      WORKER_PLATFORM_BILLING_DUE_DAYS: '20',
+      WORKER_PLATFORM_BILLING_ID: 'platform-billing-integration-test',
+    },
+    stdio: ['ignore', 'pipe', 'pipe'],
+    windowsHide: true,
+  });
+  let output = '';
+  child.stdout.on('data', chunk => {
+    output += chunk.toString();
+  });
+  child.stderr.on('data', chunk => {
+    output += chunk.toString();
+  });
+  return new Promise((resolve, reject) => {
+    child.on('error', reject);
+    child.on('exit', code => {
+      try {
+        assert.equal(code, 0, `worker should exit 0\n${output}`);
+        assert.match(output, /platform-billing batch processed=\d+/, 'worker output should include platform billing summary');
+        resolve(output);
+      } catch (error) {
+        reject(error);
+      }
+    });
+  });
+}
+
+async function cleanup(pool) {
+  await pool.query('delete from public.audit_logs where tenant_id = $1', [ids.tenant]);
+  await pool.query('delete from public.tenant_invoice_payments where tenant_id = $1', [ids.tenant]);
+  await pool.query('delete from public.tenant_invoice_items where tenant_id = $1', [ids.tenant]);
+  await pool.query('delete from public.tenant_invoices where tenant_id = $1', [ids.tenant]);
+  await pool.query('delete from public.tenant_subscriptions where tenant_id = $1', [ids.tenant]);
+  await pool.query('delete from public.tenant_billing_profiles where tenant_id = $1', [ids.tenant]);
+  await pool.query('delete from public.tenant_domains where tenant_id = $1', [ids.tenant]);
+  await pool.query('delete from public.tenants where id = $1', [ids.tenant]);
+}
+
+async function createTenantAndSubscription(pool) {
+  await pool.query(
+    `
+    insert into public.tenants (id, slug, name, legal_name, status, mode, billing_status, metadata)
+    values ($1, 'platform-billing-worker', '平台自动计费测试租户', '平台自动计费测试有限公司', 'active', 'saas', 'active', '{"source":"platform-billing-worker-test"}'::jsonb)
+    `,
+    [ids.tenant],
+  );
+  await pool.query(
+    `
+    insert into public.tenant_billing_profiles (
+      tenant_id, billing_name, tax_id, contact_name, contact_phone, contact_email,
+      invoice_title, invoice_type
+    )
+    values (
+      $1, '平台自动计费测试有限公司', '91120000BILLING', '财务测试', '13900001111',
+      'billing-worker@example.test', '平台自动计费测试有限公司', 'normal_vat'
+    )
+    `,
+    [ids.tenant],
+  );
+  await pool.query(
+    `
+    insert into public.tenant_subscriptions (
+      id, tenant_id, plan_code, status, starts_at, expires_at,
+      billing_cycle, amount_cents, metadata
+    )
+    values (
+      $1, $2, 'starter_yearly', 'active',
+      now() - interval '330 days',
+      now() + interval '20 days',
+      'yearly', 980000, '{"source":"platform-billing-worker-test"}'::jsonb
+    )
+    `,
+    [ids.subscription, ids.tenant],
+  );
+  await pool.query(
+    `
+    insert into public.tenant_subscriptions (
+      id, tenant_id, plan_code, status, starts_at, expires_at,
+      billing_cycle, amount_cents, metadata
+    )
+    values (
+      $1, $2, 'starter_yearly', 'active',
+      now() - interval '30 days',
+      null,
+      'yearly', 980000, '{"source":"platform-billing-worker-test","case":"no-expiry"}'::jsonb
+    )
+    `,
+    [ids.noExpirySubscription, ids.tenant],
+  );
+}
+
+async function main() {
+  const pool = new pg.Pool({ connectionString: databaseUrl });
+  try {
+    await cleanup(pool);
+    await createTenantAndSubscription(pool);
+
+    const firstOutput = await runWorkerOnce();
+    assert.match(firstOutput, /processed=1/, 'worker should process one subscription candidate');
+    assert.match(firstOutput, /created=1/, 'worker should create one subscription invoice');
+    assert.match(firstOutput, /failed=0/, 'worker should not fail the subscription invoice');
+
+    const invoices = await pool.query(
+      `
+      select id, invoice_no, status, invoice_type, total_cents, balance_cents,
+             billing_period_start, billing_period_end, due_date, note, metadata
+      from public.tenant_invoices
+      where tenant_id = $1 and metadata->>'subscriptionId' = $2
+      `,
+      [ids.tenant, ids.subscription],
+    );
+    assert.equal(invoices.rowCount, 1, 'worker should create exactly one invoice for the subscription');
+    const invoice = invoices.rows[0];
+    assert.equal(invoice.status, 'issued', 'worker invoice should be issued');
+    assert.equal(invoice.invoice_type, 'subscription', 'worker invoice should be subscription type');
+    assert.equal(Number(invoice.total_cents), 980000, 'worker invoice should use subscription amount');
+    assert.equal(Number(invoice.balance_cents), 980000, 'worker invoice should start unpaid');
+    assert.equal(invoice.metadata?.source, 'subscription_auto', 'worker invoice metadata should record auto source');
+    assert.equal(invoice.metadata?.workerId, 'platform-billing-integration-test', 'worker invoice metadata should record worker id');
+    assert.ok(invoice.due_date, 'worker invoice should set a due date');
+
+    const items = await pool.query(
+      `
+      select item_type, description, quantity, unit_amount_cents, amount_cents, metadata
+      from public.tenant_invoice_items
+      where tenant_id = $1 and invoice_id = $2
+      `,
+      [ids.tenant, invoice.id],
+    );
+    assert.equal(items.rowCount, 1, 'worker invoice should create one invoice item');
+    assert.equal(items.rows[0].item_type, 'subscription', 'invoice item should be subscription');
+    assert.equal(Number(items.rows[0].amount_cents), 980000, 'invoice item amount should match subscription amount');
+    assert.equal(items.rows[0].metadata?.planCode, 'starter_yearly', 'invoice item should record plan code');
+
+    const audit = await pool.query(
+      `
+      select action, target_type, target_id, details
+      from public.audit_logs
+      where tenant_id = $1 and action = 'platform.invoice.subscription_auto_created'
+      order by created_at desc
+      limit 1
+      `,
+      [ids.tenant],
+    );
+    assert.equal(audit.rows[0]?.target_type, 'tenant_invoice', 'worker should audit the generated invoice');
+    assert.equal(audit.rows[0]?.target_id, invoice.id, 'worker audit target should be the invoice id');
+    assert.equal(audit.rows[0]?.details?.subscriptionId, ids.subscription, 'worker audit should record subscription id');
+
+    const secondOutput = await runWorkerOnce();
+    assert.match(secondOutput, /created=0/, 'second worker run should not duplicate subscription invoice');
+
+    const invoiceCount = await pool.query(
+      `
+      select count(*)::integer as count
+      from public.tenant_invoices
+      where tenant_id = $1 and metadata->>'subscriptionId' = $2
+      `,
+      [ids.tenant, ids.subscription],
+    );
+    assert.equal(Number(invoiceCount.rows[0]?.count), 1, 'worker should be idempotent for subscription invoices');
+
+    const noExpiryInvoiceCount = await pool.query(
+      `
+      select count(*)::integer as count
+      from public.tenant_invoices
+      where tenant_id = $1 and metadata->>'subscriptionId' = $2
+      `,
+      [ids.tenant, ids.noExpirySubscription],
+    );
+    assert.equal(Number(noExpiryInvoiceCount.rows[0]?.count), 0, 'worker should not auto invoice subscriptions without an expiry date');
+
+    console.log('Platform billing worker integration test complete.');
+  } finally {
+    await cleanup(pool).catch(() => {});
+    await pool.end();
+  }
+}
+
+main().catch(error => {
+  console.error(error);
+  process.exit(1);
+});