From 60ad2661e16e300f60180cfb6389d96107d2311b Mon Sep 17 00:00:00 2001 From: Oliver Walters Date: Thu, 18 Jun 2026 09:30:10 +0000 Subject: [PATCH] Extend unit test --- src/backend/InvenTree/machine/test_api.py | 15 ++++++++++++--- 1 file changed, 12 insertions(+), 3 deletions(-) diff --git a/src/backend/InvenTree/machine/test_api.py b/src/backend/InvenTree/machine/test_api.py index 32a2809d2f..eb2e7584a9 100644 --- a/src/backend/InvenTree/machine/test_api.py +++ b/src/backend/InvenTree/machine/test_api.py @@ -287,14 +287,23 @@ class MachineAPITest(TestMachineRegistryMixin, InvenTreeAPITestCase): active=True, ) + restart_url = reverse('api-machine-restart', kwargs={'pk': machine.pk}) + + # Non-staff users must not be able to restart a machine + self.user.is_staff = False + self.user.save() + self.post(restart_url, expected_code=403) + + # Restore staff access + self.user.is_staff = True + self.user.save() + # verify machine status before restart response = self.get(reverse('api-machine-detail', kwargs={'pk': machine.pk})) self.assertEqual(response.data['status_text'], '') # restart the machine - response = self.post( - reverse('api-machine-restart', kwargs={'pk': machine.pk}), expected_code=200 - ) + self.post(restart_url, expected_code=200) # verify machine status after restart response = self.get(reverse('api-machine-detail', kwargs={'pk': machine.pk}))