src/Line.test.ts

   1 import { Line, InputLine, 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 });
  54
  55 test('input line', () => {
  56   interface Input {
  57     key: string;
  58     key2?: string;
  59   }
  60   class TestForm extends Form {
  61     get name() { return 'F1'; }
  62
  63     protected getLines() {
  64       return [
  65         new InputLine<string, Input>('1', 'key'),
  66         new InputLine<string, Input>('2', 'key2')
  67       ];
  68     }
  69   };
  70   const tr = new TaxReturn(2019);
  71   const f = new TestForm({ 'key': 'value' });
  72
  73   expect(f.getLine('1').value(tr)).toBe('value');
  74
  75   const l2 = f.getLine('2');
  76   expect(() => l2.value(tr)).toThrow(NotFoundError);
  77 });
  78
  79 test('line stack', () => {
  80   class FormZ extends Form {
  81     get name() { return 'Z'; }
  82
  83     protected getLines() {
  84       return [ new InputLine<any, any>('3', 'input') ];
  85     }
  86   };
  87
  88   class FormZ2 extends Form {
  89     get name() { return 'Z-2'; }
  90
  91     protected getLines() {
  92       return [
  93         new ComputedLine<number>('2c', (tr: TaxReturn, l: Line<number>): any => {
  94           return tr.getForm('Z').getLine('3').value(tr) * 0.2;
  95         })
  96       ];
  97     }
  98   };
  99
 100   const tr = new TaxReturn(2019);
 101   tr.addForm(new FormZ({ 'input': 100 }));
 102   tr.addForm(new FormZ2());
 103
 104   const l = new ReferenceLine<number>('32', 'Z-2', '2c');
 105   expect(l.value(tr)).toBe(20);
 106 });