Add Line.test.ts.

diff --git a/src/Line.test.ts b/src/Line.test.ts
new file mode 100644 (file)
index 0000000..4dc6f80
--- /dev/null
+++ b/src/Line.test.ts
@@ -0,0 +1,53 @@
+import { Line, ReferenceLine, ComputedLine } from './Line';
+import Form from './Form';
+import TaxReturn from './TaxReturn';
+import { NotFoundError } from './Errors';
+
+class ConstantLine<T> extends Line<T> {
+  private _k: T;
+
+  constructor(id: string, k: T) {
+    super(id, `Constant ${k}`);
+    this._k = k;
+  }
+
+  value(tr: TaxReturn): T {
+    return this._k;
+  }
+};
+
+test('computed line', () => {
+  const tr = new TaxReturn(2019);
+  const l = new ComputedLine<number>('A',
+    (taxReturn: TaxReturn, line: ComputedLine<number>): number => {
+      expect(taxReturn).toBe(tr);
+      expect(line).toBe(l);
+      return 42;
+    },
+    'Computed Line A');
+  expect(l.value(tr)).toBe(42);
+  expect(l.id).toBe('A');
+  expect(l.description).toBe('Computed Line A');
+});
+
+test('reference line', () => {
+  class TestForm extends Form {
+    get name() { return 'Form 1'; }
+
+    protected getLines() {
+      return [ new ConstantLine('6b', 12.34) ];
+    }
+  };
+
+  const tr = new TaxReturn(2019);
+  tr.addForm(new TestForm());
+
+  const l1 = new ReferenceLine<number>('C', 'Form 1', '6b');
+  expect(l1.value(tr)).toBe(12.34);
+
+  const l2 = new ReferenceLine<number>('x', 'Form 2', '6b');
+  expect(() => l2.value(tr)).toThrow(NotFoundError);
+
+  const l3 = new ReferenceLine<number>('y', 'Form 1', '7a');
+  expect(() => l3.value(tr)).toThrow(NotFoundError);
+});