feat: Add comprehensive tests for punch button functionality and multiple dose handling

scripts/test_multiple_dose_issue.py

@@ -0,0 +1,184 @@
+#!/usr/bin/env python3
+"""
+Test script to isolate and verify the multiple dose saving issue.
+"""
+
+import os
+import sys
+import tkinter as tk
+
+# Add the src directory to the path
+sys.path.insert(0, os.path.join(os.path.dirname(__file__), "src"))
+
+import logging
+
+from ui_manager import UIManager
+
+
+def test_parse_dose_text():
+    """Test the _parse_dose_text function directly."""
+    print("🧪 Testing _parse_dose_text function...")
+
+    # Create a minimal UIManager for testing
+    root = tk.Tk()
+    root.withdraw()
+    logger = logging.getLogger("test")
+    ui_manager = UIManager(root, logger)
+
+    # Test data: multiple doses in the format shown in the text widget
+    test_text = """21:30: 150mg
+21:35: 300mg
+21:40: 75mg"""
+
+    test_date = "07/29/2025"
+
+    result = ui_manager._parse_dose_text(test_text, test_date)
+    print(f"Input text:\n{test_text}")
+    print(f"Date: {test_date}")
+    print(f"Parsed result: {result}")
+
+    # Count how many doses were parsed
+    if result:
+        dose_count = len(result.split("|"))
+        print(f"Number of doses parsed: {dose_count}")
+
+        if dose_count == 3:
+            print("✅ _parse_dose_text is working correctly!")
+            return True
+        else:
+            print("❌ _parse_dose_text is not parsing all doses!")
+            return False
+    else:
+        print("❌ _parse_dose_text returned empty result!")
+        return False
+
+    root.destroy()
+
+
+def test_punch_button_accumulation():
+    """Test that punch buttons properly accumulate in the text widget."""
+    print("\n🧪 Testing punch button dose accumulation...")
+
+    root = tk.Tk()
+    root.title("Punch Button Test")
+    root.geometry("400x300")
+
+    logger = logging.getLogger("test")
+    ui_manager = UIManager(root, logger)
+
+    # Sample values for creating edit window
+    sample_values = (
+        "07/29/2025",  # date
+        5,
+        3,
+        7,
+        6,  # symptoms
+        1,
+        "",  # bupropion, bupropion_doses
+        0,
+        "",  # hydroxyzine, hydroxyzine_doses
+        0,
+        "",  # gabapentin, gabapentin_doses
+        0,
+        "",  # propranolol, propranolol_doses
+        "Test entry",  # note
+    )
+
+    save_called = False
+    saved_dose_data = None
+
+    def test_save(*args):
+        nonlocal save_called, saved_dose_data
+        save_called = True
+        saved_dose_data = args[-1] if args else None
+
+        print("\n💾 Save callback triggered")
+        if saved_dose_data:
+            print("Dose data received:")
+            for med, doses in saved_dose_data.items():
+                if doses:
+                    dose_count = len(doses.split("|")) if "|" in doses else 1
+                    print(f"  {med}: {dose_count} dose(s) - {doses}")
+                else:
+                    print(f"  {med}: No doses")
+
+        # Close window
+        if args and hasattr(args[0], "destroy"):
+            args[0].destroy()
+
+    callbacks = {"save": test_save, "delete": lambda x: x.destroy()}
+
+    try:
+        edit_window = ui_manager.create_edit_window(sample_values, callbacks)
+        edit_window.lift()
+        edit_window.focus_force()
+
+        print("\n📝 TEST INSTRUCTIONS:")
+        print("1. Select ANY medicine (e.g., Bupropion)")
+        print("2. Enter '100mg' in the dose field")
+        print("3. Click 'Take [Medicine]' button")
+        print("4. Enter '200mg' in the dose field")
+        print("5. Click 'Take [Medicine]' button again")
+        print("6. Enter '300mg' in the dose field")
+        print("7. Click 'Take [Medicine]' button a third time")
+        print("8. Verify you see THREE entries in the text area")
+        print("9. Click 'Save'")
+        print("\n⏳ Please perform the test...")
+
+        edit_window.wait_window()
+
+        if save_called and saved_dose_data:
+            # Check if any medicine has multiple doses
+            multiple_found = False
+            for med, doses in saved_dose_data.items():
+                if doses and "|" in doses:
+                    dose_count = len(doses.split("|"))
+                    if dose_count > 1:
+                        print(f"✅ Multiple doses found for {med}: {dose_count} doses")
+                        multiple_found = True
+
+            if multiple_found:
+                print("✅ Multiple dose accumulation is working!")
+                return True
+            else:
+                print("❌ No multiple doses found in save data")
+                return False
+        else:
+            print("❌ Save was not called or no dose data received")
+            return False
+
+    except Exception as e:
+        print(f"❌ Error during test: {e}")
+        import traceback
+
+        traceback.print_exc()
+        return False
+    finally:
+        root.destroy()
+
+
+def main():
+    print("🔬 Multiple Dose Issue Investigation")
+    print("=" * 50)
+
+    os.chdir("/home/will/Code/thechart")
+
+    # Test 1: Parse function
+    parse_test = test_parse_dose_text()
+
+    # Test 2: UI workflow
+    ui_test = test_punch_button_accumulation()
+
+    print("\n📊 Results:")
+    print(f"  Parse function test: {'✅ PASS' if parse_test else '❌ FAIL'}")
+    print(f"  UI workflow test: {'✅ PASS' if ui_test else '❌ FAIL'}")
+
+    if parse_test and ui_test:
+        print("\n🎯 Multiple dose functionality appears to be working correctly")
+        print("If you're still experiencing issues, please describe the exact steps")
+    else:
+        print("\n🚨 Issues found with multiple dose functionality")
+
+
+if __name__ == "__main__":
+    main()