From aed9b47cd74f25eea3cffbcfe796290676cf5646 Mon Sep 17 00:00:00 2001
From: Ruslan Khafizov <ruslan.khafizov@gmail.com>
Date: Mon, 13 Jan 2025 09:13:05 +0800
Subject: [PATCH] Add polling for test

---
 natural4-server/tests/test_endpoint.py | 13 +++++++++++--
 1 file changed, 11 insertions(+), 2 deletions(-)

diff --git a/natural4-server/tests/test_endpoint.py b/natural4-server/tests/test_endpoint.py
index 854e96a..f3f8c36 100644
--- a/natural4-server/tests/test_endpoint.py
+++ b/natural4-server/tests/test_endpoint.py
@@ -1,4 +1,4 @@
-from time import sleep
+from time import sleep, time
 from sanic import Sanic
 from sanic_testing.reusable import ReusableClient
 from sanic.application.constants import ServerStage
@@ -32,5 +32,14 @@ def test_post(app: Sanic, post_data):
 
             sleep(15)
 
-            request, response_pdf = client.get(f'{workdir_url}/pdf/LATEST.pdf')
+            start_time = time()
+            while time() - start_time < 60:
+                print('Waiting for PDF...')
+                request, response_pdf = client.get(f'{workdir_url}/pdf/LATEST.pdf')
+                if response_pdf.status != 200:
+                    sleep(5)
+                    continue
+                else:
+                    break
+
             assert response_pdf.status == 200