def test_insert_notification_history_delete_notifications_insert_for_key_type( sample_template): create_notification(template=sample_template, created_at=datetime.utcnow() - timedelta(hours=4), status='delivered', key_type='normal') create_notification(template=sample_template, created_at=datetime.utcnow() - timedelta(hours=4), status='delivered', key_type='team') with_test_key = create_notification(template=sample_template, created_at=datetime.utcnow() - timedelta(hours=4), status='delivered', key_type='test') del_count = insert_notification_history_delete_notifications( notification_type=sample_template.template_type, service_id=sample_template.service_id, timestamp_to_delete_backwards_from=datetime.utcnow()) assert del_count == 2 notifications = Notification.query.all() history_rows = NotificationHistory.query.all() assert len(notifications) == 1 assert with_test_key.id == notifications[0].id assert len(history_rows) == 2
def test_insert_notification_history_delete_notifications_only_insert_delete_for_given_service( sample_email_template): notification_to_move = create_notification(template=sample_email_template, created_at=datetime.utcnow() + timedelta(minutes=4), status='delivered') another_service = create_service(service_name='Another service') another_template = create_template(service=another_service, template_type='email') notification_to_stay = create_notification(template=another_template, created_at=datetime.utcnow() + timedelta(minutes=4), status='delivered') del_count = insert_notification_history_delete_notifications( notification_type=sample_email_template.template_type, service_id=sample_email_template.service_id, timestamp_to_delete_backwards_from=datetime.utcnow() + timedelta(hours=1)) assert del_count == 1 notifications = Notification.query.all() history_rows = NotificationHistory.query.all() assert len(notifications) == 1 assert len(history_rows) == 1 assert notifications[0].id == notification_to_stay.id assert history_rows[0], id == notification_to_move.id
def test_insert_notification_history_delete_notifications_more_notifications_than_query_limit( sample_template): create_notification(template=sample_template, created_at=datetime.utcnow() + timedelta(minutes=4), status='delivered') create_notification(template=sample_template, created_at=datetime.utcnow() + timedelta(minutes=20), status='permanent-failure') create_notification(template=sample_template, created_at=datetime.utcnow() + timedelta(minutes=30), status='temporary-failure') del_count = insert_notification_history_delete_notifications( notification_type=sample_template.template_type, service_id=sample_template.service_id, timestamp_to_delete_backwards_from=datetime.utcnow() + timedelta(hours=1), qry_limit=1) assert del_count == 1 notifications = Notification.query.all() history_rows = NotificationHistory.query.all() assert len(history_rows) == 1 assert len(notifications) == 2
def test_insert_notification_history_delete_notifications( sample_email_template): # should be deleted n1 = create_notification(template=sample_email_template, created_at=datetime.utcnow() - timedelta(days=1, minutes=4), status='delivered') n2 = create_notification(template=sample_email_template, created_at=datetime.utcnow() - timedelta(days=1, minutes=20), status='permanent-failure') n3 = create_notification(template=sample_email_template, created_at=datetime.utcnow() - timedelta(days=1, minutes=30), status='temporary-failure') n4 = create_notification(template=sample_email_template, created_at=datetime.utcnow() - timedelta(days=1, minutes=59), status='temporary-failure') n5 = create_notification(template=sample_email_template, created_at=datetime.utcnow() - timedelta(days=1, hours=1), status='sending') n6 = create_notification(template=sample_email_template, created_at=datetime.utcnow() - timedelta(days=1, minutes=61), status='pending') n7 = create_notification(template=sample_email_template, created_at=datetime.utcnow() - timedelta(days=1, hours=1, seconds=1), status='validation-failed') n8 = create_notification(template=sample_email_template, created_at=datetime.utcnow() - timedelta(days=1, minutes=20), status='created') # should NOT be deleted - wrong status n9 = create_notification(template=sample_email_template, created_at=datetime.utcnow() - timedelta(hours=1), status='delivered') n10 = create_notification(template=sample_email_template, created_at=datetime.utcnow() - timedelta(hours=1), status='technical-failure') n11 = create_notification(template=sample_email_template, created_at=datetime.utcnow() - timedelta(hours=23, minutes=59), status='created') ids_to_move = sorted( [n1.id, n2.id, n3.id, n4.id, n5.id, n6.id, n7.id, n8.id]) ids_to_keep = sorted([n9.id, n10.id, n11.id]) del_count = insert_notification_history_delete_notifications( notification_type=sample_email_template.template_type, service_id=sample_email_template.service_id, timestamp_to_delete_backwards_from=datetime.utcnow() - timedelta(days=1)) assert del_count == 8 notifications = Notification.query.all() history_rows = NotificationHistory.query.all() assert len(history_rows) == 8 assert ids_to_move == sorted([x.id for x in history_rows]) assert len(notifications) == 3 assert ids_to_keep == sorted([x.id for x in notifications])