From 4af53b76b726eb4bf951eb65ff71bd504584f415 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?J=C3=A9r=C3=A9my=20Lal?= Date: Tue, 30 Jan 2024 08:31:35 +0100 Subject: [PATCH] Replace aiounittest by native IsolatedAsyncioTestCase --- tests/requirements.txt | 1 - tests/unit/http/test_async_http_client.py | 9 ++++----- tests/unit/rest/test_client.py | 3 +-- 3 files changed, 5 insertions(+), 8 deletions(-) diff --git a/tests/requirements.txt b/tests/requirements.txt index 679f8e13d..a6224c982 100644 --- a/tests/requirements.txt +++ b/tests/requirements.txt @@ -2,7 +2,6 @@ Sphinx>=1.8.0 mock pytest pytest-cov -aiounittest flake8 wheel>=0.22.0 cryptography diff --git a/tests/unit/http/test_async_http_client.py b/tests/unit/http/test_async_http_client.py index df4360684..711ca6b3c 100644 --- a/tests/unit/http/test_async_http_client.py +++ b/tests/unit/http/test_async_http_client.py @@ -1,5 +1,4 @@ -import aiounittest - +import unittest from aiohttp import ClientSession from mock import patch, AsyncMock from twilio.http.async_http_client import AsyncTwilioHttpClient @@ -19,7 +18,7 @@ async def text(self): return self._text -class TestAsyncHttpClientRequest(aiounittest.AsyncTestCase): +class TestAsyncHttpClientRequest(unittest.IsolatedAsyncioTestCase): def setUp(self): self.session_mock = AsyncMock(wraps=ClientSession) self.session_mock.request.return_value = MockResponse("test", 200) @@ -58,7 +57,7 @@ async def test_invalid_request_timeout_raises_exception(self): await self.client.request("doesnt matter", "doesnt matter", timeout=-1) -class TestAsyncHttpClientRetries(aiounittest.AsyncTestCase): +class TestAsyncHttpClientRetries(unittest.IsolatedAsyncioTestCase): def setUp(self): self.session_mock = AsyncMock(wraps=ClientSession) self.session_mock.request.side_effect = [ @@ -91,7 +90,7 @@ async def test_request_retries_until_max(self): self.assertEqual(response.text, "Error") -class TestAsyncHttpClientSession(aiounittest.AsyncTestCase): +class TestAsyncHttpClientSession(unittest.IsolatedAsyncioTestCase): def setUp(self): self.session_patcher = patch("twilio.http.async_http_client.ClientSession") self.session_constructor_mock = self.session_patcher.start() diff --git a/tests/unit/rest/test_client.py b/tests/unit/rest/test_client.py index 9623dc52a..41a582a2e 100644 --- a/tests/unit/rest/test_client.py +++ b/tests/unit/rest/test_client.py @@ -1,5 +1,4 @@ import unittest -import aiounittest from mock import AsyncMock, Mock from twilio.http.response import Response @@ -106,7 +105,7 @@ def test_set_user_agent_extensions(self): self.assertEqual(user_agent_extensions, expected_user_agent_extensions) -class TestClientAsyncRequest(aiounittest.AsyncTestCase): +class TestClientAsyncRequest(unittest.IsolatedAsyncioTestCase): def setUp(self): self.mock_async_http_client = AsyncMock() self.mock_async_http_client.request.return_value = Response(200, "test")