mirror of
https://github.com/fastapi-users/fastapi-users.git
synced 2025-11-02 04:05:19 +08:00
add on delete hook (#1014)
Co-authored-by: Schwannden Kuo <schwannden@mobagel.com>
This commit is contained in:
@ -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
|
||||
|
||||
|
||||
@ -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
|
||||
|
||||
Reference in New Issue
Block a user