django-model-utils/tests/test_models/test_timestamped_model.py

from __future__ import annotations

from datetime import datetime, timedelta, timezone

import time_machine
from django.test import TestCase

from tests.models import TimeStamp, TimeStampWithStatusModel


class TimeStampedModelTests(TestCase):
    def test_created(self) -> None:
        with time_machine.travel(datetime(2016, 1, 1, tzinfo=timezone.utc)):
            t1 = TimeStamp.objects.create()
        self.assertEqual(t1.created, datetime(2016, 1, 1, tzinfo=timezone.utc))

    def test_created_sets_modified(self) -> None:
        '''
        Ensure that on creation that modified is set exactly equal to created.
        '''
        t1 = TimeStamp.objects.create()
        self.assertEqual(t1.created, t1.modified)

    def test_modified(self) -> None:
        with time_machine.travel(datetime(2016, 1, 1, tzinfo=timezone.utc)):
            t1 = TimeStamp.objects.create()

        with time_machine.travel(datetime(2016, 1, 2, tzinfo=timezone.utc)):
            t1.save()

        self.assertEqual(t1.modified, datetime(2016, 1, 2, tzinfo=timezone.utc))

    def test_overriding_created_via_object_creation_also_uses_creation_date_for_modified(self) -> None:
        """
        Setting the created date when first creating an object
        should be permissible.
        """
        different_date = datetime.today() - timedelta(weeks=52)
        t1 = TimeStamp.objects.create(created=different_date)
        self.assertEqual(t1.created, different_date)
        self.assertEqual(t1.modified, different_date)

    def test_overriding_modified_via_object_creation(self) -> None:
        """
        Setting the modified date explicitly should be possible when
        first creating an object, but not thereafter.
        """
        different_date = datetime.today() - timedelta(weeks=52)
        t1 = TimeStamp.objects.create(modified=different_date)
        self.assertEqual(t1.modified, different_date)
        self.assertNotEqual(t1.created, different_date)

    def test_overriding_created_after_object_created(self) -> None:
        """
        The created date may be changed post-create
        """
        t1 = TimeStamp.objects.create()
        different_date = datetime.today() - timedelta(weeks=52)
        t1.created = different_date
        t1.save()
        self.assertEqual(t1.created, different_date)

    def test_overriding_modified_after_object_created(self) -> None:
        """
        The modified date should always be updated when the object
        is saved, regardless of attempts to change it.
        """
        t1 = TimeStamp.objects.create()
        different_date = datetime.today() - timedelta(weeks=52)
        t1.modified = different_date
        t1.save()
        self.assertNotEqual(t1.modified, different_date)

    def test_overrides_using_save(self) -> None:
        """
        The first time an object is saved, allow modification of both
        created and modified fields.
        After that, only created may be modified manually.
        """
        t1 = TimeStamp()
        different_date = datetime.today() - timedelta(weeks=52)
        t1.created = different_date
        t1.modified = different_date
        t1.save()
        self.assertEqual(t1.created, different_date)
        self.assertEqual(t1.modified, different_date)
        different_date2 = datetime.today() - timedelta(weeks=26)
        t1.created = different_date2
        t1.modified = different_date2
        t1.save()
        self.assertEqual(t1.created, different_date2)
        self.assertNotEqual(t1.modified, different_date2)
        self.assertNotEqual(t1.modified, different_date)

    def test_save_with_update_fields_overrides_modified_provided_within_a(self) -> None:
        """
        Tests if the save method updated modified field
        accordingly when update_fields is used as an argument
        and modified is provided
        """
        tests = (
            ['modified'],  # list
            ('modified',),  # tuple
            {'modified'},  # set
        )

        for update_fields in tests:
            with self.subTest(update_fields=update_fields):
                with time_machine.travel(datetime(2020, 1, 1, tzinfo=timezone.utc)):
                    t1 = TimeStamp.objects.create()

                with time_machine.travel(datetime(2020, 1, 2, tzinfo=timezone.utc)):
                    t1.save(update_fields=update_fields)
                self.assertEqual(t1.modified, datetime(2020, 1, 2, tzinfo=timezone.utc))

    def test_save_is_skipped_for_empty_update_fields_iterable(self) -> None:
        tests = (
            [],  # list
            (),  # tuple
            set(),  # set
        )

        for update_fields in tests:
            with self.subTest(update_fields=update_fields):
                with time_machine.travel(datetime(2020, 1, 1, tzinfo=timezone.utc)):
                    t1 = TimeStamp.objects.create()

                with time_machine.travel(datetime(2020, 1, 2, tzinfo=timezone.utc)):
                    t1.test_field = 1
                    t1.save(update_fields=update_fields)

                t1.refresh_from_db()
                self.assertEqual(t1.test_field, 0)
                self.assertEqual(t1.modified, datetime(2020, 1, 1, tzinfo=timezone.utc))

    def test_save_updates_modified_value_when_update_fields_explicitly_set_to_none(self) -> None:
        with time_machine.travel(datetime(2020, 1, 1, tzinfo=timezone.utc)):
            t1 = TimeStamp.objects.create()

        with time_machine.travel(datetime(2020, 1, 2, tzinfo=timezone.utc)):
            t1.save(update_fields=None)

        self.assertEqual(t1.modified, datetime(2020, 1, 2, tzinfo=timezone.utc))

    def test_model_inherit_timestampmodel_and_statusmodel(self) -> None:
        with time_machine.travel(datetime(2020, 1, 1, tzinfo=timezone.utc)):
            t1 = TimeStampWithStatusModel.objects.create()

        with time_machine.travel(datetime(2020, 1, 2, tzinfo=timezone.utc)):
            t1.save(update_fields=['test_field', 'status'])

        self.assertEqual(t1.modified, datetime(2020, 1, 2, tzinfo=timezone.utc))