Fix imports + add test for AsyncClient context manager

parafoxia · Feb 17, 2023 · 9010e93 · 9010e93
1 parent 3606dca
commit 9010e93
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git a/analytix/client.py b/analytix/client.py
@@ -62,7 +62,6 @@
 import datetime
 import logging
 import os
-import signal
 import sys
 import typing as t
 import warnings

diff --git a/tests/client/test_async_client.py b/tests/client/test_async_client.py
@@ -108,6 +108,15 @@ async def test_client_repr(client: AsyncClient):
     assert f"{client!r}" == "AsyncClient(project_id='rickroll')"
 
 
+@mock.patch("builtins.open", mock.mock_open(read_data=create_secrets_file()))
+async def test_client_context_manager(client: AsyncClient):
+    async with AsyncClient("secrets.json") as other:
+        assert client == other
+        assert not other._session.closed
+
+    assert other._session.closed
+
+
 def test_client_active_tokens_property(client: AsyncClient):
     assert client.active_tokens == client._active_tokens
     assert not client.active_tokens