如何在 django 中模拟用户和请求

2022-01-08 00:00:00 python django mocking unit-testing

问题描述

我有与请求对象或用户对象交互的 django 代码.例如:

I have django code that interacts with request objects or user objects. For instance something like:

foo_model_instance = models.get_or_create_foo_from_user(request.user)

如果您要使用 django python shell 或在单元测试中进行测试,您会在其中传递什么?这里只需一个 User 对象就可以了,但对模拟请求对象的需求也经常出现.

If you were going to test with the django python shell or in a unittest, what would you pass in there? Here simply a User object will do, but the need for a mock request object also comes up frequently.

对于外壳或单元测试:

  • 如何模拟用户?
  • 如何模拟请求?

解决方案

对于请求,我会使用 RequestFactory 包含在 Django 中.

For request, I would use RequestFactory included with Django.

from django.test.client import RequestFactory
rf = RequestFactory()
get_request = rf.get('/hello/')
post_request = rf.post('/submit/', {'foo': 'bar'})

对于用户,我会按照@ozan 的建议使用 django.contrib.auth.models.User 并且可能使用 factory boy 提高速度(使用 factory boy 你可以选择不保存到 DB)

for users, I would use django.contrib.auth.models.User as @ozan suggested and maybe with factory boy for speed (with factory boy you can choose to not to save to DB)

相关文章