import { describe, expect, it } from 'vitest';

import { rebindOrphanWorkOwnersInMigration } from './rebind-orphan-work-owners.mjs';

const placeholderUserId = 'wx-openid-placeholder';

function table(name, rows) {
  return { name, rows };
}

describe('rebindOrphanWorkOwnersInMigration', () => {
  it('把作品表里认证表不存在的 owner_user_id 回填到占位用户', () => {
    const migration = {
      schema_version: 1,
      exported_at_micros: 1,
      tables: [
        table('user_account', [
          { user_id: 'user_alive' },
          { user_id: placeholderUserId },
        ]),
        table('puzzle_work_profile', [
          { profile_id: 'p1', owner_user_id: 'user_missing' },
          { profile_id: 'p2', owner_user_id: 'user_alive' },
          { profile_id: 'p3', owner_user_id: placeholderUserId },
        ]),
        table('puzzle_agent_session', [
          { session_id: 'draft-1', owner_user_id: '' },
        ]),
        table('tracking_event', [
          { event_id: 't1', owner_user_id: 'user_missing' },
        ]),
      ],
    };

    const result = rebindOrphanWorkOwnersInMigration(migration, {
      placeholderUserId,
      validUserIds: ['user_alive'],
    });

    expect(result.reboundRows).toEqual([
      {
        table: 'puzzle_work_profile',
        rowKey: 'p1',
        from: 'user_missing',
        to: placeholderUserId,
      },
      {
        table: 'puzzle_agent_session',
        rowKey: 'draft-1',
        from: '',
        to: placeholderUserId,
      },
    ]);
    expect(migration.tables[1].rows[0].owner_user_id).toBe(placeholderUserId);
    expect(migration.tables[1].rows[1].owner_user_id).toBe('user_alive');
    expect(migration.tables[1].rows[2].owner_user_id).toBe(placeholderUserId);
    expect(migration.tables[2].rows[0].owner_user_id).toBe(placeholderUserId);
    expect(migration.tables[3].rows[0].owner_user_id).toBe('user_missing');
  });
});