scheduler_tests.py

from inspirator import subscriberservice
from inspirator.models import Subscriber, DeliveryPreference
from pytz import timezone
from datetime import datetime, timedelta
import unittest 
from freezegun import freeze_time

class TestDetermineNextSendDate(unittest.TestCase):
    tz = timezone('Europe/Zurich')
    now = datetime.now(tz).replace(hour=9)
    weekday_num = now.weekday()
    weekday_name = subscriberservice.weekdays[weekday_num]

    freezer = freeze_time(str(now))
   
    def setUp(self):
        self.freezer.start()

    def tearDown(self):
        self.freezer.stop()

    
    def test_next_send_today(self):
        dp = DeliveryPreference()
        setattr(dp, self.weekday_name, True)

        delivery_time = self.now + timedelta(hours=1)
        dp.hour = delivery_time.hour
        dp.minute = delivery_time.minute
        dp.timezone = self.tz.zone
        send_date = subscriberservice.determine_next_send_date(deliverypreference=dp)

        self.assertEqual(send_date.hour, dp.hour)
        self.assertEqual(send_date.minute, dp.minute)
        self.assertEqual(str(send_date.date()), str(self.now.date()))

    def test_next_send_tomorrow(self):
        dp = DeliveryPreference()
        weekday = subscriberservice.weekdays[(self.weekday_num+1)%7]

        setattr(dp, weekday, True)

        delivery_time = self.now + timedelta(hours=1)
        dp.hour = delivery_time.hour
        dp.minute = delivery_time.minute
        dp.timezone = self.tz.zone
        send_date = subscriberservice.determine_next_send_date(deliverypreference=dp)

        self.assertEqual(send_date.hour, dp.hour)
        self.assertEqual(send_date.minute, dp.minute)
        self.assertEqual(
            str(send_date.date()), 
            str((self.now + timedelta(hours=24)).date())
            )

    def test_delivery_before_current_hour_next_week(self):
        dp = DeliveryPreference()
        weekday = subscriberservice.weekdays[(self.weekday_num+1)%7]

        setattr(dp, self.weekday_name, True)

        delivery_time = self.now - timedelta(hours=1)
        dp.hour = delivery_time.hour
        dp.minute = delivery_time.minute
        dp.timezone = self.tz.zone
        send_date = subscriberservice.determine_next_send_date(deliverypreference=dp)

        self.assertEqual(send_date.hour, dp.hour)
        self.assertEqual(send_date.minute, dp.minute)
        self.assertEqual(
            str(send_date.date()), 
            str((self.now + timedelta(days=7)).date())
            )

    def test_delivery_before_current_hour(self):
        dp = DeliveryPreference()
        weekday = subscriberservice.weekdays[(self.weekday_num+1)%7]

        setattr(dp, self.weekday_name, True)
        setattr(dp, weekday, True)

        delivery_time = self.now - timedelta(hours=1)
        dp.hour = delivery_time.hour
        dp.minute = delivery_time.minute
        dp.timezone = self.tz.zone
        send_date = subscriberservice.determine_next_send_date(deliverypreference=dp)

        self.assertEqual(send_date.hour, dp.hour)
        self.assertEqual(send_date.minute, dp.minute)
        self.assertEqual(
            str(send_date.date()), 
            str((self.now + timedelta(hours=24)).date())
            )