starred/netbox
1
0
Fork 0
mirror of https://github.com/netbox-community/netbox.git synced 2026-04-23 09:18:49 +02:00
Code Issues 273 Packages Projects Releases 10 Wiki Activity

cleanup

Browse Source
This commit is contained in:
Arthur
2026-03-30 17:28:19 -07:00
parent 9954a9e8a9
commit 37be38b582
1 changed files with 8 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
netbox/extras/tests/test_api.py
View File

@@ -1472,11 +1472,14 @@ class ScriptUploadTest(APITestCase):
"""POST with data_source + data_file (JSON) creates a ScriptModule with the correct file_path.""" """POST with data_source + data_file (JSON) creates a ScriptModule with the correct file_path."""
self.add_permissions('extras.add_scriptmodule', 'core.add_managedfile') self.add_permissions('extras.add_scriptmodule', 'core.add_managedfile')
mock_storage = MagicMock() mock_storage = MagicMock()
mock_storage.open.return_value.__enter__ = MagicMock(return_value=MagicMock()) # Patch storages in both the serializer (for _sync_data_file) and the model
mock_storage.open.return_value.__exit__ = MagicMock(return_value=False) # (for ManagedFile.sync_data(), which is called by SyncedDataMixin.clean() during
with patch('extras.api.serializers_.scripts.storages') as mock_storages: # ValidatedModelSerializer.validate() → full_clean()).
mock_storages.create_storage.return_value = mock_storage with patch('extras.api.serializers_.scripts.storages') as mock_ser_storages, \
mock_storages.backends = {'scripts': {}} patch('core.models.files.storages') as mock_model_storages:
for m in (mock_ser_storages, mock_model_storages):
m.create_storage.return_value = mock_storage
m.backends = {'scripts': {}}
response = self.client.post( response = self.client.post(
self.url_list, self.url_list,
{'data_source': self.data_source.pk, 'data_file': self.data_file.pk}, {'data_source': self.data_source.pk, 'data_file': self.data_file.pk},