Replace sleep in data_setup.py with wait

- Issue: #456

1 files changed, 3 insertions, 2 deletions

diff --git a/service/test/functional/features/steps/data_setup.py b/service/test/functional/features/steps/data_setup.py
index c6eab8c0..fb825aba 100644
--- a/service/test/functional/features/steps/data_setup.py
+++ b/service/test/functional/features/steps/data_setup.py
@@ -13,10 +13,10 @@
 #
 # You should have received a copy of the GNU Affero General Public License
 # along with Pixelated. If not, see <http://www.gnu.org/licenses/>.
-from time import sleep
 from uuid import uuid4
 from test.support.integration import MailBuilder
 from behave import given
+from common import wait_for_condition
 
 
 @given('I have a mail in my inbox')
@@ -25,6 +25,7 @@ def add_mail_impl(context):
 
     input_mail = MailBuilder().with_subject(subject).build_input_mail()
     context.client.add_mail_to_inbox(input_mail)
-    sleep(5)  # we need to wait for the mail to be indexed (3 secs at least)
+
+    wait_for_condition(context, lambda _: context.client.search_engine.search(subject)[1] > 0, poll_frequency=0.1)
 
     context.last_subject = subject