如何测试Django模型中的related_name及字段verbose_name？

嘿，我来帮你把这个Django模型的测试补全！针对你定义的TelegramUser模型，我们可以分两部分来测试：字段的verbose_name，以及OneToOneField的related_name反向关联。下面是具体的实现步骤和完整代码示例：

1. 测试字段的verbose_name #

要验证每个字段的verbose_name是否符合预期，我们可以通过模型的_meta属性获取字段对象，然后直接断言其verbose_name属性。如果你的项目用了国际化（gettext_lazy），记得用str()转换后再对比，避免惰性翻译对象的问题。

示例测试函数：

from django.test import TestCase
from django.utils.translation import gettext_lazy as _
from yourapp.models import TelegramUser  # 替换成你的app名称

class TelegramUserModelTests(TestCase):
    def test_field_verbose_names(self):
        # 获取模型的字段对象
        user_field = TelegramUser._meta.get_field('user')
        token_field = TelegramUser._meta.get_field('token')
        chat_id_field = TelegramUser._meta.get_field('chat_id')
        
        # 断言每个字段的verbose_name
        self.assertEqual(str(user_field.verbose_name), str(_('User')))
        self.assertEqual(str(token_field.verbose_name), str(_('Token')))
        self.assertEqual(str(chat_id_field.verbose_name), str(_('Chat id')))

2. 测试related_name反向关联 #

OneToOneField的related_name用于从关联的用户模型（这里是settings.AUTH_USER_MODEL）反向访问TelegramUser实例。我们需要验证这个反向关联是否能正常工作：

def test_related_name(self):
        from django.contrib.auth import get_user_model
        User = get_user_model()
        
        # 创建测试用户和对应的TelegramUser实例
        test_user = User.objects.create_user(username='testuser', password='testpass123')
        telegram_user = TelegramUser.objects.create(
            user=test_user,
            token='test-token-123',
            chat_id='123456'
        )
        
        # 验证反向关联是否可用，且指向正确的实例
        self.assertEqual(test_user.telegramuser, telegram_user)
        # 额外验证：如果没有关联的TelegramUser，访问会抛出DoesNotExist异常
        another_user = User.objects.create_user(username='anotheruser', password='testpass456')
        with self.assertRaises(TelegramUser.DoesNotExist):
            _ = another_user.telegramuser

完整测试类 #

把上面两部分整合起来，就是一个完整的测试类，你可以直接放到你的app的tests.py文件里：

from django.test import TestCase
from django.utils.translation import gettext_lazy as _
from django.contrib.auth import get_user_model
from yourapp.models import TelegramUser  # 替换成实际的app名称

class TelegramUserModelTests(TestCase):
    def test_field_verbose_names(self):
        # 测试各字段的verbose_name
        user_field = TelegramUser._meta.get_field('user')
        token_field = TelegramUser._meta.get_field('token')
        chat_id_field = TelegramUser._meta.get_field('chat_id')
        
        self.assertEqual(str(user_field.verbose_name), str(_('User')))
        self.assertEqual(str(token_field.verbose_name), str(_('Token')))
        self.assertEqual(str(chat_id_field.verbose_name), str(_('Chat id')))

    def test_related_name(self):
        # 测试反向关联related_name
        User = get_user_model()
        
        # 创建关联实例
        test_user = User.objects.create_user(username='testuser', password='testpass123')
        telegram_user = TelegramUser.objects.create(
            user=test_user,
            token='test-token-123',
            chat_id='123456'
        )
        
        # 验证反向关联
        self.assertEqual(test_user.telegramuser, telegram_user)
        
        # 验证无关联时的异常
        another_user = User.objects.create_user(username='anotheruser', password='testpass456')
        with self.assertRaises(TelegramUser.DoesNotExist):
            _ = another_user.telegramuser

额外提示 #

• 如果你的项目没有启用国际化，或者直接用字符串而不是_()包裹verbose_name，可以去掉str()和gettext_lazy的引用，直接对比字符串。
• 运行测试的命令是：python manage.py test yourapp（替换成你的app名称）。

内容的提问来源于stack exchange，提问作者Narnik Gamarnik