paradizelost/paperless-ngx
1
0
Fork 0
mirror of https://github.com/paperless-ngx/paperless-ngx.git synced 2025-04-02 13:45:10 -05:00
Code Issues Packages Projects Releases Wiki Activity

Mock this way

Browse Source
This commit is contained in:
shamoon 2025-02-25 11:28:34 -08:00
parent 1f354f2d06
commit 500fa58cbe
1 changed files with 9 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
src/documents/tests/test_api_tasks.py
View File

@ -9,6 +9,7 @@ from rest_framework.test import APITestCase
from documents.models import PaperlessTask from documents.models import PaperlessTask
from documents.tests.utils import DirectoriesMixin from documents.tests.utils import DirectoriesMixin
from documents.views import TasksViewSet
class TestTasks(DirectoriesMixin, APITestCase): class TestTasks(DirectoriesMixin, APITestCase):
@ -311,8 +312,7 @@ class TestTasks(DirectoriesMixin, APITestCase):
self.assertEqual(returned_data["related_document"], "1234") self.assertEqual(returned_data["related_document"], "1234")
@mock.patch("documents.tasks.train_classifier") def test_run_train_classifier_task(self):
def test_run_train_classifier_task(self, mock_train_classifier):
""" """
GIVEN: GIVEN:
- A superuser - A superuser
@ -321,7 +321,13 @@ class TestTasks(DirectoriesMixin, APITestCase):
THEN: THEN:
- The task is run - The task is run
""" """
mock_train_classifier.return_value = "Task started" mock_train_classifier = mock.Mock(return_value="Task started")
TasksViewSet.TASK_AND_ARGS_BY_NAME = {
PaperlessTask.TaskName.TRAIN_CLASSIFIER: (
mock_train_classifier,
{"scheduled": False},
),
}
response = self.client.post( response = self.client.post(
self.ENDPOINT + "run/", self.ENDPOINT + "run/",
{"task_name": PaperlessTask.TaskName.TRAIN_CLASSIFIER}, {"task_name": PaperlessTask.TaskName.TRAIN_CLASSIFIER},