Improve tests for pruning images

This commit is contained in:
grossmj committed 2025-01-03 19:56:41 +07:00
1 parent e17a79e261
commit ac4c5f5985
3 files changed
+22 -7

No files matched your search

+4 -1
View File
@@ -261,10 +261,13 @@ class TestImageRoutes:
async def test_prune_images(self, app: FastAPI, client: AsyncClient, db_session: AsyncSession) -> None:
images_repo = ImagesRepository(db_session)
images_in_db = await images_repo.get_images()
assert len(images_in_db) != 0
response = await client.delete(app.url_path_for("prune_images"))
assert response.status_code == status.HTTP_204_NO_CONTENT
images_repo = ImagesRepository(db_session)
images_in_db = await images_repo.get_images()
assert len(images_in_db) == 0