From a7209f64136d7cc8bd446f6801d8695fc367608f Mon Sep 17 00:00:00 2001
From: Silvano Cerza <3314350+silvanocerza@users.noreply.github.com>
Date: Thu, 15 Feb 2024 19:33:34 +0100
Subject: [PATCH] Mark OpenAPIServiceConnector integration test as flaky
 (#7007)

---
 test/components/connectors/test_openapi_service.py | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/test/components/connectors/test_openapi_service.py b/test/components/connectors/test_openapi_service.py
index 29441d1f7..d9c669b1b 100644
--- a/test/components/connectors/test_openapi_service.py
+++ b/test/components/connectors/test_openapi_service.py
@@ -1,12 +1,14 @@
 import json
 import os
-
-import pytest
+import time
 from unittest.mock import MagicMock, Mock
 
+import flaky
+import pytest
 import requests
 from openapi3 import OpenAPI
 from openapi3.schemas import Model
+
 from haystack.components.connectors import OpenAPIServiceConnector
 from haystack.dataclasses import ChatMessage
 
@@ -140,6 +142,7 @@ class TestOpenAPIServiceConnector:
             " to get the raw data from the service response"
         )
 
+    @flaky.flaky(max_runs=5, rerun_filter=lambda *_: time.sleep(5))
     @pytest.mark.integration
     @pytest.mark.skipif(not os.getenv("GITHUB_TOKEN"), reason="GITHUB_TOKEN is not set")
     def test_run(self, genuine_fc_message, test_files_path):