From 4f31c1bfc055759ed8ade25f4ebc11ba063764e6 Mon Sep 17 00:00:00 2001
From: Arman Ozak <arman@armanozak.com>
Date: Fri, 10 Apr 2020 00:12:37 +0300
Subject: [PATCH] feat: add container strategies

---
 .../src/lib/strategies/container.strategy.ts  | 44 ++++++++++
 .../packages/core/src/lib/strategies/index.ts |  1 +
 .../src/lib/tests/container.strategy.spec.ts  | 80 +++++++++++++++++++
 3 files changed, 125 insertions(+)
 create mode 100644 npm/ng-packs/packages/core/src/lib/strategies/container.strategy.ts
 create mode 100644 npm/ng-packs/packages/core/src/lib/tests/container.strategy.spec.ts

diff --git a/npm/ng-packs/packages/core/src/lib/strategies/container.strategy.ts b/npm/ng-packs/packages/core/src/lib/strategies/container.strategy.ts
new file mode 100644
index 0000000000..dfe169e402
--- /dev/null
+++ b/npm/ng-packs/packages/core/src/lib/strategies/container.strategy.ts
@@ -0,0 +1,44 @@
+import { ViewContainerRef } from '@angular/core';
+
+export abstract class ContainerStrategy {
+  constructor(public containerRef: ViewContainerRef) {}
+
+  abstract getIndex(): number;
+
+  prepare(): void {}
+}
+
+export class ClearContainerStrategy extends ContainerStrategy {
+  getIndex(): number {
+    return 0;
+  }
+
+  prepare() {
+    this.containerRef.clear();
+  }
+}
+
+export class InsertIntoContainerStrategy extends ContainerStrategy {
+  constructor(containerRef: ViewContainerRef, private index: number) {
+    super(containerRef);
+  }
+
+  getIndex() {
+    return Math.min(Math.max(0, this.index), this.containerRef.length);
+  }
+}
+
+export const CONTAINER_STRATEGY = {
+  Clear(containerRef: ViewContainerRef) {
+    return new ClearContainerStrategy(containerRef);
+  },
+  Append(containerRef: ViewContainerRef) {
+    return new InsertIntoContainerStrategy(containerRef, containerRef.length);
+  },
+  Prepend(containerRef: ViewContainerRef) {
+    return new InsertIntoContainerStrategy(containerRef, 0);
+  },
+  Insert(containerRef: ViewContainerRef, index: number) {
+    return new InsertIntoContainerStrategy(containerRef, index);
+  },
+};
diff --git a/npm/ng-packs/packages/core/src/lib/strategies/index.ts b/npm/ng-packs/packages/core/src/lib/strategies/index.ts
index 28447748f7..8c36b230ef 100644
--- a/npm/ng-packs/packages/core/src/lib/strategies/index.ts
+++ b/npm/ng-packs/packages/core/src/lib/strategies/index.ts
@@ -1,3 +1,4 @@
+export * from './container.strategy';
 export * from './content-security.strategy';
 export * from './content.strategy';
 export * from './context.strategy';
diff --git a/npm/ng-packs/packages/core/src/lib/tests/container.strategy.spec.ts b/npm/ng-packs/packages/core/src/lib/tests/container.strategy.spec.ts
new file mode 100644
index 0000000000..e85e7b5a50
--- /dev/null
+++ b/npm/ng-packs/packages/core/src/lib/tests/container.strategy.spec.ts
@@ -0,0 +1,80 @@
+import { ViewContainerRef } from '@angular/core';
+import {
+  ClearContainerStrategy,
+  CONTAINER_STRATEGY,
+  InsertIntoContainerStrategy,
+} from '../strategies';
+
+describe('ClearContainerStrategy', () => {
+  const containerRef = ({
+    clear: jest.fn(),
+    length: 7,
+  } as any) as ViewContainerRef;
+
+  describe('#getIndex', () => {
+    it('should return 0', () => {
+      const strategy = new ClearContainerStrategy(containerRef);
+      expect(strategy.getIndex()).toBe(0);
+    });
+  });
+
+  describe('#prepare', () => {
+    it('should call clear method of containerRef once', () => {
+      const strategy = new ClearContainerStrategy(containerRef);
+      strategy.prepare();
+      expect(strategy.getIndex()).toBe(0);
+      expect(containerRef.clear).toHaveBeenCalledTimes(1);
+    });
+  });
+});
+
+describe('InsertIntoContainerStrategy', () => {
+  const containerRef = ({
+    clear: jest.fn(),
+    length: 7,
+  } as any) as ViewContainerRef;
+
+  describe('#getIndex', () => {
+    test.each`
+      index       | expected
+      ${0}        | ${0}
+      ${4}        | ${4}
+      ${9}        | ${7}
+      ${-1}       | ${0}
+      ${Infinity} | ${7}
+    `(
+      'should return $expected when index is given $index',
+      ({ index, expected }: { index: number; expected: number }) => {
+        const strategy = new InsertIntoContainerStrategy(containerRef, index);
+        expect(strategy.getIndex()).toBe(expected);
+      },
+    );
+  });
+
+  describe('#prepare', () => {
+    it('should not call clear method of containerRef', () => {
+      const strategy = new InsertIntoContainerStrategy(containerRef, 0);
+      strategy.prepare();
+      expect(containerRef.clear).not.toHaveBeenCalled();
+    });
+  });
+});
+
+describe('CONTAINER_STRATEGY', () => {
+  const containerRef = ({
+    clear: jest.fn(),
+    length: 7,
+  } as any) as ViewContainerRef;
+
+  test.each`
+    name         | Strategy                       | index
+    ${'Clear'}   | ${ClearContainerStrategy}      | ${undefined}
+    ${'Append'}  | ${InsertIntoContainerStrategy} | ${containerRef.length}
+    ${'Prepend'} | ${InsertIntoContainerStrategy} | ${0}
+    ${'Insert'}  | ${InsertIntoContainerStrategy} | ${4}
+  `('should successfully map $name to $Strategy.name', ({ name, Strategy, index }) => {
+    expect(CONTAINER_STRATEGY[name](containerRef, index)).toEqual(
+      new Strategy(containerRef, index),
+    );
+  });
+});