Более элегантный подход к написанию модульных тестов Django

В настоящее время я пишу тесты, используя модульные тесты Django (на основе модуля стандартной библиотеки Python: unittest). Я написал этот тест для своей модели контакта, который проходит:

class ContactTestCase(TestCase):
    def setUp(self):
        """Create model objects."""
        Contact.objects.create(
            name='Jane Doe',
            email='[email protected]',
            phone='+2348123940567',
            subject='Sample Subject',
            message='This is my test message for Contact object.'
        )


    def test_user_can_compose_message(self):
        """ Test whether a user can compose a messsage in the contact form."""
        test_user = Contact.objects.get(name='Jane Doe')
        self.assertEqual(test_user.email, '[email protected]')
        self.assertEqual(test_user.phone, '+2348123940567')
        self.assertEqual(test_user.subject, 'Sample Subject')
        self.assertEqual(test_user.message, 'This is my test message for Contact object.')
Found 1 test(s).
Creating test database for alias 'default'...
System check identified no issues (0 silenced).
.
----------------------------------------------------------------------
Ran 1 test in 0.005s

OK
Destroying test database for alias 'default'...

Однако в этом тесте мне пришлось использовать метод assertEqual 4 раза (может быть и больше, если тестировать модели с большим количеством полей). Похоже, это не соответствует принципу DRY.

Из документации я знаю, что assertEqual(first, second, msg=None) тесты, в которых первое и второе равны. Если значения не сравниваются равными, тест завершится неудачно.

Есть ли обходной путь или более элегантный подход к написанию таких тестов?

🤔 А знаете ли вы, что...
Python обладает обширной документацией и активным сообществом разработчиков.


51
1

Ответ:

Решено

Идеальный вопрос! Добро пожаловать в сообщество тестов качества!

  1. Создайте список/диктант с тестовыми примерами.
  2. Просмотрите список и вызовите Assert в контексте подтеста.
  3. Наслаждаться.

Пример здесь:

def test_user_can_compose_message(self):
    """ Test whether a user can compose a messsage in the contact form."""
    test_user = Contact.objects.get(name='Jane Doe')
    test_cases = {'email': '[email protected]', 'phone': '+2348123940567', 'subject': 'Sample Subject', 'message': 'This is my test message for Contact object.'}
    for field, value in test_cases.items():
        with self.subTest("wrong user subtest", field=field, ):
            self.assertEqual(getattr(test_user, field), value)

Кстати, не забудьте использовать Factory Boy и старайтесь избегать жестко запрограммированных констант. Вместо этого вы можете использовать django.utils.crypto.render_string и т. д.