diff options
| -rw-r--r-- | app/test/test_next_essay.py | 18 |
1 files changed, 10 insertions, 8 deletions
diff --git a/app/test/test_next_essay.py b/app/test/test_next_essay.py index 9075d7c..c34f540 100644 --- a/app/test/test_next_essay.py +++ b/app/test/test_next_essay.py @@ -40,14 +40,16 @@ def test_next(): # click Next differ = 0 - elem = driver.find_element_by_xpath('//form[1]/input[1]') - elem.click() - driver.save_screenshot('./app/test/test_next_essay_pic1.png') - elem = driver.find_element_by_id('text-content') - current_essay_content = elem.text - - if current_essay_content != essay_content: - diff = 1 + for i in range(3): + elem = driver.find_element_by_xpath('//form[1]/input[1]') + elem.click() + driver.save_screenshot('./app/test/test_next_essay_pic1.png') + elem = driver.find_element_by_id('text-content') + current_essay_content = elem.text + + if current_essay_content != essay_content: + diff = 1 + break assert diff == 1 |