refactor(focus-origin-monitor): complete stream when stopping monitor (…

…angular#3404)
crisbeto · Mar 4, 2017 · 94adecd · 94adecd
1 parent 63d4359
commit 94adecd
Showing 1 changed file with 8 additions and 5 deletions.
diff --git a/src/lib/core/style/focus-origin-monitor.ts b/src/lib/core/style/focus-origin-monitor.ts
@@ -102,12 +102,15 @@ export class FocusOriginMonitor {
    * @param element The element to stop monitoring.
    */
   unmonitor(element: Element): void {
-    if (!this._elementInfo.has(element)) {
-      return;
+    let elementInfo = this._elementInfo.get(element);
+
+    if (elementInfo) {
+      elementInfo.unlisten();
+      elementInfo.subject.complete();
+
+      this._setClasses(element, null);
+      this._elementInfo.delete(element);
     }
-    this._elementInfo.get(element).unlisten();
-    this._setClasses(element, null);
-    this._elementInfo.delete(element);
   }
 
   /**