Django: как макетировать класс в представлении API

Question

Название может быть немного запутанным.

Скажем, у меня есть APIView с post методом.Внутри метода post я представил класс, у которого есть собственный метод.В данном случае это класс, который занимается загрузкой на S3, и это то, что я хочу пропустить при запуске unittest.

class SomeView(APIView):
    def post(self):
        # do something here
        input1 = some_process(payload_arg1)
        input2 = some_other_process(payload_arg2)
        uploader = S3Uploader()
        s3_response = uploader.upload_with_aux_fxn(input1, input2)
        if s3_response['status_code'] == 200:
            # do something else
            return Response('Good job I did it!', status_code=200)
        else:
            return Response('noooo you're horrible!', status_code=400)

Реальный код имеет разные вызовы функций и ответы, очевидно.

Теперь мне нужно высмеять эти uploader и uploader.upload_with_aux_fxn, чтобы я на самом деле не звонил S3.Как мне поиздеваться?

Я пробовал в своем тестовом скрипте

from some_place import S3Uploader
class SomeViewTestCase(TestCase): 
    def setUp(self):        
        self.client = APIClient()
        uploader_mock = S3Uploader()
        uploader_mock.upload_support_doc = MagicMock(return_value={'status_code': 200, 'message': 'asdasdad'}
        response = self.client.post(url, payload, format='multipart')

Но я все же запустил загрузку S3 (как показывает файл в S3).Как мне правильно подшучивать над этим?

EDIT1:

Моя попытка исправить

def setUp(self):
    self.factory = APIRequestFactory()
    self.view = ViewToTest.as_view()
    self.url = reverse('some_url')


@patch('some_place.S3Uploader', FakeUploader)
def test_uplaod(self):
    payload = {'some': 'data', 'other': 'stuff'}
    request = self.factory.post(self.url, payload, format='json')
    force_authenticate(request, user=self.user)
    response = self.view(request)

, где FakeUplaoder

class FakeUplaoder(object):
    def __init__(self):
        pass
    def upload_something(self, data, arg1, arg2, arg3):
        return {'status_code': 200, 'message': 'unit test', 's3_path': 
                'unit/test/path.pdf'}

    def downlaod_something(self, s3_path):
        return {'status_code': 200, 'message': '', 'body': 'some base64 
                stuff'}

к сожалению этоне успешный.Я все еще использую класс

РЕДАКТИРОВАТЬ 2:

Я использую Django 1.11 и Python 2.7, на случай, если людям понадобится эта информация

Answer 1

Взгляните на vcrpy .Он записывает запрос во внешний API один раз, а затем повторяет ответ каждый раз, когда вы запускаете тесты.Не нужно ничего издеваться вручную.

Answer 2

Полагаю, что правильным подходом было бы сохранить файл в модели с FileField, а затем подключить Boto для обработки загрузки в производственном сценарии.
Внимательно посмотрите на:
https://docs.djangoproject.com/en/2.2/ref/models/fields/#filefield
и
https://django -storages.readthedocs.io / en / latest / backends / amazon-S3.html # model
этот подход позволит сохранить поведение Django по умолчанию, что сделает вещи более тестируемыми с помощью Djangoтестовый клиент по умолчанию.

Answer 3

Вот пример того, как я мог бы высмеять это S3Uploader в APITestCase.

from rest_framework import status
from unittest import mock
from unittest.mock import MagicMock

class SomeViewTestCase(APITestCase): 

   @mock.patch("path.to.view_file.S3Uploader")
   def test_upload(self, s3_uploader_mock):
       """Test with mocked S3Uploader"""
       concrete_uploader_mock = MagicMock(**{
           "upload_with_aux_fxn__return_value": {"status_code": 200}
       })
       s3_uploader_mock.return_value = concrete_uploader_mock
       response = self.client.post(url, payload, format='multipart')
       self.assertEqual(response.status_code, status.HTTP_200_OK)
       s3_uploader_mock.assert_called_once()
       concrete_uploader_mock.upload_with_aux_fx.assert_called_once()

Answer 4

Попробуйте использовать MagicMock, как показано ниже

from unittest import mock
from storages.backends.s3boto3 import S3Boto3Storage


class SomeTestCase(TestCase):
  def setUp(self):
    self.factory = APIRequestFactory()
    self.view = ViewToTest.as_view()
    self.url = reverse('some_url')


  @mock.patch.object(S3Boto3Storage, '_save', MagicMock(return_value='/tmp/somefile.png'))
  def test_uplaod(self):
    payload = {'some': 'data', 'other': 'stuff'}
    request = self.factory.post(self.url, payload, format='json')
    force_authenticate(request, user=self.user)
    response = self.view(request)