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