bkinnightskytw
/
InvenTree
mirror of https://github.com/inventree/InvenTree.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Extend unit test

This commit is contained in:
Oliver Walters 2026-06-18 09:30:10 +00:00
parent 8c9f73e607
commit 60ad2661e1
1 changed files with 12 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
src/backend/InvenTree/machine/test_api.py
View File

@ -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}))