Mark OpenAPIServiceConnector integration test as flaky (#7007)

This commit is contained in:
Silvano Cerza 2024-02-15 19:33:34 +01:00 committed by GitHub
parent 8a101701fa
commit a7209f6413
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -1,12 +1,14 @@
import json import json
import os import os
import time
import pytest
from unittest.mock import MagicMock, Mock from unittest.mock import MagicMock, Mock
import flaky
import pytest
import requests import requests
from openapi3 import OpenAPI from openapi3 import OpenAPI
from openapi3.schemas import Model from openapi3.schemas import Model
from haystack.components.connectors import OpenAPIServiceConnector from haystack.components.connectors import OpenAPIServiceConnector
from haystack.dataclasses import ChatMessage from haystack.dataclasses import ChatMessage
@ -140,6 +142,7 @@ class TestOpenAPIServiceConnector:
" to get the raw data from the service response" " 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.integration
@pytest.mark.skipif(not os.getenv("GITHUB_TOKEN"), reason="GITHUB_TOKEN is not set") @pytest.mark.skipif(not os.getenv("GITHUB_TOKEN"), reason="GITHUB_TOKEN is not set")
def test_run(self, genuine_fc_message, test_files_path): def test_run(self, genuine_fc_message, test_files_path):