def test_only_return_unnotified_expired_allocations(self):
sdate_tz = localize_timezone(2016, 2, 2, 'America/Denver')
edate_tz = localize_timezone(2017, 2, 2, 'America/Denver')
alloc_dict_not_notified = {
'project': self.proj,
'allocation_id': 'ucb7_summit2',
'amount': '50000',
'start_date': sdate_tz,
'end_date': edate_tz,
'expiration_notice_sent': False
}
alloc_not_notified = Allocation.objects.create(
**alloc_dict_not_notified)
alloc_dict_notified = {
'project': self.proj,
'allocation_id': 'ucb6_summit2',
'amount': '50000',
'start_date': sdate_tz,
'end_date': edate_tz,
'expiration_notice_sent': True
}
alloc_notified = Allocation.objects.create(**alloc_dict_notified)
expiration_notifier = ExpirationNotifier(False)
actual_allocations = expiration_notifier.get_expired_allocations_with_no_sent_notification(
)
expected_query_set = [alloc_not_notified]
self.assertEqual(expected_query_set, list(actual_allocations))
def test_send_upcoming_expiration_notice(self):
sdate_tz = localize_timezone(2016, 2, 2, 'America/Denver')
edate_notice_needed_tz = localize_timezone(2018, 2, 21,
'America/Denver')
alloc_dict_notification_needed = {
'project': self.proj,
'allocation_id': 'ucb7_summit2',
'amount': '50000',
'start_date': sdate_tz,
'end_date': edate_notice_needed_tz,
'expiration_notice_sent': False
}
alloc_notification_needed = Allocation.objects.create(
**alloc_dict_notification_needed)
current_date_for_test_tz = localize_timezone(2018, 2, 1,
'America/Denver')
with mock.patch('mailer.signals.allocation_expiring.send') as mock_send, \
mock.patch('django.utils.timezone.now', return_value=current_date_for_test_tz):
notification_interval = 20
upcoming_expiration_notifier = UpcomingExpirationNotifier(
False, notification_interval)
upcoming_expiration_notifier.send_upcoming_expiration_notices()
self.assertEqual(mock_send.call_count, 1)
def test_return_soon_to_expire_allocations_at_interval_date(self):
sdate_tz = localize_timezone(2016, 2, 2, 'America/Denver')
edate_notice_needed_tz = localize_timezone(2018, 2, 21,
'America/Denver')
alloc_dict_notification_needed = {
'project': self.proj,
'allocation_id': 'ucb7_summit2',
'amount': '50000',
'start_date': sdate_tz,
'end_date': edate_notice_needed_tz,
'expiration_notice_sent': False
}
alloc_notification_needed = Allocation.objects.create(
**alloc_dict_notification_needed)
# Expired allocations don't need upcoming notification
edate_expired_tz = localize_timezone(2017, 2, 2, 'America/Denver')
alloc_dict_expired = {
'project': self.proj,
'allocation_id': 'ucb6_summit2',
'amount': '50000',
'start_date': sdate_tz,
'end_date': edate_expired_tz,
'expiration_notice_sent': True
}
alloc_expired = Allocation.objects.create(**alloc_dict_expired)
# Notification date in future
edate_far_in_future_tz = localize_timezone(2030, 2, 2,
'America/Denver')
alloc_dict_no_notification = {
'project': self.proj,
'allocation_id': 'ucb8_summit3',
'amount': '50000',
'start_date': sdate_tz,
'end_date': edate_far_in_future_tz,
'expiration_notice_sent': False
}
alloc_no_notification = Allocation.objects.create(
**alloc_dict_no_notification)
current_date_for_test_tz = localize_timezone(2018, 2, 1,
'America/Denver')
with mock.patch('django.utils.timezone.now',
return_value=current_date_for_test_tz):
notification_interval = 20
upcoming_expiration_notifier = UpcomingExpirationNotifier(
False, notification_interval)
actual_allocations = upcoming_expiration_notifier.get_expiring_allocations_for_interval(
)
expected_query_set = [alloc_notification_needed]
self.assertEqual(expected_query_set, list(actual_allocations))
def test_only_return_allocations_with_passed_expiration_date(self):
sdate_tz = localize_timezone(2016, 2, 2, 'America/Denver')
current_date_for_test_tz = localize_timezone(2018, 2, 2,
'America/Denver')
with mock.patch('django.utils.timezone.now',
return_value=current_date_for_test_tz):
edate_in_past_tz = localize_timezone(2017, 2, 2, 'America/Denver')
alloc_dict_end_date_passed = {
'project': self.proj,
'allocation_id': 'ucb7_summit2',
'amount': '50000',
'start_date': sdate_tz,
'end_date': edate_in_past_tz,
'expiration_notice_sent': False
}
alloc_end_date_passed = Allocation.objects.create(
**alloc_dict_end_date_passed)
edate_in_future_tz = localize_timezone(2019, 2, 2,
'America/Denver')
alloc_dict_end_date_future = {
'project': self.proj,
'allocation_id': 'ucb6_summit2',
'amount': '50000',
'start_date': sdate_tz,
'end_date': edate_in_future_tz,
'expiration_notice_sent': False
}
alloc_end_date_future = Allocation.objects.create(
**alloc_dict_end_date_future)
expiration_notifier = ExpirationNotifier(False)
actual_allocations = expiration_notifier.get_expired_allocations_with_no_sent_notification(
)
expected_query_set = [alloc_end_date_passed]
self.assertEqual(expected_query_set, list(actual_allocations))
def test_expiration_notice_field_set_when_notice_sent(self):
sdate_tz = localize_timezone(2016, 2, 2, 'America/Denver')
edate_tz = localize_timezone(2017, 2, 2, 'America/Denver')
alloc_dict_not_notified = {
'project': self.proj,
'allocation_id': 'ucb7_summit2',
'amount': '50000',
'start_date': sdate_tz,
'end_date': edate_tz,
'expiration_notice_sent': False
}
alloc_not_notified = Allocation.objects.create(
**alloc_dict_not_notified)
with mock.patch('mailer.signals.allocation_expired.send'):
expiration_notifier = ExpirationNotifier(False)
expiration_notifier.send_expiration_notices()
actual_allocation_object = Allocation.objects.get(
allocation_id='ucb7_summit2')
self.assertEqual(True,
actual_allocation_object.expiration_notice_sent)
def test_send_expiration_notice_for_expired_allocation(self):
sdate_tz = localize_timezone(2016, 2, 2, 'America/Denver')
edate_tz = localize_timezone(2017, 2, 2, 'America/Denver')
alloc_dict_not_notified = {
'project': self.proj,
'allocation_id': 'ucb7_summit2',
'amount': '50000',
'start_date': sdate_tz,
'end_date': edate_tz,
'expiration_notice_sent': False
}
alloc_not_notified = Allocation.objects.create(
**alloc_dict_not_notified)
with mock.patch('mailer.signals.allocation_expired.send') as mock_send:
expiration_notifier = ExpirationNotifier(False)
expiration_notifier.send_expiration_notices()
args, kwargs = mock_send.call_args
actual_model = kwargs['allocation']
self.assertEqual(mock_send.call_count, 1)
self.assertTrue(isinstance(actual_model, Allocation))