Refactored tests to play well with older versions of Django.

Author: Michael Barr

.coveragerc

@@ -0,0 +1,2 @@
+[run]
+omit = timezone_utils/*migrations*

tests/models.py

@@ -0,0 +1,162 @@
+# ==============================================================================
+# IMPORTS
+# ==============================================================================
+# ==============================================================================
+# IMPORTS
+# ==============================================================================
+# Python
+from __future__ import unicode_literals
+from datetime import datetime, time as datetime_time
+import pytz
+
+# Django
+from django.conf import settings
+from django.db import models
+from django.utils.translation import ugettext_lazy as _
+
+# App
+from timezone_utils.fields import LinkedTZDateTimeField, TimeZoneField
+from timezone_utils.choices import PRETTY_ALL_TIMEZONES_CHOICES
+
+
+# ==============================================================================
+# MODELS
+# ==============================================================================
+class TZWithLowMaxLength(models.Model):
+    """Test should check to make sure that the `max_length` is set to the
+    maximum length of the longest value provided by pytz.all_timezones.
+
+    """
+    timezone = TimeZoneField(
+        max_length=15,
+        null=True,
+    )
+
+
+class TZWithBadStringDefault(models.Model):
+    """Test should check that a ValidationError is raised when..."""
+    timezone = TimeZoneField(
+        default='Bad/Worse',
+        max_length=64,
+        null=True,
+    )
+
+
+class ModelWithBadTimeOverride(models.Model):
+    timestamp = LinkedTZDateTimeField(
+        default=settings.TEST_DATETIME,
+        time_override='<invalid>',
+    )
+
+
+class ModelWithBadPopulateFrom(models.Model):
+    timezone = TimeZoneField(default='US/Eastern')
+    timestamp = LinkedTZDateTimeField(
+        default=settings.TEST_DATETIME,
+        populate_from='invalid_field_reference',
+    )
+
+
+class ModelWithBadTimeZoneCharField(models.Model):
+    timezone = models.CharField(default='Bad/Worse', max_length=64)
+    timestamp = LinkedTZDateTimeField(
+        default=settings.TEST_DATETIME,
+        populate_from='timezone',
+    )
+
+
+class LocationTimeZone(models.Model):
+    timezone = TimeZoneField(
+        verbose_name=_('timezone'),
+        max_length=64,
+        null=True,
+        blank=True
+    )
+
+
+class LocationTimeZoneChoices(models.Model):
+    timezone = TimeZoneField(
+        verbose_name=_('timezone'),
+        max_length=64,
+        null=True,
+        blank=True,
+        choices=PRETTY_ALL_TIMEZONES_CHOICES,
+    )
+
+
+class TZWithGoodStringDefault(models.Model):
+    """Test should validate that"""
+    timezone = TimeZoneField(
+        default='US/Eastern',
+        max_length=64,
+        null=True,
+    )
+
+
+class TZWithGoodTZInfoDefault(models.Model):
+    timezone = TimeZoneField(
+        default=pytz.timezone('US/Pacific'),
+        max_length=64,
+        null=True,
+    )
+
+
+class ModelWithDateTimeOnly(models.Model):
+    timestamp = LinkedTZDateTimeField(
+        default=settings.TEST_DATETIME,
+    )
+
+
+class NullModelWithDateTimeOnly(models.Model):
+    timestamp = LinkedTZDateTimeField(
+        null=True,
+    )
+
+
+class CallableTimeStampedModel(models.Model):
+    start = LinkedTZDateTimeField(
+        default=settings.TEST_DATETIME,
+        time_override=datetime.min.time
+    )
+    end = LinkedTZDateTimeField(
+        default=settings.TEST_DATETIME,
+        time_override=datetime.max.time
+    )
+
+
+class StaticTimeStampedModel(models.Model):
+    start = LinkedTZDateTimeField(
+        default=settings.TEST_DATETIME,
+        time_override=datetime_time(0, 0)
+    )
+    end = LinkedTZDateTimeField(
+        default=settings.TEST_DATETIME,
+        time_override=datetime_time(23, 59, 59, 999999)
+    )
+
+
+class ModelWithForeignKeyToTimeZone(models.Model):
+    other_model = models.ForeignKey(
+        to='tests.TZWithGoodStringDefault',
+        related_name='fk_to_tz'
+    )
+    timestamp = LinkedTZDateTimeField(
+        default=settings.TEST_DATETIME,
+        populate_from=lambda instance: instance.other_model.timezone
+    )
+
+
+class ModelWithLocalTimeZone(models.Model):
+    timezone = TimeZoneField(default='US/Eastern')
+    timestamp = LinkedTZDateTimeField(
+        default=settings.TEST_DATETIME,
+        populate_from='timezone'
+    )
+
+
+class ModelWithLocalTZCharField(models.Model):
+    timezone = models.CharField(default='US/Eastern', max_length=64)
+    timestamp = LinkedTZDateTimeField(
+        default=settings.TEST_DATETIME,
+        populate_from='timezone'
+    )

tests/models/__init__.py

@@ -9,7 +9,7 @@
 from django.test import TestCase
 
 # App
-from .models.timezonefield.invalid import (TZWithBadStringDefault,
+from tests.models import (TZWithBadStringDefault,
                                            TZWithLowMaxLength)