从django 1.5升级到1.7后的测试中的异常

时间:2015-02-16 03:38:46

标签: python django exception upgrade

这是破坏的测试:

def test_registering_user_twice_cause_error_msg(self):

    user=User(name='pyRock',email='python@rocks.com')
    user.save()

    #create the request used to test the view
    self.request.session = {}
    self.request.method = 'POST'
    self.request.POST = {
        'email':'python@rocks.com',
        'name':'pyRock',
        'stripe_token':'...',
        'last_4_digits':'4242',
        'password':'bad_password',
        'ver_password':'bad_password'
    }

    expected_form=UserForm(self.request.POST)
    expected_form.is_valid()
    expected_form.addError('python@rocks.com is already a member')

    #create the expected html
    html = render_to_response(
        'register.html',
        {
            'form': expected_form,
            'months': range(1, 12),
            'publishable': settings.STRIPE_PUBLISHABLE,
            'soon': soon(),
            'user': None,
            'years': list(range(2011, 2036)),
        }
    )

    #mock out stripe so we don't hit their server
    with mock.patch('stripe.Customer') as stripe_mock:

        config = {'create.return_value': mock.Mock()}
        stripe_mock.configure_mock(**config)

        #run the test
        resp = register(self.request)

        #verify that we did things correctly

        self.assertEqual(resp.status_code, 200)
        self.assertEqual(self.request.session, {})
        self.assertEqual(resp.content, html.content)

        #assert there is no records in the database.
        users = User.objects.filter(email="python@rocks.com")
        self.assertEqual(len(users), 1)

因此,在运行测试时会抛出以下内容:

ERROR: test_registering_user_twice_cause_error_msg (payments.tests.RegisterPageTests)
----------------------------------------------------------------------
Traceback (most recent call last):
  File "/Users/1111/_realpython/d3/django_ecommerce/payments/tests.py", line 278, in test_registering_user_twice_cause_error_msg
    self.assertEqual(len(users), 1)
  File "/Users/1111/.virtualenvs/d317/lib/python2.7/site-packages/django/db/models/query.py", line 122, in __len__
    self._fetch_all()
  File "/Users/1111/.virtualenvs/d317/lib/python2.7/site-packages/django/db/models/query.py", line 966, in _fetch_all
    self._result_cache = list(self.iterator())
  File "/Users/1111/.virtualenvs/d317/lib/python2.7/site-packages/django/db/models/query.py", line 265, in iterator
    for row in compiler.results_iter():
  File "/Users/1111/.virtualenvs/d317/lib/python2.7/site-packages/django/db/models/sql/compiler.py", line 700, in results_iter
    for rows in self.execute_sql(MULTI):
  File "/Users/1111/.virtualenvs/d317/lib/python2.7/site-packages/django/db/models/sql/compiler.py", line 786, in execute_sql
    cursor.execute(sql, params)
  File "/Users/1111/.virtualenvs/d317/lib/python2.7/site-packages/django/db/backends/utils.py", line 59, in execute
    self.db.validate_no_broken_transaction()
  File "/Users/1111/.virtualenvs/d317/lib/python2.7/site-packages/django/db/backends/__init__.py", line 386, in validate_no_broken_transaction
    "An error occurred in the current transaction. You can't "
TransactionManagementError: An error occurred in the current transaction. You can't execute queries until the end of the 'atomic' block.

我读了几个关于Django在事务中包装每个测试的答案,但我不能做任何事情因为我是Django noob,上面的代码就是教程。

在断言中发生异常更有趣。如果我将其删除,则即使查询users = User.objects.filter(email="python@rocks.com")存在,测试也会通过。但是,如果我以某种方式引用users,请说print users测试再次中断。

那是怎么回事?

谢谢!

1 个答案:

答案 0 :(得分:0)

  

那是怎么回事?

如上所述,this is a duplicate

但如果您访问users,我会单独说明它中断的原因。发生这种情况的原因是Django数据库调用是懒惰的并且延迟执行,直到你最终使用查询。

因此,如果您从不使用或遍历users,它将为您节省数据库调用。这就是为什么它看起来有效,是因为如果你删除len(users)它永远不需要users的结果,那么就不会查询它。