diff --git a/src/documents/tests/test_api_bulk_edit.py b/src/documents/tests/test_api_bulk_edit.py index ebef6a329..ba0e1139f 100644 --- a/src/documents/tests/test_api_bulk_edit.py +++ b/src/documents/tests/test_api_bulk_edit.py @@ -20,6 +20,7 @@ class TestBulkEditAPI(DirectoriesMixin, APITestCase): super().setUp() user = User.objects.create_superuser(username="temp_admin") + self.user = user self.client.force_authenticate(user=user) patcher = mock.patch("documents.bulk_edit.bulk_update_documents.delay") @@ -993,6 +994,7 @@ class TestBulkEditAPI(DirectoriesMixin, APITestCase): args, kwargs = m.call_args self.assertCountEqual(args[0], [self.doc2.id, self.doc3.id]) self.assertEqual(kwargs["metadata_document_id"], self.doc3.id) + self.assertEqual(kwargs["user"], self.user) @mock.patch("documents.serialisers.bulk_edit.merge") def test_merge_and_delete_insufficient_permissions(self, m): @@ -1092,6 +1094,7 @@ class TestBulkEditAPI(DirectoriesMixin, APITestCase): args, kwargs = m.call_args self.assertCountEqual(args[0], [self.doc2.id]) self.assertEqual(kwargs["pages"], [[1], [2, 3, 4], [5, 6], [7]]) + self.assertEqual(kwargs["user"], self.user) def test_split_invalid_params(self): response = self.client.post( diff --git a/src/documents/tests/test_bulk_edit.py b/src/documents/tests/test_bulk_edit.py index 37ebc35fc..fed93cd01 100644 --- a/src/documents/tests/test_bulk_edit.py +++ b/src/documents/tests/test_bulk_edit.py @@ -422,8 +422,9 @@ class TestPDFActions(DirectoriesMixin, TestCase): """ doc_ids = [self.doc1.id, self.doc2.id, self.doc3.id] metadata_document_id = self.doc1.id + user = User.objects.create(username="test_user") - result = bulk_edit.merge(doc_ids) + result = bulk_edit.merge(doc_ids, None, False, user) expected_filename = ( f"{'_'.join([str(doc_id) for doc_id in doc_ids])[:100]}_merged.pdf" @@ -525,7 +526,8 @@ class TestPDFActions(DirectoriesMixin, TestCase): """ doc_ids = [self.doc2.id] pages = [[1, 2], [3]] - result = bulk_edit.split(doc_ids, pages) + user = User.objects.create(username="test_user") + result = bulk_edit.split(doc_ids, pages, False, user) self.assertEqual(mock_consume_file.call_count, 2) consume_file_args, _ = mock_consume_file.call_args self.assertEqual(consume_file_args[1].title, "B (split 2)")