KeyError: 'HTTP_HOST' when running django tests

Question

我是单元测试的新手，所以我不知道我做错了什么。我使用 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/ 时一切正常，所以我想错误出在测试中。如果您需要任何其他代码，我会在此处粘贴。

Answer 1

HTTP_HOST header默认情况下Django测试客户端没有设置。您的多主机中间件假定 header 始终存在，因此当测试运行.

时您会得到 KeyError

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

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')

或者您可以将测试客户端设置为在所有请求中包含 header：

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/')

有关详细信息，请参阅 making requests 上的文档。

KeyError: 'HTTP_HOST' when running django tests

KeyError: 'HTTP_HOST' when running django tests

python

django

django-testing

django-tests