Mock()函数在django2中给出TypeError

时间:2018-06-20 19:59:14

标签: python django mocking tdd django-2.0

我正在关注this tutorial

运行test_views.py时,笔者发现错误不应该出现:TypeError: quote_from_bytes() expected bytes

我的views和我的test_views与本书相同,但是我使用的是django 2.0.6而不是django 1.11,所以我的url.py发生了变化,所以也许是问题所在。

修改:

第二个问题似乎出在mock()函数中。

当我使用patch('lists.views.List')时,视图中的Print(list_)会给出<MagicMock name='List()' id='79765800'>而不是List object (1)

/编辑

我的lists/urls.py

urlpatterns = [
    path('new', views.new_list, name='new_list'),
    path('<slug:list_id>/',
        views.view_list, name='view_list'),
    path('users/<email>/',         # I'm not sure about this one but it works in other tests
        views.my_lists, name='my_lists'),
]
#instead of:
#urlpatterns = [
#    url(r'^new$', views.new_list, name='new_list'),
#    url(r'^(\d+)/$', views.view_list, name='view_list'),
#    url(r'^users/(.+)/$', views.my_lists, name='my_lists'),
#]

我的lists/views.py

[...]
def new_list(request):
    form = ItemForm(data=request.POST)
    if form.is_valid():
        list_ = List()
        list_.owner = request.user
        list_.save()
        form.save(for_list=list_)
        Print(list_)
        return redirect(list_)
    else:
        return render(request, 'home.html', {"form": form})

我的lists/tests/test_views.py

@patch('lists.views.List')
@patch('lists.views.ItemForm')
def test_list_owner_is_saved_if_user_is_authenticated(self, 
    mockItemFormClass, mockListClass
):
    user = User.objects.create(email='a@b.com')
    self.client.force_login(user)
    self.client.post('/lists/new', data={'text': 'new item'})
    mock_list = mockListClass.return_value
    self.assertEqual(mock_list.owner, user)

我的完整追溯:

traceback

可以是什么?

谢谢

2 个答案:

答案 0 :(得分:1)

最后我找到了在线解决方案。

Django 2 doesn't support anymore bytestrings在某些地方,因此,当视图重定向模拟类列表时,它将作为模拟对象执行,而iri_to_uri django函数将引发错误。在django 1.11中,iri_to_uri将iri强制为字节return quote(force_bytes(iri), safe="/#%[]=:;$&()+,!?*@'~"),而现在是return quote(iri, safe="/#%[]=:;$&()+,!?*@'~")。因此,解决方案是使用return redirect(str(list_.get_absolute_url()))

中的return redirect(list_)而不是lists.views.py
def new_list(request):
    form = ItemForm(data=request.POST)
    if form.is_valid():
        list_ = List()
        list_.owner = request.user
        list_.save()
        form.save(for_list=list_)
        #return redirect(list_)
        return redirect(str(list_.get_absolute_url()))
    else:
        return render(request, 'home.html', {"form": form})

我希望这对其他人有帮助

答案 1 :(得分:1)

我已经在测试代码中解决了这个问题,而没有更改所需的生产代码,如下所示:

@patch('lists.views.NewListForm')
class NewListViewUnitTest(unittest.TestCase):
    def setUp(self):
        self.request = HttpRequest()
        self.request.POST['text'] = 'new list item'
        self.request.user = Mock()

def test_passes_POST_data_to_NewListForm(self, mockNewListForm):
    mock_form = mockNewListForm.return_value
    returned_object = mock_form.save.return_value
    returned_object.get_absolute_url.return_value = 'fakeurl'

    new_list2(self.request)

    mockNewListForm.assert_called_once_with(data=self.request.POST)

def test_saves_form_with_owner_if_form_valid(self, mockNewListForm):
    mock_form = mockNewListForm.return_value
    mock_form.is_valid.return_value = True
    returned_object = mock_form.save.return_value
    returned_object.get_absolute_url.return_value = 'fakeurl'

    new_list2(self.request)

    mock_form.save.assert_called_once_with(owner=self.request.user)

@patch('lists.views.redirect')
def test_redirects_to_form_returned_object_if_form_valid(
    self, mock_redirect, mockNewListForm
):
    mock_form = mockNewListForm.return_value
    mock_form.is_valid.return_value = True

    response = new_list2(self.request)

    self.assertEqual(response, mock_redirect.return_value)
    mock_redirect.assert_called_once_with(mock_form.save.return_value)

请注意,赋值 some_method.return_value 设置了 some_method 的响应,不调用 some_method(),因此我们也可以测试该方法是否只调用了一次。< /p>

我喜欢这个解决方案的地方在于它产生了所需的生产代码:

def new_list2(request):
    form = NewListForm(data=request.POST)
    list_ = form.save(owner=request.user)
    return redirect(list_)

.. 而不是在生产代码中使用像 return redirect(str(list_.get_absolute_url())) 这样的解决方法,这是不可取的,因为它:

  1. 不是所需的生产代码
  2. 不太优雅的生产代码
  3. 只是将模拟对象的名称作为字符串返回(即 <MagicMock name='NewListForm().save().get_absolute_url()' id='4363470544'>),这不是我们想要的:我们想要调用 get_absolute_url() 方法并且那个方法(不是 str())应该以字符串形式返回一个 url。