test_time.py

import datetime
from functools import partial
from unittest import TestCase

from piccolo.table import Table
from piccolo.columns.column_types import Time
from piccolo.columns.defaults.time import TimeNow


class MyTable(Table):
    created_on = Time()


class MyTableDefault(Table):
    created_on = Time(default=TimeNow())


class TestTime(TestCase):
    def setUp(self):
        MyTable.create_table().run_sync()

    def tearDown(self):
        MyTable.alter().drop_table().run_sync()

    def test_timestamp(self):
        created_on = datetime.datetime.now().time()
        row = MyTable(created_on=created_on)
        row.save().run_sync()

        result = MyTable.objects().first().run_sync()
        self.assertEqual(result.created_on, created_on)


class TestTimeDefault(TestCase):
    def setUp(self):
        MyTableDefault.create_table().run_sync()

    def tearDown(self):
        MyTableDefault.alter().drop_table().run_sync()

    def test_timestamp(self):
        created_on = datetime.datetime.now().time()
        row = MyTableDefault()
        row.save().run_sync()

        _datetime = partial(datetime.datetime, year=2020, month=1, day=1)

        result = MyTableDefault.objects().first().run_sync()
        self.assertTrue(
            _datetime(
                hour=result.created_on.hour,
                minute=result.created_on.minute,
                second=result.created_on.second,
            )
            - _datetime(
                hour=created_on.hour,
                minute=created_on.minute,
                second=created_on.second,
            )
            < datetime.timedelta(seconds=1)
        )