#!/usr/bin/env python3
"""
Test script to demonstrate the dose tracking functionality.
"""

import os
import sys
from datetime import datetime

sys.path.append(os.path.join(os.path.dirname(__file__), "src"))

from data_manager import DataManager
from init import logger


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

    # Initialize data manager
    data_manager = DataManager("thechart_data.csv", logger)

    # Test adding a dose
    today = datetime.now().strftime("%m/%d/%Y")
    print(f"Testing dose tracking for date: {today}")

    # Add some test doses
    test_doses = [
        ("bupropion", "150mg"),
        ("propranolol", "10mg"),
        ("bupropion", "150mg"),  # Second dose of same medicine
    ]

    for medicine, dose in test_doses:
        success = data_manager.add_medicine_dose(today, medicine, dose)
        if success:
            print(f"✓ Added {medicine} dose: {dose}")
        else:
            print(f"✗ Failed to add {medicine} dose: {dose}")

    # Retrieve and display doses
    print(f"\nDoses recorded for {today}:")
    medicines = ["bupropion", "hydroxyzine", "gabapentin", "propranolol"]

    for medicine in medicines:
        doses = data_manager.get_today_medicine_doses(today, medicine)
        if doses:
            print(f"{medicine.title()}:")
            for timestamp, dose in doses:
                print(f"  - {timestamp}: {dose}")
        else:
            print(f"{medicine.title()}: No doses recorded")


if __name__ == "__main__":
    test_dose_tracking()