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

Question

我有与请求对象或用户对象交互的 Django 代码。 例如类似的东西：

foo_model_instance = models.get_or_create_foo_from_user(request.user)

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

对于外壳或单元测试：

• 你如何模拟用户？
• 你如何模拟请求？

Answer 1

对于请求，我将使用 Django 中包含的RequestFactory 。

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 并且可能使用工厂男孩来提高速度（对于工厂男孩，您可以选择不保存到数据库）

Answer 2

你如何模拟用户？

初始化一个django.contrib.auth.models.User对象。 User.objects.create_user使这变得容易。

你如何模拟请求？

初始化一个django.http.HttpRequest对象。

当然，根据您想要做什么，有快捷方式。 如果您只需要一个带有指向user属性的对象，只需创建一些东西（任何东西）并赋予它该属性。

Answer 3

您可以按照 Anurag Uniyal 的建议推出自己的模拟，也可以使用模拟框架。

对于那些说你可以像在 Django 中一样创建一个普通用户的人的回应......我建议这违背了单元测试的意义。 单元测试不应该触及数据库，但是通过创建用户，您已经更改了数据库，因此我们为什么要模拟一个。

Answer 4

在这里阅读模拟对象
http://en.wikipedia.org/wiki/Mock_object
http://www.mockobjects.com/

并使用此 python 库来模拟用户
http://python-mock.sourceforge.net/

否则你可以自己写一个简单的 User 类，以此为起点

class MockUser(object):
    def __call__(self, *args, **kwargs):
        return self

    def __getattr__(Self, name):
        return self

添加特定案例等

Answer 5

您不需要模拟用户，因为您可以在测试中创建一个 - 测试完成后数据库将被销毁。

要模拟请求，请使用 Simon Willison 的这个片段。