Provide unlayout callback for amp-dailymotion (ampproject#33588)

* Provide unlayout callback for amp-dailymotion * fix spelling
rochapablo · Aug 30, 2021 · 006594e · 006594e
1 parent 6f9212d
commit 006594e
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 0 deletions.
diff --git a/extensions/amp-dailymotion/0.1/amp-dailymotion.js b/extensions/amp-dailymotion/0.1/amp-dailymotion.js
@@ -176,6 +176,16 @@ class AmpDailymotion extends AMP.BaseElement {
     return this.loadPromise(this.iframe_);
   }
 
+  /** @override */
+  unlayoutCallback() {
+    const iframe = this.iframe_;
+    if (iframe) {
+      this.element.removeChild(iframe);
+      this.iframe_ = null;
+    }
+    return true;
+  }
+
   /**
    * @param {!Event} event
    * @private

diff --git a/extensions/amp-dailymotion/0.1/test/test-amp-dailymotion.js b/extensions/amp-dailymotion/0.1/test/test-amp-dailymotion.js
@@ -83,5 +83,17 @@ describes.realWin(
         );
       });
     });
+
+    it('unlayout and relayout', async () => {
+      const dailymotion = await getDailymotion('x2m8jpp');
+      expect(dailymotion.querySelector('iframe')).to.exist;
+
+      const unlayoutResult = dailymotion.unlayoutCallback();
+      expect(unlayoutResult).to.be.true;
+      expect(dailymotion.querySelector('iframe')).to.not.exist;
+
+      await dailymotion.layoutCallback();
+      expect(dailymotion.querySelector('iframe')).to.exist;
+    });
   }
 );