From dfdd329ddf0512a484b67dd452165c57dbb4c45c Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Fri, 5 Jan 2024 22:28:34 +0530 Subject: [PATCH] (ci/cd) pytest event loop fixture --- litellm/tests/test_proxy_server_keys.py | 17 +++++++++++++++++ 1 file changed, 17 insertions(+) diff --git a/litellm/tests/test_proxy_server_keys.py b/litellm/tests/test_proxy_server_keys.py index 7f25ae9b8..b372681a5 100644 --- a/litellm/tests/test_proxy_server_keys.py +++ b/litellm/tests/test_proxy_server_keys.py @@ -61,6 +61,23 @@ async def wrapper_startup_event(): await startup_event() +import asyncio + + +@pytest.yield_fixture +def event_loop(): + """Create an instance of the default event loop for each test case.""" + policy = asyncio.WindowsSelectorEventLoopPolicy() + res = policy.new_event_loop() + asyncio.set_event_loop(res) + res._close = res.close + res.close = lambda: None + + yield res + + res._close() + + # Here you create a fixture that will be used by your tests # Make sure the fixture returns TestClient(app) @pytest.fixture(scope="function")