thib8956/wallabag-api-client
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

fix: use aiohttp ClientSession as a context manager

Browse Source
This commit is contained in:
Thibaud Gasser
2026-03-02 16:36:41 +01:00
parent 9ee4f70428
commit 4ef3de646e
1 changed files with 16 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

32
wallabag_api/test_wallabag.py
View File

@@ -35,24 +35,24 @@ class TestWallabag:
"original_url": "http://localhost", "original_url": "http://localhost",
} }
@pytest.fixture(scope="class") @pytest.fixture(scope="function")
async def configuration(self): async def configuration(self):
return load_test_config(CONFIG_PATH) return load_test_config(CONFIG_PATH)
@pytest.fixture(scope="class") @pytest.fixture(scope="function")
async def setup_class(self, configuration): async def setup_class(self, configuration):
access_token = await Wallabag.get_token(**configuration) access_token = await Wallabag.get_token(**configuration)
wallabag = Wallabag( async with aiohttp.ClientSession() as session:
host=configuration["host"], wallabag = Wallabag(
token=access_token, host=configuration["host"],
client_id=configuration["client_id"], token=access_token,
client_secret=configuration["client_secret"], client_id=configuration["client_id"],
aio_sess=aiohttp.ClientSession(), client_secret=configuration["client_secret"],
) aio_sess=session,
entry = await wallabag.post_entries(**self.entry_data) )
yield wallabag, entry entry = await wallabag.post_entries(**self.entry_data)
await wallabag.delete_entries(entry["id"]) yield wallabag, entry
await wallabag.aio_sess.close() await wallabag.delete_entries(entry["id"])
@pytest.mark.asyncio @pytest.mark.asyncio
async def test_get_token(self, configuration): async def test_get_token(self, configuration):
@@ -105,7 +105,7 @@ class TestWallabag:
entry_data = self.entry_data.copy() entry_data = self.entry_data.copy()
entry_data["url"] = "https://example.org" entry_data["url"] = "https://example.org"
data = await wallabag.post_entries(**entry_data) data = await wallabag.post_entries(**entry_data)
assert data is True assert data
assert isinstance(data["id"], int) assert isinstance(data["id"], int)
await wallabag.delete_entries(data["id"]) await wallabag.delete_entries(data["id"])
@@ -133,7 +133,7 @@ class TestWallabag:
entry_id = await self._create_entry(wallabag) entry_id = await self._create_entry(wallabag)
assert isinstance(entry_id, int) assert isinstance(entry_id, int)
data = await wallabag.delete_entries(entry_id) data = await wallabag.delete_entries(entry_id)
assert data is True assert data
@pytest.mark.asyncio @pytest.mark.asyncio
async def test_post_entry_tags(self, setup_class): async def test_post_entry_tags(self, setup_class):
@@ -143,7 +143,7 @@ class TestWallabag:
tags = ["foo", "bar"] tags = ["foo", "bar"]
assert isinstance(tags, list) assert isinstance(tags, list)
data = await wallabag.post_entry_tags(entry_id, tags) data = await wallabag.post_entry_tags(entry_id, tags)
assert data is True assert data
@pytest.mark.asyncio @pytest.mark.asyncio
async def test_delete_entry_tag(self, setup_class): async def test_delete_entry_tag(self, setup_class):