如何在 django 中模拟用户和请求
问题描述
我有与请求对象或用户对象交互的 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)
相关文章