Как протестировать auto_now_add в django

У меня есть приложение django 1.11, и я хочу написать модульный тест для своего решения.

Я хочу протестировать функцию даты регистрации.

model.py:

class User(models.Model):
    first_name = models.CharField(max_length=30)
    last_name = models.CharField(max_length=30)
    registration_date = models.DateTimeField(auto_now_add=True)

    def get_registration_date(self):
        return self.registration_date

Я также использую django-boy для фабрик моделей: factoryies.py

  class UserFactory(factory.DjangoModelFactory):
        class Meta:
            model = models.User
        first_name = 'This is first name'
        last_name = 'This is last name'
        registration_date = timezone.now()

test.py

def test_get_registration_date(self):
    user = factories.UserFactory.create()
    self.assertEqual(user.get_registration_date(), timezone.now())

Проблема в том, что я получил AssertionError:

AssertionError: datetime.datetime(2018, 4, 17, 9, 39, 36, 707927, tzinfo=<UTC>) != datetime.datetime(2018, 4, 17, 9, 39, 36, 708069, tzinfo=<UTC>)

почему бы не дать макету конкретное время при создании для упрощения

— 17.04.2018 11:57

python django unit-testing datetime factory-boy

17.04.2018 11:50

Почему в Python есть оператор "pass"?

Оператор pass в Python - это простая концепция, которую могут быстро освоить даже новички без опыта программирования.

Некоторые методы, о которых вы не знали, что они существуют в Python

Python - самый известный и самый простой в изучении язык в наши дни. Имея широкий спектр применения в области машинного обучения, Data Science,...

Основы Python Часть I

Вы когда-нибудь задумывались, почему в программах на Python вы видите приведенный ниже код?

LeetCode - 1579. Удаление максимального числа ребер для сохранения полной проходимости графа

Алиса и Боб имеют неориентированный граф из n узлов и трех типов ребер:

Оптимизация кода с помощью тернарного оператора Python

И последнее, что мы хотели бы показать вам, прежде чем двигаться дальше, это

Советы по эффективной веб-разработке с помощью Python

Как веб-разработчик, Python может стать мощным инструментом для создания эффективных и масштабируемых веб-приложений.

2 357

Перейти к ответу Данный вопрос помечен как решенный

Ответы 3

Ответ принят как подходящий

Вы можете использовать mock:

import pytz
from unittest import mock

def test_get_registration_date(self):
    mocked = datetime.datetime(2018, 4, 4, 0, 0, 0, tzinfo=pytz.utc)
    with mock.patch('django.utils.timezone.now', mock.Mock(return_value=mocked)):
        user = factories.UserFactory.create()
        self.assertEqual(user.get_registration_date(), mocked)

Спасибо! Также работает для полей auto_now.

— 27.08.2020 21:57

17.04.2018 11:59

Вы можете использовать пакет замораживающего пистолета. https://github.com/spulec/freezegun, который исправляет datetime.now ().

from freezegun import freeze_time
...
    @freeze_time("2017-06-23 07:28:00")
    def test_get_registration_date(self):
        user = factories.UserFactory.create()
        self.assertEqual(
            datetime.strftime(user.get_registration_date(), "%Y-%m-%d %H:%M:%S")
            "2017-06-23 07:28:00"
        )

17.04.2018 12:03

Просто используйте декоратор factory.post_generation:

class UserFactory(factory.DjangoModelFactory):
    ...
    @factory.post_generation
    def registration_date(self, create, extracted, **kwargs):
        if extracted:
            self.registration_date = extracted

22.01.2019 17:23

Другие вопросы по теме

Передача переменных Python в качестве аргументов скрипту Perl

Django возвращает варианты с помощью AjaxView

Django обслуживает статические файлы локально

Как отключить файлы проекта только для чтения в PyCharm?

Как реализовать отношения "один ко многим" в Django RestFramework?

Csv-файл не печатается на веб-странице django

Django: использовать несколько файлов CSS в одном HTML

Тег шаблона Django `url` не работает должным образом

Dockerfile_Django REST_ERROR _Не удается удалить "шесть". Это установленный дистрибутив

Chart.js с Django - Использование REST

Как протестировать auto_now_add в django

Ответы 3

Другие вопросы по теме

Похожие вопросы