Find a driver instance under WebElement (#15)

wgnet · Nov 19, 2016 · 19a8af0 · 19a8af0
1 parent a4720e6
commit 19a8af0
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/webium/base_page.py b/webium/base_page.py
@@ -10,7 +10,10 @@
 
 def is_element_present(self, element_name, just_in_dom=False, timeout=0):
     def _get_driver():
-        driver = getattr(self, '_driver', None)
+        try:
+            driver = getattr(self, '_driver')
+        except AttributeError:
+            driver = getattr(self, 'parent', None)
         if driver:
             return driver
         return get_driver()