lib/luci/utils_unittest.py

# Copyright 2019 The ChromiumOS Authors
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.

"""Tests for utils.py."""

from datetime import date
from datetime import datetime

from chromite.third_party.google.protobuf.timestamp_pb2 import Timestamp

from chromite.lib import cros_test_lib
from chromite.lib.luci import utils


class TestTimeParsingFunctions(cros_test_lib.MockTestCase):
    """Test time parsing functions in luci/utils.py."""

    def testTimestampToDatetime(self) -> None:
        # Test None input.
        self.assertIsNone(utils.TimestampToDatetime(None))
        # Test empty input.
        time1 = Timestamp()
        self.assertIsNone(utils.TimestampToDatetime(time1))
        # Test valid input.
        time1.GetCurrentTime()
        formatted_time = utils.TimestampToDatetime(time1)
        self.assertIsNotNone(formatted_time)
        self.assertIsInstance(formatted_time, datetime)

    def testDateToTimestamp(self) -> None:
        result = utils.DatetimeToTimestamp(date(2019, 4, 15))
        self.assertEqual(result.seconds, 1555286400)
        result = utils.DatetimeToTimestamp(date(2019, 4, 15), end_of_day=True)
        self.assertEqual(result.seconds, 1555372740)
        with self.assertRaises(AssertionError):
            utils.DatetimeToTimestamp(None)