update: adjust steps to use get iframe body

Exabyte-io · Apr 10, 2024 · 9f96b39 · 9f96b39
1 parent cacd4fa
commit 9f96b39
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 11 deletions.
diff --git a/tests/cypress/support/step_definitions/I see file opened.ts b/tests/cypress/support/step_definitions/I see file opened.ts
@@ -4,6 +4,6 @@ import MaterialDesignerPage from "../widgets/MaterialDesignerPage";
 
 When("I see file {string} opened", (filename: string) => {
     const { jupyterLiteSession } = new MaterialDesignerPage().designerWidget;
-    jupyterLiteSession.waitForVisible("md");
+    jupyterLiteSession.waitForVisible();
     jupyterLiteSession.checkFileOpened(filename);
 });
diff --git a/tests/cypress/support/widgets/JupyterLiteSession.ts b/tests/cypress/support/widgets/JupyterLiteSession.ts
@@ -1,5 +1,5 @@
 import Widget from "./Widget";
-
+// TODO: celanup eveyrthing
 const selectors = {
     wrapper: "iframe#jupyter-lite-iframe",
     main: "#main",
@@ -13,19 +13,14 @@ export default class JupyterLiteSession extends Widget {
         this.wrappedSelectors = this.getWrappedSelectors(selectors);
     }
 
-    waitForVisible(timeout: BrowserTimeout) {
-        return cy
-            .get("#jupyter-lite-iframe", { log: true, timeout: 20000 })
-            .iframe()
-            .get("body #main")
-            .should("exist");
+    waitForVisible() {
+        return cy.getIframeBody("#jupyter-lite-iframe").find("#main").should("exist");
     }
 
     checkFileOpened(fileName: string) {
         return cy
-            .get("#jupyter-lite-iframe", { log: true, timeout: 20000 })
-            .iframe()
-            .get("body #main")
+            .getIframeBody("#jupyter-lite-iframe")
+            .find("#main")
             .find(`li[title="Name: ${fileName}"]`)
             .should("exist");
     }