Django测试:伪造用户创建

时间:2012-09-28 03:34:02

标签: python django testing

我想更好地编写这个测试:

def test_profile_created(self):
        self.client.post(reverse('registration_register'), data={
            'username':'ygam',
            'email':'ygam@example.com',
            'password1':'ygam',
            'password2':'ygam'
        })
        """
        Test if a profile is created on save
        """
        user = User.objects.get(username='ygam')
        self.assertTrue(UserProfile.objects.filter(user=user).exists())

我刚刚在django-registration测试中发现了这个代码,它实际上没有“创建”用户:

def test_registration_signal(self):
        def receiver(sender, **kwargs):
            self.failUnless('user' in kwargs)
            self.assertEqual(kwargs['user'].username, 'bob')
            self.failUnless('request' in kwargs)
            self.failUnless(isinstance(kwargs['request'], WSGIRequest))
            received_signals.append(kwargs.get('signal'))

        received_signals = []
        signals.user_registered.connect(receiver, sender=self.backend.__class__)

        self.backend.register(_mock_request(),
                              username='bob',
                              email='bob@example.com',
                              password1='secret')

        self.assertEqual(len(received_signals), 1)
        self.assertEqual(received_signals, [signals.user_registered])

但是他为这个“_mock_request”使用了自定义函数:

class _MockRequestClient(Client):
    def request(self, **request):
        environ = {
            'HTTP_COOKIE': self.cookies,
            'PATH_INFO': '/',
            'QUERY_STRING': '',
            'REMOTE_ADDR': '127.0.0.1',
            'REQUEST_METHOD': 'GET',
            'SCRIPT_NAME': '',
            'SERVER_NAME': 'testserver',
            'SERVER_PORT': '80',
            'SERVER_PROTOCOL': 'HTTP/1.1',
            'wsgi.version': (1,0),
            'wsgi.url_scheme': 'http',
            'wsgi.errors': self.errors,
            'wsgi.multiprocess':True,
            'wsgi.multithread': False,
            'wsgi.run_once': False,
            'wsgi.input': None,
            }
        environ.update(self.defaults)
        environ.update(request)
        request = WSGIRequest(environ)

        # We have to manually add a session since we'll be bypassing
        # the middleware chain.
        session_middleware = SessionMiddleware()
        session_middleware.process_request(request)
        return request


def _mock_request():
    return _MockRequestClient().request()

但是,根据我的需要,它可能太长了。我希望能够以某种方式“伪造”帐户创建。我对模拟和存根的经验不多,所以任何帮助都可以。谢谢!

2 个答案:

答案 0 :(得分:2)

这两种方法正在测试不同的东西:

  1. 您的测试用例测试模型层 - 它会询问“如果我创建用户并保存它,是否自动创建配置文件记录?”

    它不依赖于用户对象的创建方式 - 通过管理视图,用户注册视图或管理命令 - 只要此测试通过,您就知道创建用户的任何方法通过ORM将添加个人资料。

    它也不关心如何创建配置文件。它可以通过预保存信号,保存后信号,重写的save()方法中的一些代码或魔法。只要创建了记录,测试就会通过。

  2. 您在django-registration中找到的代码正在测试视图层。它调用一个特定的视图函数register,向它传递三个URL参数,并询问问题“这个视图导致user_registered信号被触发了吗?”

    此测试用例不会考虑其他创建用户的方法;只是注册视图。它不关心创建用户的任何其他方法。

    它也不会查看是否创建了个人资料。所有它想知道的是,无论在任何特定应用程序中 的信号是什么,信号都被触发了。

  3. 有什么区别?

    您的代码是集成测试的一个很好的例子 - “我是否按顺序将所有这些部分(模型,信号)放在一起,以便创建配置文件?” - 以及回归测试,以便在将来向您发出是否已做过任何破坏用户个人资料创建的事情的信号。

    django注册码是一个实际的单元测试。它非常谨慎地隔离视图代码,在受控环境中调用它并提出一个简单的问题。

    哪个更好?

    在您的情况下,您可能不需要编写单元测试。 (虽然可能会有很多争论;在某些方面它实际上是宗教问题)你想要一种方法来验证你所放在一起的代码是按照你期望的那样工作,并且你的测试是最好的方法。

    不要担心实际创建用户 - 在测试运行期间,您正在执行事务中的所有操作,并且在测试方法完成后立即回滚该事务。它可能甚至没有击中磁盘。并且,正如@Platinum Azure所提到的,无论如何它都在测试数据库中。

答案 1 :(得分:0)

为什么要模拟用户创建?如果你运行Django的测试引擎,你正在测试测试数据库。只需调用视图或使用请求客户端工厂,让测试运行器在完成后删除测试数据库。