Как смоделировать метод модели из юнит-теста вида django?

Question

Я пытаюсь смоделировать вызов функции в нисходящем направлении, чтобы заставить мой unittest пройти представление DRF.

У меня есть метод create () в моей модели, который вызывает внешний upload_files () функция. Я пытаюсь понять, как смоделировать вызов upload_files () (но из проверки моих представлений)

Поскольку представление вызывает сериализатор, а сериализатор вызывает модель, мне нужно выяснить правильное пространство имен для того, чтобы издеваться над путем. Вот мой код, который показывает, что я пытаюсь сделать, однако я думаю, что путь, переданный mock.patch, неверен.

Получение этой ошибки при запуске unittest

ERROR: test_post (tests.api.views.test_stage_execution_view.StageExecutionViewTest)
ModuleNotFoundError: No module named 'myapp.models.stage_execution.StageExecution'; 'myapp.models.stage_execution' is not a package

тестирование вида из API-интерфейса drf:

# myapp/tests/api/views/test_stage_execution_view.py

class StageExecutionViewTest(APITestCase):
    @mock.patch('myapp.models.stage_execution.StageExecution.create.upload_files', side_effect=mocked_upload_files)
    def setUp(self, arg):
        self.stage_execution = StageExecution.manager.create(
            stage="test",
            job_execution="test",
            time_start="2018-12-20T12:05:00.258797-08:00",
        )

    @mock.patch('myapp.models.stage_execution.StageExecution.create.upload_files', side_effect=mocked_upload_files)
    def test_post(self):
        start_stage_execution_payload = {"bla"}
        response = self.client.post(
        "/api/v2/stage_execution", start_stage_execution_payload, format="json"
    )
        assert response.status_code == 201

модели:

#myapp/models/stage_execution.py

class StageExecution(JsonableModel):
    stage = models.CharField(max_length=150, blank=True, null=True)
    job_execution = models.CharField(max_length=150, blank=True, null=True)
    time_start = models.DateTimeField()

class StageExecutionManager(models.Manager):
    def get_queryset(self):
        return StageExecutionQuerySet(self.model, using=self._db)

    def create(self, **kwargs):
        from myapp.tasks import upload_files
        upload_files()

сериализатор:

# myapp/api/serializers/stage_execution_serializer.py

class StageExecutionSerializer(serializers.ModelSerializer):

    class Meta:
        model = StageExecution
        fields = (
            "stage",
            "job_execution",
            "time_start"

просмотров:

# myapp/api/views/stage_execution_view.py

class StageExecutionView(GenericAPIView):
    serializer_class = StageExecutionSerializer

    def get_queryset(self):
        return StageExecution.manager.all()

    def get(self, request, format=None):
        queryset = self.get_queryset()

    if queryset.exists():
        serializer = StageExecutionSerializer(queryset, many=True)
        return Response(serializer.data)

Answer 1

Вы можете использовать это, чтобы высмеивать upload_files :

@mock.patch('myapp.tasks.upload_files')
def setUp(self, mocked_upload_files):
    mocked_upload_files.return_value = ""  # You can change here to get return value you need