from datetime import datetime as dt, timedelta as td
from django.test import TestCase

from hc.lib.date import format_hms, choose_next_report_date


class DateFormattingTestCase(TestCase):
    def test_mins_secs_work(self):
        s = format_hms(td(seconds=0))
        self.assertEqual(s, "0 sec")

        s = format_hms(td(seconds=1))
        self.assertEqual(s, "1 sec")

        s = format_hms(td(seconds=61))
        self.assertEqual(s, "1 min 1 sec")

        s = format_hms(td(seconds=62))
        self.assertEqual(s, "1 min 2 sec")

    def test_hours_work(self):
        s = format_hms(td(seconds=62 + 60 * 60))
        self.assertEqual(s, "1 h 1 min 2 sec")

        s = format_hms(td(seconds=60 * 60))
        self.assertEqual(s, "1 h 0 min 0 sec")


class NextReportDateTestCase(TestCase):
    def test_it_works(self):
        # October
        nao = dt(year=2019, month=10, day=15, hour=6)
        result = choose_next_report_date(nao)
        self.assertEqual(result.year, 2019)
        self.assertEqual(result.month, 11)
        self.assertEqual(result.day, 1)
        self.assertTrue(result.hour >= 12)

        # December
        nao = dt(year=2019, month=12, day=15, hour=6)
        result = choose_next_report_date(nao)
        self.assertEqual(result.year, 2020)
        self.assertEqual(result.month, 1)
        self.assertEqual(result.day, 1)
        self.assertTrue(result.hour >= 12)
