app/test/test_next_essay.py: click Next three times so that it is very unlikely the essay won't change.

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