From 7747d78b9db669afe4b873d5c1b14b1c200aa16b Mon Sep 17 00:00:00 2001
From: Sashank Aryal <sashank2012@gmail.com>
Date: Fri, 4 Oct 2024 13:00:38 -0500
Subject: [PATCH] wait until buffering is done

---
 e2e-pw/src/oss/poms/modal/imavid-controls.ts | 12 ++++++++++++
 1 file changed, 12 insertions(+)

diff --git a/e2e-pw/src/oss/poms/modal/imavid-controls.ts b/e2e-pw/src/oss/poms/modal/imavid-controls.ts
index acc7af3f25..81228ef68b 100644
--- a/e2e-pw/src/oss/poms/modal/imavid-controls.ts
+++ b/e2e-pw/src/oss/poms/modal/imavid-controls.ts
@@ -32,10 +32,22 @@ export class ModalImaAsVideoControlsPom {
     return timelineId;
   }
 
+  private async waitUntilBufferingIsFinished() {
+    await this.page.waitForFunction(
+      () =>
+        document
+          .querySelector("[data-cy=imavid-playhead]")
+          ?.getAttribute("data-playhead-state") !== "buffering"
+    );
+  }
+
   private async togglePlay() {
+    await this.waitUntilBufferingIsFinished();
+
     let currentPlayHeadStatus = await this.playPauseButton.getAttribute(
       "data-playhead-state"
     );
+
     const original = currentPlayHeadStatus;
 
     // keep pressing space until play head status changes