From b4316a07de9267c64964e7d7ea602e5d56e03bdf Mon Sep 17 00:00:00 2001 From: appel_c Date: Thu, 5 Jun 2025 14:38:43 +0200 Subject: [PATCH] test: add ttest for on_destroy hook --- tests/test_psi_device_base.py | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/tests/test_psi_device_base.py b/tests/test_psi_device_base.py index b40bddf..becfad9 100644 --- a/tests/test_psi_device_base.py +++ b/tests/test_psi_device_base.py @@ -89,6 +89,15 @@ def test_on_stage_hook(device): mock_on_stage.assert_called_once() +def test_on_destroy_hook(device): + """Test on destroy hook""" + assert device.destroyed is False + with mock.patch.object(device, "on_destroy") as mock_on_destroy: + device.destroy() + mock_on_destroy.assert_called_once() + assert device.destroyed is True + + def test_on_unstage_hook(device): """Test user method hooks""" with mock.patch.object(device, "on_unstage") as mock_on_unstage: