python - Django - 使用站点框架进行单元测试

Question

我尝试使用站点框架运行单元测试。 这是测试函数:

def test_main(self):
  response = self.client.get('/')
  self.assertEqual(response.status_code, 200)

这是命令:

python2.7 manage.py test --settings=myproject.settings_01

但是我遇到了一个错误，因为测试函数不知道我想要测试哪个站点。 --settings 不起作用。

我该怎么说我想要测试SITE_ID=1

错误:

======================================================================
enter code here`ERROR: test_main (sport.tests.SimpleTest)
----------------------------------------------------------------------
Traceback (most recent call last):
  File "/usr/lib/python2.7/site-packages/django/test/utils.py", line 216, in inner
    return test_func(*args, **kwargs)
  File "/home/torm/BookingSystem/sport/tests.py", line 12, in test_main
    response = self.client.get('/')
  File "/usr/lib/python2.7/site-packages/django/test/client.py", line 470, in get
    **extra)
  File "/usr/lib/python2.7/site-packages/django/test/client.py", line 286, in get
    return self.generic('GET', path, secure=secure, **r)
  File "/usr/lib/python2.7/site-packages/django/test/client.py", line 358, in generic
    return self.request(**r)
  File "/usr/lib/python2.7/site-packages/django/test/client.py", line 440, in request
    six.reraise(*exc_info)
  File "/usr/lib/python2.7/site-packages/django/core/handlers/base.py", line 130, in get_response
    % (callback.__module__, view_name))
ValueError: The view system.views.main_page didn't return an HttpResponse object. It returned None instead

主要功能:

def main_page(request):
    site = get_current_site(request)

    if site.name == 'sport':
        return sport.views.main_page(request)

Answer 1

我已经找到解决办法了。

我没有更改测试数据库中的站点名称。

def setUp(self):
    site = Site.objects.get(id=1)
    site.name = 'sport'
    site.save()

当我将此代码添加到测试类时，一切正常。