如何为 django 多租户应用程序编写测试？

Question

我在 django 中有带有 postgresql 模式的多租户应用程序。我试图在租户文件夹“t002”中编写测试代码：

from django.test import TestCase
from path.customers.models import Client
# TestClass to test tenant setup
class TestTenantSetup(TestCase):
    def setUp(self):
        Client.objects.create(
            name='t002',
            schema_name='t002', 
            domain_url='t002.cmpny.com'
            )

    def test_tenant_exists(self):
        client = Client.objects.get(name='t002')
        self.assertEquals(client.schema_name, "t002")

但是当我使用 python manage.py test path/t002/tests 运行测试时，它正在为我的实际数据库创建一个测试数据库。所以假设我会遵循这个 SO 答案 (django unit tests without a db) 并避免创建数据库，但我不确定我是否遵循正确的路径来测试多租户 django 项目。当我查看 django-tenant-schemas 文档 (https://django-tenant-schemas.readthedocs.io/en/latest/test.html) 时，我无法轻易掌握它。谁能告诉如何开始对 django 多租户应用程序进行测试？并且请检查上面的代码是否正确。

规格 - python==2.7、django==1.11、postgres==9.6

Answer 1

取决于您在多租户应用程序中使用的隔离软。

如果您使用不同的数据库设置或任何其他设置，您可以使用 @django.test.override_settings 在您的测试中覆盖它们的测试版本。

例如：

from django.test import override_settings

@override_settings(DATABASES=<...>)  # you can also load and/or override other settings for a specific application
class TestTenantSetup(TestCase):
    def setUp(self):
        Client.objects.create(
            name='t002',
            schema_name='t002', 
            domain_url='t002.cmpny.com'
            )

    def test_tenant_exists(self):
        client = Client.objects.get(name='t002')
        self.assertEquals(client.schema_name, "t002")