diff --git a/tests/test_forms.py b/tests/test_forms.py index 41dcc34..30d141d 100644 --- a/tests/test_forms.py +++ b/tests/test_forms.py @@ -159,9 +159,14 @@ class TestHeavySelect2Mixin(TestSelect2Mixin): elem1, elem2 = driver.find_elements_by_css_selector('.select2-selection') elem1.click() - result1 = driver.find_element_by_css_selector('.select2-results li:first-child').text + + result1 = WebDriverWait(driver, 10).until( + expected_conditions.presence_of_element_located((By.CSS_SELECTOR, '.select2-results li:first-child')) + ).text elem2.click() - result2 = driver.find_element_by_css_selector('.select2-results li:first-child').text + result2 = WebDriverWait(driver, 10).until( + expected_conditions.presence_of_element_located((By.CSS_SELECTOR, '.select2-results li:first-child')) + ).text assert result1 != result2