1 files changed, 10 insertions, 1 deletions
diff --git a/src/main/java/com/mavlushechka/a1qa/elements/BaseElement.java b/src/main/java/com/mavlushechka/a1qa/elements/BaseElement.java
index edcf1e7..8a7408d 100644
--- a/src/main/java/com/mavlushechka/a1qa/elements/BaseElement.java
+++ b/src/main/java/com/mavlushechka/a1qa/elements/BaseElement.java
@@ -1,10 +1,12 @@
 package com.mavlushechka.a1qa.elements;
 
 import com.mavlushechka.a1qa.driverUtils.WebDriverUtils;
+import com.mavlushechka.a1qa.driverUtils.WebDriverWaitFactory;
 import com.mavlushechka.a1qa.utils.LoggerUtils;
 import org.openqa.selenium.By;
 import org.openqa.selenium.Dimension;
 import org.openqa.selenium.WebElement;
+import org.openqa.selenium.support.ui.ExpectedConditions;
 
 public abstract class BaseElement {
 
@@ -42,7 +44,14 @@ public abstract class BaseElement {
         WebElement webElement = find();
 
         LoggerUtils.info("Getting visibility of the \"" + name + "\" element.");
-        return webElement.isDisplayed();
+        return WebDriverWaitFactory.createWebDriverWait().until(ExpectedConditions.visibilityOf(webElement)).isDisplayed();
+    }
+
+    public boolean isInvisible() {
+        WebElement webElement = find();
+
+        LoggerUtils.info("Getting invisibility of the \"" + name + "\" element.");
+        return WebDriverWaitFactory.createWebDriverWait().until(ExpectedConditions.invisibilityOf(webElement));
     }
 
     public String getName() {