add async tests

Author: skysomorphic

poetry.lock

@@ -2,14 +2,18 @@
 name = "qbreader"
 version = "1.0.0-beta.1"
 description = "The official qbreader API python wrapper"
-authors = ["Sky \"g3ner1c\" Hong <g3ner1c-sky@pm.me>", "Rohan Arni", "Geoffrey Wu <geoffreywu1000@gmail.com>"]
+authors = [
+    "Sky \"g3ner1c\" Hong <g3ner1c-sky@pm.me>",
+    "Rohan Arni",
+    "Geoffrey Wu <geoffreywu1000@gmail.com>",
+]
 license = "MIT"
 readme = "README.md"
 homepage = "https://github.com/qbreader/python-module"
 repository = "https://github.com/qbreader/python-module"
 documentation = "https://github.com/qbreader/python-module#readme"
 keywords = ["quizbowl", "quiz bowl"]
-classifiers=[
+classifiers = [
     "Development Status :: 4 - Beta",
     "Intended Audience :: Developers",
     "Programming Language :: Python :: 3",
@@ -34,7 +38,7 @@ types-requests = "^2.31.0.1"
 tox = "^4.6.4"
 pytest = "^7.4.0"
 pytest-cov = "^4.1.0"
-pytest-async = "^0.1.1"
+pytest-asyncio = "^0.21.1"
 
 [tool.black]
 line-length = 88
@@ -49,6 +53,7 @@ convention = "numpy"
 [tool.pytest.ini_options]
 addopts = "-v --cov=qbreader --cov-report=term-missing --cov-report=xml --cov-report=html"
 testpaths = "tests"
+asyncio_mode = "strict"
 
 [build-system]
 requires = ["poetry-core>=1.0.0"]

pyproject.toml

@@ -1,6 +1,6 @@
 """Directly access the qbreader API asynchronously."""
 
-from typing import Optional, Self
+from typing import Optional, Self, Type
 
 import aiohttp
 
@@ -23,24 +23,40 @@
 class Async:
     """The asynchronous qbreader API wrapper."""
 
-    def __init__(self, session: Optional[aiohttp.ClientSession] = None):
-        """Initialize a new Async instance.
+    session: aiohttp.ClientSession
+
+    @classmethod
+    async def create(
+        cls: Type[Self], session: Optional[aiohttp.ClientSession] = None
+    ) -> Self:
+        """Create a new Async instance. `__init__()` is not async, so this is necessary.
 
         Parameters
         ----------
         session : aiohttp.ClientSession, optional
             The aiohttp session to use for requests. If none is provided, a new session
             will be created.
+
+        Returns
+        -------
+        Async
+            The new Async instance.
         """
+        self = cls()
         self.session = session or aiohttp.ClientSession()
+        return self
+
+    async def close(self: Self) -> None:
+        """Close the aiohttp session."""
+        await self.session.close()
 
     async def __aenter__(self: Self) -> Self:
         """Enter an async context."""
         return self
 
     async def __aexit__(self: Self) -> None:
         """Exit an async context."""
-        await self.session.close()
+        await self.close()
 
     async def query(
         self: Self,
@@ -323,7 +339,7 @@ async def random_bonus(
                 raise Exception(str(response.status) + " bad request")
 
             json = await response.json()
-            return tuple(Bonus.from_json(b) for b in await json["bonuses"])
+            return tuple(Bonus.from_json(b) for b in json["bonuses"])
 
     async def random_name(self: Self) -> str:
         """Get a random adjective-noun pair that can be used as a name.

qbreader/asynchronous.py

@@ -168,7 +168,7 @@ async def check_answer_async(
         cls: Type[Self],
         answerline: str,
         givenAnswer: str,
-        session: aiohttp.ClientSession,
+        session: aiohttp.ClientSession | None = None,
     ) -> Self:
         """Asynchronously create an AnswerJudgement given an answerline and an answer.
 
@@ -199,6 +199,15 @@ async def check_answer_async(
 
         data = {"answerline": answerline, "givenAnswer": givenAnswer}
 
+        if session is None:
+            async with aiohttp.ClientSession() as session:
+                async with session.get(url, params=data) as response:
+                    if response.status != 200:
+                        raise Exception(str(response.status) + " bad request")
+
+                    json = await response.json()
+                    return cls.from_json(json)
+
         async with session.get(url, params=data) as response:
             if response.status != 200:
                 raise Exception(str(response.status) + " bad request")
@@ -258,7 +267,7 @@ def check_answer_sync(self, givenAnswer: str) -> AnswerJudgement:
         return AnswerJudgement.check_answer_sync(self.formatted_answer, givenAnswer)
 
     async def check_answer_async(
-        self, givenAnswer: str, session: aiohttp.ClientSession
+        self, givenAnswer: str, session: aiohttp.ClientSession | None = None
     ) -> AnswerJudgement:
         """Asynchronously check whether an answer is correct."""
         return await AnswerJudgement.check_answer_async(

qbreader/types.py

@@ -28,6 +28,17 @@ def assert_exception(
         func(*args, **kwargs)
 
 
+async def async_assert_exception(
+    func: Callable,
+    exception,
+    *args,
+    **kwargs,
+):
+    """Assert that an async function raises an exception."""
+    with pytest.raises(exception):
+        await func(*args, **kwargs)
+
+
 def assert_warning(
     func: Callable,
     warning,
@@ -37,3 +48,14 @@ def assert_warning(
     """Assert that a function raises a warning."""
     with pytest.warns(warning):
         return func(*args, **kwargs)
+
+
+async def async_assert_warning(
+    func: Callable,
+    warning,
+    *args,
+    **kwargs,
+):
+    """Assert that an async function raises a warning."""
+    with pytest.warns(warning):
+        return await func(*args, **kwargs)