From 1e19b042d3483eafc005e3e4815e7f555ca99d0b Mon Sep 17 00:00:00 2001
From: Recolic Keghart <root@recolic.net>
Date: Mon, 17 Feb 2020 05:37:22 -0800
Subject: [PATCH] Add another wait, to fetch correct result

---
 run.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/run.py b/run.py
index 4b4f826..f16b3fb 100755
--- a/run.py
+++ b/run.py
@@ -10,6 +10,7 @@ def wait_for_element_txt(element_txt):
         print('.', end='', flush=True)
         time.sleep(0.2)
     print()
+
 def wait_for_element_class(element_class_name):
     print('Waiting for loading.', end='', flush=True)
     while driver.find_elements_by_class_name(element_class_name) == []:
@@ -53,7 +54,9 @@ wait_for_element_class('dialog_button')
 driver.find_element_by_class_name('dialog_button').click() # first one is 'Ok', second one is 'Cancel'.
 
 wait_for_element_class('dialog_content')
-while 'If you have anything to comment,please click here, or you can add reviews in the form.' == driver.find_element_by_class_name('dialog_content').text:
+while 'If you have anything to comment' in driver.find_element_by_class_name('dialog_content').text:
+    wait_for_element_class('dialog_content')
+while '' == driver.find_element_by_class_name('dialog_content').text:
     wait_for_element_class('dialog_content')
 result = driver.find_element_by_class_name('dialog_content').text
 print(result)
-- 
GitLab