add on delete hook (#1014)

Co-authored-by: Schwannden Kuo <schwannden@mobagel.com>
This commit is contained in:
schwannden
2022-06-20 19:26:02 +08:00
committed by GitHub
parent 0efbab42cd
commit 3bb3728261
4 changed files with 90 additions and 1 deletions

View File

@ -120,6 +120,8 @@ class UserManagerMock(BaseTestUserManager[models.UP]):
on_after_forgot_password: MagicMock
on_after_reset_password: MagicMock
on_after_update: MagicMock
on_before_delete: MagicMock
on_after_delete: MagicMock
_update: MagicMock
@ -475,6 +477,8 @@ def make_user_manager(mocker: MockerFixture):
mocker.spy(user_manager, "on_after_forgot_password")
mocker.spy(user_manager, "on_after_reset_password")
mocker.spy(user_manager, "on_after_update")
mocker.spy(user_manager, "on_before_delete")
mocker.spy(user_manager, "on_after_delete")
mocker.spy(user_manager, "_update")
return user_manager

View File

@ -532,6 +532,10 @@ class TestDelete:
):
await user_manager.delete(user)
assert user_manager.on_before_delete.called is True
assert user_manager.on_after_delete.called is True
@pytest.mark.asyncio
@pytest.mark.manager