cfait 1.1.8

// SPDX-License-Identifier: GPL-3.0-or-later
//! Tests for recurrence date calculation logic.
use cfait::model::{DateType, Task, TaskStatus};
use chrono::{Duration, Utc};
use std::collections::HashMap;

// Helper to create a task due at Now + Offset days
fn create_task_due_in_days(days: i64, recurrence: &str) -> Task {
    let mut t = Task::new("Task", &HashMap::new(), None);
    // Use Utc directly as internal storage uses Utc
    let dt = Utc::now() + Duration::days(days);
    t.due = Some(DateType::Specific(dt));
    t.rrule = Some(recurrence.to_string());
    t.status = TaskStatus::Completed;
    t
}

#[test]
fn test_daily_recurrence() {
    // Create task due yesterday (-1 day)
    let mut t = create_task_due_in_days(-1, "FREQ=DAILY");
    let original_due = match t.due.as_ref().unwrap() {
        DateType::Specific(d) => *d,
        _ => panic!(),
    };

    let advanced = t.advance_recurrence();
    assert!(advanced);
    assert_eq!(t.status, TaskStatus::NeedsAction);

    let new_due = match t.due.as_ref().unwrap() {
        DateType::Specific(d) => *d,
        _ => panic!(),
    };

    // Since task was due yesterday, and we are "now" (today),
    // next occurrence after now should be Today (if time allows) or Tomorrow.
    // logic: find > now.
    // if original = yesterday. recurrence = daily.
    // sequence: yesterday, today, tomorrow...
    // if yesterday < now, check today.
    // if today > now (e.g. task was due at 23:00 yesterday, now is 10:00 today, recurrence is 23:00),
    // then today 23:00 > now 10:00. So next is today.
    // if task was due at 09:00 yesterday, now is 10:00 today.
    // yesterday 09:00 < now.
    // today 09:00 < now.
    // next is tomorrow 09:00.

    // Since we created it with `Utc::now() + days`, the time component matches `now`.
    // original = now - 24h.
    // next candidate = now.
    // is now > now? No (equal).
    // next candidate = now + 24h.
    // is now + 24h > now? Yes.
    // So expected is original + 2 days (i.e. Tomorrow relative to Now).

    let original_local = original_due.with_timezone(&chrono::Local).naive_local();
    let new_local = new_due.with_timezone(&chrono::Local).naive_local();
    let expected_local = original_local + Duration::days(2);

    let diff = new_local - expected_local;
    assert!(diff.num_seconds().abs() < 5, "Expected around tomorrow");
}

#[test]
fn test_weekly_recurrence() {
    // Due 8 days ago
    let mut t = create_task_due_in_days(-8, "FREQ=WEEKLY");
    let original_due = match t.due.as_ref().unwrap() {
        DateType::Specific(d) => *d,
        _ => panic!(),
    };

    let advanced = t.advance_recurrence();
    assert!(advanced);

    let new_due = match t.due.as_ref().unwrap() {
        DateType::Specific(d) => *d,
        _ => panic!(),
    };

    // Sequence: now-8d, now-1d, now+6d...
    // now-1d < now.
    // now+6d > now.
    // Expected: now + 6 days.
    // original (-8) + 14 = +6.

    let original_local = original_due.with_timezone(&chrono::Local).naive_local();
    let new_local = new_due.with_timezone(&chrono::Local).naive_local();
    let expected_local = original_local + Duration::days(14);

    let diff = new_local - expected_local;
    assert!(diff.num_seconds().abs() < 5);
}

#[test]
fn test_monthly_recurrence() {
    // Due 35 days ago (~1 month ago)
    let mut t = create_task_due_in_days(-35, "FREQ=MONTHLY");

    // We can't easily predict exact day due to variable month lengths without complex logic matching the rrule crate.
    // But we know it should be in the future relative to now.

    let advanced = t.advance_recurrence();
    assert!(advanced);

    let new_due = match t.due.as_ref().unwrap() {
        DateType::Specific(d) => *d,
        _ => panic!(),
    };

    assert!(new_due > Utc::now());
}

#[test]
fn test_custom_interval() {
    // Every 3 days. Due 4 days ago.
    // Sequence: -4, -1, +2.
    // -1 < now.
    // +2 > now.
    // Next should be +2 days from now.
    let mut t = create_task_due_in_days(-4, "FREQ=DAILY;INTERVAL=3");
    let original_due = match t.due.as_ref().unwrap() {
        DateType::Specific(d) => *d,
        _ => panic!(),
    };

    let advanced = t.advance_recurrence();
    assert!(advanced);

    let new_due = match t.due.as_ref().unwrap() {
        DateType::Specific(d) => *d,
        _ => panic!(),
    };

    let original_local = original_due.with_timezone(&chrono::Local).naive_local();
    let new_local = new_due.with_timezone(&chrono::Local).naive_local();
    let expected_local = original_local + Duration::days(6);

    let diff = new_local - expected_local;
    assert!(diff.num_seconds().abs() < 5);
}

#[test]
fn test_complex_weekday_recurrence() {
    // This implies we need to match specific weekdays.
    // It's safer to just check it advances to future.
    let mut t = create_task_due_in_days(-10, "FREQ=WEEKLY;BYDAY=MO");
    t.advance_recurrence();
    let new_due = match t.due.as_ref().unwrap() {
        DateType::Specific(d) => *d,
        _ => panic!(),
    };
    assert!(new_due > Utc::now());
}

#[test]
fn test_recurrence_preserves_time() {
    let mut t = Task::new("Time Test", &HashMap::new(), None);
    let dt = Utc::now();
    t.due = Some(DateType::Specific(dt));
    t.rrule = Some("FREQ=DAILY".to_string());

    t.advance_recurrence();

    let new_due = match t.due.as_ref().unwrap() {
        DateType::Specific(d) => *d,
        _ => panic!(),
    };

    // Check hour/minute match (allowing slight drift if rrule recalculates, but usually it preserves)
    // Actually rrule calculated from DTSTART preserves time.
    assert_eq!(
        dt.with_timezone(&chrono::Local).format("%H:%M").to_string(),
        new_due
            .with_timezone(&chrono::Local)
            .format("%H:%M")
            .to_string()
    );
}

#[test]
fn test_cancel_single_occurrence_daily() {
    // Create a task with daily recurrence due yesterday
    let t = create_task_due_in_days(-1, "FREQ=DAILY");
    let original_due = match t.due.as_ref().unwrap() {
        DateType::Specific(d) => *d,
        _ => panic!(),
    };

    // Recycle the task to a Cancelled state
    let (history, recycled) = t.recycle(TaskStatus::Cancelled, false);
    let advanced_task = recycled.expect("Should advance");

    assert_eq!(history.status, TaskStatus::Cancelled);
    assert_eq!(
        history.due.as_ref().unwrap().to_comparison_time(),
        original_due
    );

    assert_eq!(advanced_task.status, TaskStatus::NeedsAction);
    assert!(advanced_task.due.as_ref().unwrap().to_comparison_time() > Utc::now());
}