Add attribute_to_be_include method to expected_conditions when the at… (#9200)

* Add attribute_to_be_include method to expected_conditions when the attribute given is include the element's attributes

Co-authored-by: David Burns <david.burns@theautomatedtester.co.uk>

Author: sunnyyukaige

py/selenium/webdriver/support/expected_conditions.py

@@ -364,6 +364,21 @@ def _predicate(driver):
     return _predicate
 
 
+def element_attribute_to_include(locator, attribute_):
+    """ An expectation for checking if the given attribute is include in the
+    specified element.
+    locator, attribute
+    """
+    def _predicate(driver):
+        try:
+            element_attribute = driver.find_element(*locator).get_attribute(attribute_)
+            return element_attribute is not None
+        except StaleElementReferenceException:
+            return False
+
+    return _predicate
+
+
 def any_of(*expected_conditions):
     """ An expectation that any of multiple expected conditions is true.
     Equivalent to a logical 'OR'.

py/test/selenium/webdriver/common/webdriverwait_tests.py

@@ -315,3 +315,11 @@ def testExpectedConditionAlertIsPresent(driver, pages):
     alert = driver.switch_to.alert
     assert 'alerty' == alert.text
     alert.dismiss()
+
+
+def testExpectedConditionAttributeToBeIncludeInElement(driver, pages):
+    pages.load('booleanAttributes.html')
+    with pytest.raises(TimeoutException):
+        WebDriverWait(driver, 1).until(EC.element_attribute_to_include((By.ID, 'inputRequired'), 'test'))
+    value = WebDriverWait(driver, 1).until(EC.element_attribute_to_include((By.ID, 'inputRequired'), 'value'))
+    assert value is not None