feat: Add test scripts for dose parsing and UI tracking functionality

medicines.json

@@ -51,6 +51,7 @@
       "display_name": "Quetiapine",
       "dosage_info": "25 mg",
       "quick_doses": [
+        "12",
         "25",
         "50",
         "100"

scripts/test_dose_parsing.py

@@ -0,0 +1,95 @@
+#!/usr/bin/env python3
+"""
+Test the complete dose tracking flow: load -> display -> add -> save
+"""
+
+import os
+import sys
+from datetime import datetime
+
+# Add the src directory to Python path
+sys.path.insert(0, os.path.join(os.path.dirname(__file__), "..", "src"))
+
+from init import logger
+from ui_manager import UIManager
+
+
+def test_dose_parsing():
+    """Test dose parsing functions directly."""
+
+    # Mock a UI manager instance for testing
+    class MockManager:
+        def get_all_medicines(self):
+            return ["bupropion"]
+
+        def get_all_pathologies(self):
+            return []
+
+    ui_manager = UIManager(None, logger, MockManager(), MockManager(), None)
+
+    # Test 1: Parse storage format to display format
+    print("=== Test 1: Storage to Display Format ===")
+    storage_format = "2025-08-07 08:00:00:150mg|2025-08-07 12:00:00:150mg"
+    print(f"Input (storage): {storage_format}")
+
+    # This would normally be done by _populate_dose_history
+    formatted_doses = []
+    for dose_entry in storage_format.split("|"):
+        if ":" in dose_entry:
+            parts = dose_entry.rsplit(":", 1)
+            if len(parts) == 2:
+                timestamp, dose = parts
+                try:
+                    dt = datetime.strptime(timestamp, "%Y-%m-%d %H:%M:%S")
+                    time_str = dt.strftime("%I:%M %p")
+                    formatted_doses.append(f"• {time_str} - {dose}")
+                except ValueError:
+                    formatted_doses.append(f"• {dose_entry}")
+            else:
+                formatted_doses.append(f"• {dose_entry}")
+        else:
+            formatted_doses.append(f"• {dose_entry}")
+
+    display_format = "\n".join(formatted_doses)
+    print(f"Output (display): {display_format}")
+
+    # Test 2: Add new dose in display format
+    print("\n=== Test 2: Add New Dose ===")
+    new_timestamp = datetime.now().strftime("%I:%M %p")
+    new_dose = f"• {new_timestamp} - 150mg"
+    print(f"New dose to add: {new_dose}")
+
+    updated_display = display_format + f"\n{new_dose}"
+    print(f"Updated display: {updated_display}")
+
+    # Test 3: Parse display format back to storage format
+    print("\n=== Test 3: Display to Storage Format ===")
+    test_date = "2025-08-07"
+    parsed_storage = ui_manager._parse_dose_history_for_saving(
+        updated_display, test_date
+    )
+    print(f"Input (display): {updated_display}")
+    print(f"Output (storage): {parsed_storage}")
+
+    # Test 4: Verify round-trip integrity
+    print("\n=== Test 4: Round-trip Test ===")
+    print(f"Original storage: {storage_format}")
+    print(f"Final storage:    {parsed_storage}")
+
+    # Check if we preserved the original doses
+    original_count = len(storage_format.split("|"))
+    final_count = len(parsed_storage.split("|")) if parsed_storage else 0
+    print(f"Dose count: {original_count} -> {final_count}")
+
+    if final_count == original_count + 1:
+        print("✅ SUCCESS: New dose was added without replacing existing ones")
+    elif final_count == original_count:
+        print("❌ FAILURE: No new dose was added")
+    elif final_count < original_count:
+        print("❌ FAILURE: Existing doses were lost")
+    else:
+        print(f"⚠️  UNEXPECTED: Dose count changed unexpectedly ({final_count})")
+
+
+if __name__ == "__main__":
+    test_dose_parsing()

scripts/test_dose_tracking_ui.py

@@ -0,0 +1,111 @@
+#!/usr/bin/env python3
+"""
+Test script for dose tracking UI in edit window.
+Tests the specific issue where adding new doses replaces existing ones.
+"""
+
+import os
+import sys
+import tkinter as tk
+from datetime import datetime
+
+# Add the src directory to Python path
+sys.path.insert(0, os.path.join(os.path.dirname(__file__), "..", "src"))
+
+from init import logger
+from medicine_manager import MedicineManager
+from pathology_manager import PathologyManager
+from theme_manager import ThemeManager
+from ui_manager import UIManager
+
+
+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:
+        from medicine_manager import Medicine
+
+        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()

src/ui_manager.py

@@ -1033,12 +1033,17 @@ class UIManager:
                     if dose:
                         from datetime import datetime
 
-                        timestamp = datetime.now().strftime("%H:%M")
-                        new_dose = f"{timestamp}: {dose}"
+                        # Format timestamp for display (12-hour format with AM/PM)
+                        timestamp = datetime.now().strftime("%I:%M %p")
+                        new_dose = f"• {timestamp} - {dose}"
 
                         current_doses = dose_var.get()
                         if current_doses and current_doses.strip():
-                            dose_var.set(current_doses + f"\n{new_dose}")
+                            # Check if current content is placeholder text
+                            if "No doses recorded" in current_doses:
+                                dose_var.set(new_dose)
+                            else:
+                                dose_var.set(current_doses + f"\n{new_dose}")
                         else:
                             dose_var.set(new_dose)
 

verify_docs_consolidation.py

@@ -1,125 +0,0 @@
-#!/usr/bin/env python3
-"""
-Documentation Consolidation Verification Script
-
-This script verifies that the documentation consolidation was successful
-and provides a summary of the new documentation structure.
-"""
-
-import sys
-from pathlib import Path
-
-
-def check_file_exists(filename):
-    """Check if a file exists and return its size."""
-    path = Path(filename)
-    if path.exists():
-        size = path.stat().st_size
-        lines = len(path.read_text().splitlines()) if path.suffix == ".md" else 0
-        return True, size, lines
-    return False, 0, 0
-
-
-def main():
-    """Verify the documentation consolidation."""
-
-    print("TheChart Documentation Consolidation Verification")
-    print("=" * 55)
-    print()
-
-    # Primary consolidated documentation
-    primary_docs = [
-        ("CONSOLIDATED_DOCS.md", "Complete comprehensive documentation"),
-        ("README.md", "Updated project overview with consolidated refs"),
-        ("DOCS_CONSOLIDATION_SUMMARY.md", "Consolidation process summary"),
-    ]
-
-    # Existing documentation (preserved)
-    existing_docs = [
-        ("USER_GUIDE.md", "User manual and features"),
-        ("DEVELOPER_GUIDE.md", "Development setup and architecture"),
-        ("API_REFERENCE.md", "Technical API documentation"),
-        ("CHANGELOG.md", "Version history"),
-        ("UI_FLICKERING_FIX_SUMMARY.md", "Latest performance improvements"),
-        ("IMPROVEMENTS_SUMMARY.md", "Recent feature additions"),
-    ]
-
-    # Documentation hub
-    hub_docs = [
-        ("docs/README.md", "Documentation navigation hub"),
-    ]
-
-    print("📚 PRIMARY CONSOLIDATED DOCUMENTATION")
-    print("-" * 45)
-    for filename, description in primary_docs:
-        exists, size, lines = check_file_exists(filename)
-        status = "✅" if exists else "❌"
-        size_info = f"({size:,} bytes, {lines} lines)" if exists else ""
-        print(f"{status} {filename:<35} - {description}")
-        if size_info:
-            print(f"   {size_info}")
-
-    print("\n📖 EXISTING DOCUMENTATION (PRESERVED)")
-    print("-" * 45)
-    for filename, description in existing_docs:
-        exists, size, lines = check_file_exists(filename)
-        status = "✅" if exists else "❌"
-        print(f"{status} {filename:<35} - {description}")
-
-    print("\n🏠 DOCUMENTATION HUB")
-    print("-" * 45)
-    for filename, description in hub_docs:
-        exists, size, lines = check_file_exists(filename)
-        status = "✅" if exists else "❌"
-        print(f"{status} {filename:<35} - {description}")
-
-    # Verify consolidated docs content
-    print("\n🔍 CONSOLIDATED DOCS CONTENT VERIFICATION")
-    print("-" * 45)
-
-    consolidated_path = Path("CONSOLIDATED_DOCS.md")
-    if consolidated_path.exists():
-        content = consolidated_path.read_text()
-
-        required_sections = [
-            "Quick Start",
-            "User Guide",
-            "Developer Guide",
-            "Features & Capabilities",
-            "Technical Architecture",
-            "Recent Improvements",
-            "API Reference",
-            "Troubleshooting",
-            "Contributing",
-        ]
-
-        for section in required_sections:
-            if f"# {section}" in content or f"## {section}" in content:
-                print(f"✅ Section: {section}")
-            else:
-                print(f"❌ Missing: {section}")
-    else:
-        print("❌ CONSOLIDATED_DOCS.md not found")
-
-    print("\n📊 CONSOLIDATION SUMMARY")
-    print("-" * 45)
-    print("✅ Created comprehensive CONSOLIDATED_DOCS.md")
-    print("✅ Updated README.md with consolidated references")
-    print("✅ Updated docs/README.md as navigation hub")
-    print("✅ Preserved all existing documentation")
-    print("✅ Added documentation consolidation summary")
-    print("✅ Maintained backward compatibility")
-
-    print("\n💡 USAGE RECOMMENDATIONS")
-    print("-" * 45)
-    print("🌟 For comprehensive info: CONSOLIDATED_DOCS.md")
-    print("⚡ For quick user access: USER_GUIDE.md")
-    print("⚡ For quick dev access: DEVELOPER_GUIDE.md")
-    print("📚 For navigation help: docs/README.md")
-
-    print("\n✅ Documentation consolidation completed successfully!")
-    return 0
-
-
-if __name__ == "__main__":
-    sys.exit(main())