import { ColumnType, FormulaResultType, NumberFormat } from '../types';
import { getFinalColumnType } from './columns';

describe('getFinalColumnType', () => {
  it('returns non-formula types as-is', () => {
    expect(getFinalColumnType({ type: ColumnType.TEXT })).toEqual(
      ColumnType.TEXT
    );
  });

  it('returns formula types based on resultType', () => {
    expect(
      getFinalColumnType({
        type: ColumnType.FORMULA,
        formula: '',
        resultType: FormulaResultType.NUMBER,
        numberFormat: NumberFormat.DECIMAL,
      })
    ).toEqual(ColumnType.NUMBER);
  });

  it('returns formula lookup types based on lookupTypeOptions', () => {
    expect(
      getFinalColumnType({
        type: ColumnType.LOOKUP,
        lookupColumnType: ColumnType.FORMULA,
        lookupTypeOptions: {
          type: ColumnType.FORMULA,
          formula: '',
          resultType: FormulaResultType.NUMBER,
          numberFormat: NumberFormat.DECIMAL,
        },
        foreignLookupColumnId: '',
        recordReferenceColumnId: '',
      })
    ).toEqual(ColumnType.NUMBER);
  });
});