模拟，预期str实例MagicMock找到

Question

我试图使用Mock with Django做两件事: 1.获取用户(我们有一个自定义后端来进行我想要避免的网络调用)。2.模拟另一种方法，该方法通过网络获得密码重置的令牌。

下面是一个测试：

@patch('core.views.KagisoUser')
def test_forgot_password_sends_reset_email(self, MockKagisoUser):  # noqa
    email = 'mock@user.com'
    user = MagicMock()
    user.email = email
    user.generate_reset_password_token.return_value = email
    MockKagisoUser.objects.first.return_value = user
    data = {'email': email}
    client = Client()

    response = client.post('/forgot_password/', data, follow=True)
    message = list(response.context['messages'])[0].message

    assert response.status_code == 200
    assert 'You will receive an email with reset instructions shortly' \
        == message

    assert len(mail.outbox) == 1
    assert mail.outbox[0].to[0] == email
    assert mail.outbox[0].subject == 'Password Reset'
    assert mail.outbox[0].alternatives[0]  # HTML body

我知道这个错误：

self = <django.core.mail.message.EmailMultiAlternatives object at 0x7f52eacd3390>

        def message(self):
            encoding = self.encoding or settings.DEFAULT_CHARSET
            msg = SafeMIMEText(self.body, self.content_subtype, encoding)
            msg = self._create_message(msg)
            msg['Subject'] = self.subject
            msg['From'] = self.extra_headers.get('From', self.from_email)
    >       msg['To'] = self.extra_headers.get('To', ', '.join(self.to))
    E       TypeError: sequence item 0: expected str instance, MagicMock found

以下是实现：

def forgot_password(request):
    reset_message = 'You will receive an email with reset instructions shortly'
    not_found_message = 'We could not find a user for that email address'

    if request.method == 'POST':
        form = forms.ForgotPasswordForm(request.POST)

        if form.is_valid():
            email = form.cleaned_data['email']
            user = KagisoUser.objects.filter(email=email).first()

            if user:
                print(user.email)
                token = user.generate_reset_password_token()
                url = request.build_absolute_uri(reverse('reset_password'))
                message = render_to_string(
                    'core/emails/reset_password.html',
                    {
                        'url': url,
                        'token': token,
                        'user_id': user.id
                    }
                )
                send_mail(
                    'Password Reset',
                    '',
                    'development@kagiso.io',
                    [user.email],
                    html_message=message
                )
                messages.success(request, reset_message)
                return HttpResponseRedirect('/')
            else:
                messages.error(request, not_found_message)
                return HttpResponseRedirect(reverse('forgot_password'))
    else:
        form = forms.ForgotPasswordForm()

    return render(
        request,
        'core/auth/forgot_password.html',
        {'form': form}
    )

我做错了什么？

Answer 1

堆栈跟踪非常有用..。

但除此之外，您似乎是在嘲弄MockKagisoUser.objects.first，而您的视图从KagisoUser.objects.filter(email=email).first()中获取用户，因此您缺少了filter

mock_filter = MagicMock()
mock_filter.first.return_value = user
MockKagisoUser.objects.filter.return_value = mock_filter

上面的过滤器应该返回一个筛选器，在获取first时返回用户，然后返回视图中的用户，而不是返回MagicMock实例。