Selenium Expected Conditions - possible to use 'or'?

0
0

I’m using Selenium 2 / WebDriver with the Python API, as follows:

from selenium.webdriver.support import expected_conditions as EC
# code that causes an ajax query to be run
WebDriverWait(driver, 10).until( EC.presence_of_element_located( \
(By.CSS_SELECTOR, "div.some_result")));

I want to wait for either a result to be returned (div.some_result) or a “Not found” string. Is that possible? Kind of:

WebDriverWait(driver, 10).until( \
EC
.presence_of_element_located( \
(By.CSS_SELECTOR, "div.some_result")) \
or
EC
.presence_of_element_located( \
(By.CSS_SELECTOR, "div.no_result")) \
);

I realise I could do this with a CSS selector (div.no_result, div.some_result), but is there a way to do it using the Selenium expected conditions method?

  • You must to post comments
0
0

I did it like this:

class AnyEc:
""" Use with WebDriverWait to combine expected_conditions
in an OR.
"""

def __init__(self, *args):
self
.ecs = args
def __call__(self, driver):
for fn in self.ecs:
try:
if fn(driver): return True
except:
pass

Then call it like…

from selenium.webdriver.support import expected_conditions as EC
# ...
WebDriverWait(driver, 10).until( AnyEc(
EC
.presence_of_element_located(
(By.CSS_SELECTOR, "div.some_result")),
EC
.presence_of_element_located(
(By.CSS_SELECTOR, "div.no_result")) ))

Obviously it would be trivial to also implement an AllEc class likewise.

Nb. the try: block is odd. I was confused because some ECs return true/false while others will throw exceptions for False. The Exceptions are caught by WebDriverWait so my AnyEc thing was producing odd results because the first one to throw an exception meant AnyEc didn’t proceed to the next test.

  • You must to post comments
Showing 1 result
Your Answer
Post as a guest by filling out the fields below or if you already have an account.
Name*
E-mail*
Website