Django 测试客户端和子域

使用Django的测试客户端和子域进行测试是一种非常方便的方法。测试客户端允许我们在不启动服务器的情况下进行测试，而子域测试则允许我们模拟不同的子域名进行测试。在本文中，我们将探讨如何使用这些功能进行测试，并通过一些案例代码来说明。

pip install django

django-admin startproject myproject
cd myproject
python manage.py startapp myapp

python
INSTALLED_APPS = [
    ...
    'myapp',
]

python
from django.test import Client
class SubdomainClient(Client):
    def __init__(self, subdomain=None, *args, **kwargs):
        super(SubdomainClient, self).__init__(*args, **kwargs)
        if subdomain:
            self.defaults['HTTP_HOST'] = subdomain + '.localhost'
def subdomain_client(subdomain=None, *args, **kwargs):
    return SubdomainClient(subdomain, *args, **kwargs)

python
MIDDLEWARE = [
    ...
    'myapp.middleware.SubdomainMiddleware',
]
SUBDOMAIN_MIDDLEWARE_URLCONF = 'myproject.urls'

python
from django.test import TestCase
from django.urls import reverse
class MyTestCase(TestCase):
    def test_homepage(self):
        client = subdomain_client()
        response = client.get(reverse('myapp:home'))
        self.assertEqual(response.status_code, 200)
        self.assertContains(response, "欢迎访问主页")
    def test_subdomain_page(self):
        client = subdomain_client(subdomain="subdomain")
        response = client.get(reverse('myapp:subdomain_page'))
        self.assertEqual(response.status_code, 200)
        self.assertContains(response, "欢迎访问子域页面")

python manage.py test myapp