Skip to content

Commit 53ee84d

Browse files
committed
Correct test memory issue
1 parent 577859f commit 53ee84d

1 file changed

Lines changed: 13 additions & 6 deletions

File tree

unittests/test_rest_framework.py

Lines changed: 13 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -869,7 +869,7 @@ def test_notes_can_be_added_by_users_with_read_only_permissions(self):
869869

870870
@parameterized.expand(
871871
[
872-
("files", {"title": "test", "file": b"empty"}),
872+
("files", {"title": "test"}),
873873
("tags", {"tags": ["apple", "banana", "cherry"]}),
874874
],
875875
)
@@ -898,12 +898,19 @@ def test_related_objects(self, related_object_path, payload):
898898
# Retry adding the related object
899899
if related_object_path == "files":
900900
# Convert bytes to a mock uploaded file
901-
payload["file"] = SimpleUploadedFile(
902-
name="test_file.txt",
903-
content=payload["file"], # the b"empty"
904-
content_type="text/plain",
901+
response = self.client.post(
902+
relative_url,
903+
{
904+
"file": SimpleUploadedFile(
905+
name="test_file.txt",
906+
content=b"empty",
907+
content_type="text/plain",
908+
),
909+
**payload,
910+
},
905911
)
906-
response = self.client.post(relative_url, payload)
912+
else:
913+
response = self.client.post(relative_url, payload)
907914
self.assertIn(response.status_code, [200, 201], response.content[:1000])
908915

909916

0 commit comments

Comments
 (0)