0

我正在尝试提供对通过 boto3 生成的 s3 预签名 url 的重定向，它看起来应该可以工作，但我想为它编写一个测试用例。

我已经检查过类似于下面的内容。test_redirect_to_presigned_url()失败并没有test_bucket/testfile.png找到。虽然test_motos3_presigned_url作品如预期。

from django.test import TestCase
from django.test import Client


class MotoS3TestCase(TestCase):
    def setUp(self):
        self.mock = mock_s3()
        self.mock.start()
        self.bucket_name = 'test_bucket'
        self.conn = boto3.resource('s3')
        self.conn.create_bucket(Bucket=self.bucket_name)
        self.s3 = boto3.client('s3')

    def tearDown(self):
        self.mock.stop()

    def test_redirect_to_presigned_url(self):
        client = Client()
        response = client.get('/url/to/view/', follow=True)   
        self.assertTrue(response.status_code == 200, f'response: {response}, response.content: {response.content}')         

    def test_motos3_presigned_url(self):
        keyname = 'testfile.png'  
        self.s3.upload_fileobj(io.BytesIO(b'somefile content'), self.bucket_name, keyname)
        params = {
            'Bucket': self.bucket_name,
            'Key': keyname,
        }

        presigned_url = self.s3.generate_presigned_url(
            ClientMethod='get_object',
            Params=params,
            ExpiresIn=3600,
            HttpMethod='GET'
        )
        import requests
        response = requests.get(presigned_url)

        self.assertTrue(response.status_code == 200, f'response: {response}, response.content: {response.content}')

视图.py

import boto3
from django.http import HttpResponseRedirect


def redirect_to_presigned_url(request):
    s3 = boto3.client('s3')
    keyname = 'testfile.png'  
    bucket_name = 'test_bucket'
    s3.upload_fileobj(io.BytesIO(b'somefile content'), bucket_name, keyname)
    params = {
                'Bucket': bucket_name,
                'Key': keyname,
    }

    presigned_url = s3.generate_presigned_url(
                ClientMethod='get_object',
                Params=params,
                ExpiresIn=3600,
                HttpMethod='GET'
    )
    return HttpResponseRedirect(redirect_to=presigned_url)

看来 moto 不能很好地与 django 测试客户端集成。在重定向尝试时，moto 似乎返回 a404不存在。

因此，即使我添加了对象，我也无法使用生成的presigned_url.

为什么我可以创建一个测试用例来检查这个重定向？

4

0 回答 0