diff --git a/tests/test_services_trash.py b/tests/test_services_trash.py index cb169ba..cb7cec6 100644 --- a/tests/test_services_trash.py +++ b/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 - session.execute = AsyncMock(side_effect=[_rowcount_result(i) for i in [2, 0, 1, 1, 0, 0, 0]]) + # 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]]) 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 session.execute.await_count == 7 + assert n == 6 + assert session.execute.await_count == 6 def test_owner_clause_scopes_every_model():