From 0e27a1a327207dd900e13cdc92d070a47bdd1752 Mon Sep 17 00:00:00 2001 From: shamoon <4887959+shamoon@users.noreply.github.com> Date: Thu, 8 Oct 2026 09:28:02 -0700 Subject: [PATCH] documentfactory --- src/documents/tests/test_api_trash.py | 27 +++++---------------------- 1 file changed, 5 insertions(+), 22 deletions(-) diff --git a/src/documents/tests/test_api_trash.py b/src/documents/tests/test_api_trash.py index f1d656eaf..8dbdee2db 100644 --- a/src/documents/tests/test_api_trash.py +++ b/src/documents/tests/test_api_trash.py @@ -6,6 +6,7 @@ from rest_framework.test import APITestCase from documents.models import Document from paperless_testing.dirs import DirectoriesMixin +from paperless_testing.factories import DocumentFactory from paperless_testing.factories import UserFactory from paperless_testing.permissions import grant_all_global @@ -293,32 +294,14 @@ class TestTrashAPI(DirectoriesMixin, APITestCase): - The version of the user's own document can be restored """ user2 = UserFactory(username="user2") - other_root = Document.objects.create( - title="other root", - checksum="other-root", - mime_type="application/pdf", - owner=user2, - ) - other_version = Document.objects.create( - title="other version", - checksum="other-version", - mime_type="application/pdf", - root_document=other_root, + other_version = DocumentFactory( + root_document=DocumentFactory(owner=user2), version_index=1, ) other_version.delete() - own_root = Document.objects.create( - title="own root", - checksum="own-root", - mime_type="application/pdf", - owner=self.user, - ) - own_version = Document.objects.create( - title="own version", - checksum="own-version", - mime_type="application/pdf", + own_version = DocumentFactory( owner=user2, - root_document=own_root, + root_document=DocumentFactory(owner=self.user), version_index=1, ) own_version.delete()