added tests for Interval columns in migrations

Author: dantownsend

piccolo/columns/base.py

@@ -28,6 +28,7 @@
 from piccolo.columns.combination import Where
 from piccolo.columns.choices import Choice
 from piccolo.columns.defaults.base import Default
+from piccolo.columns.defaults.interval import IntervalCustom
 from piccolo.columns.reference import LazyTableReference
 from piccolo.columns.indexes import IndexMethod
 from piccolo.querystring import QueryString
@@ -545,6 +546,9 @@ def get_sql_value(self, value: t.Any) -> t.Any:
             output = f"'{value.isoformat()}'"
         elif isinstance(value, datetime.time):
             output = f"'{value.isoformat()}'"
+        elif isinstance(value, datetime.timedelta):
+            interval = IntervalCustom.from_timedelta(value)
+            output = getattr(interval, self._meta.engine_type)
         elif isinstance(value, bytes):
             output = f"'{value.hex()}'"
         elif isinstance(value, uuid.UUID):

tests/apps/migrations/auto/integration/test_migrations.py

@@ -14,6 +14,7 @@
     BigInt,
     Date,
     Integer,
+    Interval,
     SmallInt,
     Text,
     Time,
@@ -59,6 +60,10 @@ def date_default():
     return datetime.datetime.now().date()
 
 
+def timedelta_default():
+    return datetime.timedelta(days=1)
+
+
 @postgres_only
 class TestMigrations(TestCase):
     def tearDown(self):
@@ -267,3 +272,19 @@ def test_date_column(self):
                 ]
             ]
         )
+
+    def test_interval_column(self):
+        self._test_migrations(
+            table_classes=[
+                self.table(column)
+                for column in [
+                    Interval(),
+                    Interval(default=datetime.timedelta(days=1)),
+                    Interval(default=timedelta_default),
+                    Interval(null=True, default=None),
+                    Interval(null=False),
+                    Interval(index=True),
+                    Interval(index=False),
+                ]
+            ]
+        )