tests/test_services_trash.py

@@ -97,14 +97,14 @@ def _rowcount_result(n):
 @pytest.mark.asyncio
 async def test_restore_clears_batch_across_all_models():
     session = _make_mock_session()
-    # 7 models, each returns a rowcount
+    # 6 soft-deletable models, each returns a rowcount
-    session.execute = AsyncMock(side_effect=[_rowcount_result(i) for i in [2, 0, 1, 1, 0, 0, 0]])
+    session.execute = AsyncMock(side_effect=[_rowcount_result(i) for i in [2, 0, 1, 1, 0, 0]])
     with patch("scribe.services.trash.async_session") as cls:
         cls.return_value = session
         from scribe.services.trash import restore
         n = await restore(user_id=1, batch_id="abc")
     assert n == 4
-    assert session.execute.await_count == 7
+    assert session.execute.await_count == 6
     assert session.commit.called
 
 
@@ -123,13 +123,13 @@ async def test_purge_expired_skips_when_retention_zero():
 @pytest.mark.asyncio
 async def test_purge_expired_deletes_across_models_when_positive():
     session = _make_mock_session()
-    session.execute = AsyncMock(side_effect=[_rowcount_result(1) for _ in range(7)])
+    session.execute = AsyncMock(side_effect=[_rowcount_result(1) for _ in range(6)])
     with patch("scribe.services.trash.async_session") as cls:
         cls.return_value = session
         from scribe.services.trash import purge_expired
         n = await purge_expired(1, 90)
-    assert n == 7
+    assert n == 6
-    assert session.execute.await_count == 7
+    assert session.execute.await_count == 6
 
 
 def test_owner_clause_scopes_every_model():