Django 单元测试 function 带有基于模拟 return_value 的重定向

Question

我有一个类似于 function 的视图

def my_function(request):
    session = create_something('some_random_string')
    return redirect(session.url, code=303)

测试一下

import unittest
from django.test import TestCase
from unittest.mock import patch
from my_app.views import my_function

class TestMyFunction(TestCase):

    @patch('my_app.views.create_something', return_value={
        "url": "https://tiagoperes.eu/"
    })
    def test_my_function(self, mock_create_something):
        response = self.client.get("/my-function/")

这给

AttributeError: 'dict' object 没有属性 'url'

这个问题类似于以下问题

• 当模拟 Django model class 时无法重定向（因为我使用的是redirect() ，它采用来自模拟 function 的动态值，因此也可以轻松达到错误TypeError: quote_from_bytes() expected bytes ）
• 如何使用 unittest.mock 调试修补方法（如果没有return_value ，我会得到TypeError: expected string or bytes-like object ）

Answer 1

我必须替换补丁中的 return_value 才能使用MagicMock

@patch('my_app.views.create_something', return_value=MagicMock(url="https://tiagoperes.eu"))