diff --git a/game/end_to_end_tests/test_level_editor.py b/game/end_to_end_tests/test_level_editor.py index 13938e491..084ebe526 100644 --- a/game/end_to_end_tests/test_level_editor.py +++ b/game/end_to_end_tests/test_level_editor.py @@ -306,6 +306,10 @@ def test_pigeon(self): "city" ) + assert WebDriverWait(self.selenium, DELAY_TIME).until( + EC.presence_of_element_located((By.XPATH, "//image[contains(@href, 'pigeon')]")) + ) + pigeon = self.selenium.find_elements(By.XPATH, "//image[contains(@href, 'pigeon')]") nonexistent_cow = self.selenium.find_elements(By.XPATH, "//image[contains(@href, 'Clarice')]")