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"))