Django：如何在 API 视图中模拟一个类

Question

标题可能有点混乱。

假设我有一个带有post 方法的APIView。在 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 中）。我如何正确模拟这个？

编辑1：

我尝试修补

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

我想正确的方法是使用 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 2

看看vcrpy。它记录对外部 API 的请求一次，然后在每次运行测试时重放答案。无需手动模拟任何东西。

Answer 3

这是我如何在 APITestCase 中模拟 S3Uploader 的示例。

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)