#!/usr/bin/env python3
"""
Test script for dose tracking UI in edit window.
Tests the specific issue where adding new doses replaces existing ones.
"""
# ruff: noqa: E402

import sys
import tkinter as tk
from datetime import datetime
from pathlib import Path


def _ensure_src_on_path() -> None:
    src_dir = Path(__file__).resolve().parent.parent / "src"
    if str(src_dir) not in sys.path:
        sys.path.insert(0, str(src_dir))


_ensure_src_on_path()
from thechart.core.constants import LOG_LEVEL
from thechart.core.logger import init_logger
from thechart.managers import Medicine, MedicineManager, PathologyManager
from thechart.ui import ThemeManager
from thechart.ui.ui_manager import UIManager

logger = init_logger(__name__, testing_mode=(LOG_LEVEL == "DEBUG"))


def test_dose_tracking():
    """Test the dose tracking functionality."""

    # Create test window
    root = tk.Tk()
    root.title("Dose Tracking Test")
    root.geometry("800x600")

    # Initialize managers
    medicine_manager = MedicineManager(logger=logger)
    pathology_manager = PathologyManager(logger=logger)
    theme_manager = ThemeManager(root, logger)

    ui_manager = UIManager(
        root, logger, medicine_manager, pathology_manager, theme_manager
    )

    # Add a test medicine if none exist
    medicines = medicine_manager.get_all_medicines()
    if not medicines:
        test_medicine = Medicine(
            key="bupropion",
            display_name="Bupropion",
            dosage="150mg",
            color="#4CAF50",
            quick_doses=["150", "300"],
            is_default=True,
        )
        medicine_manager.add_medicine(test_medicine)
        print("Added test medicine: Bupropion")

    # Test data - simulate existing doses for today
    test_date = datetime.now().strftime("%Y-%m-%d")
    existing_doses = {"bupropion": "• 08:00 AM - 150mg\n• 12:00 PM - 150mg"}

    # Create test callbacks
    def test_save_callback(edit_win, *args):
        print(f"Save callback called with {len(args)} arguments")
        print(f"Arguments: {args}")
        # Don't actually save, just print for testing

    def test_delete_callback(edit_win):
        print("Delete callback called")
        edit_win.destroy()

    callbacks = {"save": test_save_callback, "delete": test_delete_callback}

    # Test values to populate the edit window
    test_values = (
        test_date,  # date
        0,  # pathology score (if any)
        1,  # medicine taken (bupropion)
        existing_doses["bupropion"],  # existing doses
        "Test note",  # note
    )

    print(f"Creating edit window with test values: {test_values}")

    # Create the edit window
    _ = ui_manager.create_edit_window(test_values, callbacks)

    # Add instructions label
    instructions = tk.Label(
        root,
        text="Instructions:\n"
        "1. The edit window should show existing doses: 08:00 AM and 12:00 PM\n"
        "2. Enter a new dose (e.g., 150) and click 'Take Bupropion'\n"
        "3. The new dose should be ADDED to existing doses, not replace them\n"
        "4. Click Save to see the final dose data in console",
        justify=tk.LEFT,
        wraplength=500,
        bg="lightyellow",
        padx=10,
        pady=10,
    )
    instructions.pack(pady=10, padx=10, fill=tk.X)

    print("Test setup complete. Check the edit window for dose tracking behavior.")
    print(
        "Expected behavior: New doses should be added to existing ones, "
        "not replace them."
    )

    root.mainloop()


if __name__ == "__main__":
    test_dose_tracking()