From 9a0c250374195522cc87cfde7437dfa26bce5a93 Mon Sep 17 00:00:00 2001
From: Arman Ozak <arman@armanozak.com>
Date: Thu, 28 May 2020 23:09:44 +0300
Subject: [PATCH] feat: keep element as public property on loading strategy

---
 .../packages/core/src/lib/strategies/loading.strategy.ts  | 8 +++++---
 .../packages/core/src/lib/tests/loading.strategy.spec.ts  | 2 ++
 2 files changed, 7 insertions(+), 3 deletions(-)

diff --git a/npm/ng-packs/packages/core/src/lib/strategies/loading.strategy.ts b/npm/ng-packs/packages/core/src/lib/strategies/loading.strategy.ts
index be89a751ed..f1f3abe30c 100644
--- a/npm/ng-packs/packages/core/src/lib/strategies/loading.strategy.ts
+++ b/npm/ng-packs/packages/core/src/lib/strategies/loading.strategy.ts
@@ -5,6 +5,8 @@ import { CrossOriginStrategy, CROSS_ORIGIN_STRATEGY } from './cross-origin.strat
 import { DomStrategy, DOM_STRATEGY } from './dom.strategy';
 
 export abstract class LoadingStrategy<T extends HTMLScriptElement | HTMLLinkElement = any> {
+  element: T;
+
   constructor(
     public path: string,
     protected domStrategy: DomStrategy = DOM_STRATEGY.AppendToHead(),
@@ -14,10 +16,10 @@ export abstract class LoadingStrategy<T extends HTMLScriptElement | HTMLLinkElem
   abstract createElement(): T;
 
   createStream<E extends Event>(): Observable<E> {
+    this.element = this.createElement();
+
     return of(null).pipe(
-      switchMap(() =>
-        fromLazyLoad<E>(this.createElement(), this.domStrategy, this.crossOriginStrategy),
-      ),
+      switchMap(() => fromLazyLoad<E>(this.element, this.domStrategy, this.crossOriginStrategy)),
     );
   }
 }
diff --git a/npm/ng-packs/packages/core/src/lib/tests/loading.strategy.spec.ts b/npm/ng-packs/packages/core/src/lib/tests/loading.strategy.spec.ts
index 4b950cf533..4215e6eee5 100644
--- a/npm/ng-packs/packages/core/src/lib/tests/loading.strategy.spec.ts
+++ b/npm/ng-packs/packages/core/src/lib/tests/loading.strategy.spec.ts
@@ -39,6 +39,7 @@ describe('ScriptLoadingStrategy', () => {
       const strategy = new ScriptLoadingStrategy(path, domStrategy, crossOriginStrategy);
 
       strategy.createStream<CustomEvent>().subscribe(event => {
+        expect(strategy.element.tagName).toBe('SCRIPT');
         expect(event.detail.crossOrigin).toBe('use-credentials');
         done();
       });
@@ -78,6 +79,7 @@ describe('StyleLoadingStrategy', () => {
       const strategy = new StyleLoadingStrategy(path, domStrategy, crossOriginStrategy);
 
       strategy.createStream<CustomEvent>().subscribe(event => {
+        expect(strategy.element.tagName).toBe('LINK');
         expect(event.detail.crossOrigin).toBe('use-credentials');
         done();
       });