8

我是单元测试的新手,所以我不知道我做错了什么。我使用 python2.7 和 Django1.8

当我跑

python manage.py test myapp --keepdb

我明白了

======================================================================
ERROR: test_view_content (myproject.news.tests.test_views.EntryTestCase)
----------------------------------------------------------------------
Traceback (most recent call last):
  File "/home/zoli/projects/project_dict/myproject/news/tests/test_views.py", line 27, in test_view_content
    response = client.get(reverse('news_list', kwargs={'page': 1}))
  File "/home/zoli/.virtualenvs/project_dict/local/lib/python2.7/site-packages/django/test/client.py", line 500, in get
    **extra)
  File "/home/zoli/.virtualenvs/project_dict/local/lib/python2.7/site-packages/django/test/client.py", line 303, in get
    return self.generic('GET', path, secure=secure, **r)
  File "/home/zoli/.virtualenvs/project_dict/local/lib/python2.7/site-packages/django/test/client.py", line 379, in generic
    return self.request(**r)
  File "/home/zoli/.virtualenvs/project_dict/local/lib/python2.7/site-packages/django/test/client.py", line 466, in request
    six.reraise(*exc_info)
  File "/home/zoli/.virtualenvs/project_dict/local/lib/python2.7/site-packages/django/core/handlers/base.py", line 108, in get_response
    response = middleware_method(request)
  File "/home/zoli/projects/project_dict/myproject/middleware/multihostname.py", line 18, in process_request
    host = request.META['HTTP_HOST'].split(':')[0]
KeyError: u'HTTP_HOST'

----------------------------------------------------------------------

我的测试看起来像

from django.test import TestCase, Client
from django.contrib.sites.models import Site
from myproject.news.models import Entry
from myproject.people.models import User
from django.core.urlresolvers import reverse


class EntryTestCase(TestCase):
    def setUp(self):
        user1 = User.objects.create(username='zoli')
        site1 = Site.objects.create(domain='mysite.sk', name='mysite')
        entry = Entry(author=user1, title='Titulok', text='Toto je obsah')
        entry.save()
        entry.sites.add(site1)
        entry.save()

    def test_view_content(self):
        client = Client()

        response = client.get(reverse('news_list', kwargs={'page': 1})) # This is raising and error
        print response.content

当我访问 /novinky/strana/1/ 时,一切正常,所以我认为错误在测试中。如果您需要任何其他代码,我会在此处粘贴。

4

1 回答 1

19

HTTP_HOST默认情况下,Django 测试客户端不设置标头。您的多主机中间件假定标头始终存在,因此您会KeyError在测试运行时得到一个。

您可能想要更改您的多主机名中间件,以便当标头不在请求中时它不会导致错误。

if 'HTTP_HOST' in request.META:
    host = request.META['HTTP_HOST'].split(':')[0]
    ...
else:
    # do something else

或者也许有一个默认主机:

host = request.META.get('HTTP_HOST', 'defaulthost.com').split(':')[0]

如果要测试不同header的效果,可以在发出请求时包含header:

client = Client()
# Make a request, setting the header manually
client.get('/my_url', HTTP_HOST='example.com')

或者您可以设置测试客户端以在所有请求中包含标头:

client = Client(HTTP_HOST='example.com')
# The header will be set for both of the following requests
client.get('/my_url/')
client.get('/my_second_url/')

请参阅有关提出请求的文档以获取更多信息。

于 2016-01-19T12:25:17.343 回答