src/Line.test.ts

   1 import { Line, ReferenceLine, ComputedLine } from './Line';
   2 import Form from './Form';
   3 import TaxReturn from './TaxReturn';
   4 import { NotFoundError } from './Errors';
   5
   6 class ConstantLine<T> extends Line<T> {
   7   private _k: T;
   8
   9   constructor(id: string, k: T) {
  10     super(id, `Constant ${k}`);
  11     this._k = k;
  12   }
  13
  14   value(tr: TaxReturn): T {
  15     return this._k;
  16   }
  17 };
  18
  19 test('computed line', () => {
  20   const tr = new TaxReturn(2019);
  21   const l = new ComputedLine<number>('A',
  22     (taxReturn: TaxReturn, line: ComputedLine<number>): number => {
  23       expect(taxReturn).toBe(tr);
  24       expect(line).toBe(l);
  25       return 42;
  26     },
  27     'Computed Line A');
  28   expect(l.value(tr)).toBe(42);
  29   expect(l.id).toBe('A');
  30   expect(l.description).toBe('Computed Line A');
  31 });
  32
  33 test('reference line', () => {
  34   class TestForm extends Form {
  35     get name() { return 'Form 1'; }
  36
  37     protected getLines() {
  38       return [ new ConstantLine('6b', 12.34) ];
  39     }
  40   };
  41
  42   const tr = new TaxReturn(2019);
  43   tr.addForm(new TestForm());
  44
  45   const l1 = new ReferenceLine<number>('C', 'Form 1', '6b');
  46   expect(l1.value(tr)).toBe(12.34);
  47
  48   const l2 = new ReferenceLine<number>('x', 'Form 2', '6b');
  49   expect(() => l2.value(tr)).toThrow(NotFoundError);
  50
  51   const l3 = new ReferenceLine<number>('y', 'Form 1', '7a');
  52   expect(() => l3.value(tr)).toThrow(NotFoundError);
  53 });