feat: Add test script for verifying multiple dose punching and saving behavior

test_multiple_punch_save.py

@@ -0,0 +1,141 @@
+#!/usr/bin/env python3
+"""
+Test script to verify multiple dose punching and saving behavior.
+"""
+
+import os
+import sys
+import tkinter as tk
+
+# Add the src directory to the path so we can import our modules
+sys.path.insert(0, os.path.join(os.path.dirname(__file__), "src"))
+
+import logging
+
+from ui_manager import UIManager
+
+
+def test_multiple_punch_and_save():
+    """Test multiple dose punching followed by save."""
+    print("Testing multiple dose punching and save functionality...")
+
+    # Create a test Tkinter root
+    root = tk.Tk()
+    root.title("Test Root Window")
+    root.geometry("200x100")  # Small root window
+
+    # Create a logger
+    logger = logging.getLogger("test_logger")
+    logger.setLevel(logging.DEBUG)
+
+    # Create UIManager
+    ui_manager = UIManager(root, logger)
+
+    # Sample dose data for testing
+    sample_dose_data = {
+        "bupropion": "2025-01-15 08:00:00:300mg",
+        "hydroxyzine": "",
+        "gabapentin": "",
+        "propranolol": "",
+    }
+
+    # Sample values for the edit window (14 fields for new CSV format)
+    sample_values = (
+        "01/15/2025",  # date
+        5,  # depression
+        3,  # anxiety
+        7,  # sleep
+        6,  # appetite
+        1,  # bupropion
+        sample_dose_data["bupropion"],  # bupropion_doses
+        0,  # hydroxyzine
+        sample_dose_data["hydroxyzine"],  # hydroxyzine_doses
+        0,  # gabapentin
+        sample_dose_data["gabapentin"],  # gabapentin_doses
+        0,  # propranolol
+        sample_dose_data["propranolol"],  # propranolol_doses
+        "Test entry for multiple punch testing",  # note
+    )
+
+    # Track save calls
+    save_calls = []
+
+    # Define test callbacks
+    def test_save(*args):
+        save_calls.append(args)
+        print(f"✓ Save called with {len(args)} arguments")
+
+        # Print dose data specifically
+        if len(args) >= 12:  # Should have dose_data as last argument
+            dose_data = args[-1]  # Last argument should be dose_data
+            print("  Dose data received:")
+            for med, doses in dose_data.items():
+                print(f"    {med}: {doses}")
+
+        # Close window after save
+        if args and hasattr(args[0], "destroy"):
+            args[0].destroy()
+
+    def test_delete(*args):
+        print("Delete callback triggered")
+        if args and hasattr(args[0], "destroy"):
+            args[0].destroy()
+
+    callbacks = {
+        "save": test_save,
+        "delete": test_delete,
+    }
+
+    try:
+        # Create the edit window
+        edit_window = ui_manager.create_edit_window(sample_values, callbacks)
+        edit_window.geometry("600x400")  # Set a reasonable size
+        edit_window.lift()  # Bring to front
+        edit_window.focus_force()  # Force focus
+
+        print("✓ Edit window created")
+        print("✓ Now simulating multiple dose punches...")
+
+        # Let's simulate the manual process
+
+        print("\n=== Manual Test Instructions ===")
+        print("1. In the Bupropion field, enter '150mg' and click 'Take Bupropion'")
+        print("2. Enter '300mg' and click 'Take Bupropion' again")
+        print("3. You should see both doses in the text area")
+        print("4. Click 'Save' to persist the changes")
+        print("5. Check if both doses are saved to the CSV")
+        print("\nWindow will stay open for manual testing...")
+
+        # Wait for user to manually test
+        edit_window.wait_window()
+
+        # Check if save was called
+        if save_calls:
+            print("✓ Save was called successfully")
+            return True
+        else:
+            print("✗ Save was not called")
+            return False
+
+    except Exception as e:
+        print(f"✗ Error during test: {e}")
+        import traceback
+
+        traceback.print_exc()
+        return False
+
+    finally:
+        root.destroy()
+
+
+if __name__ == "__main__":
+    print("Testing Multiple Dose Punching and Save")
+    print("=" * 40)
+
+    success = test_multiple_punch_and_save()
+
+    if success:
+        print("\n✅ Multiple punch and save test completed!")
+    else:
+        print("\n❌ Multiple punch and save test failed!")
+        sys.exit(1)