import { rowsFromUpdateReturning } from './query-update-returning';

describe('rowsFromUpdateReturning', () => {
  it('unwraps TypeORM UPDATE … RETURNING shape [rows, rowCount]', () => {
    const rows = [{ id: 'ledger-1', balance: '800.00' }];
    expect(rowsFromUpdateReturning([rows, 1])).toEqual(rows);
  });

  it('passes through a bare row array (SELECT shape)', () => {
    const rows = [{ amount_settled: '500.00', status: 'PARTIAL' }];
    expect(rowsFromUpdateReturning(rows)).toEqual(rows);
  });

  it('returns empty array for non-array input', () => {
    expect(rowsFromUpdateReturning(null)).toEqual([]);
    expect(rowsFromUpdateReturning(undefined)).toEqual([]);
  });
});