Add test script for etag

Author: haacked

test_etag.py

@@ -0,0 +1,96 @@
+#!/usr/bin/env python
+"""
+Test script to observe ETag support for local evaluation polling.
+
+This script starts a PostHog client with local evaluation enabled and
+watches the polling behavior to see ETag/304 responses in action.
+
+Usage:
+    1. Copy .env.example to .env and fill in your credentials
+    2. Run: uv run python test_etag.py
+"""
+
+import logging
+import os
+import time
+
+# Set up logging BEFORE importing posthog
+logging.basicConfig(
+    level=logging.DEBUG,
+    format="%(asctime)s [%(levelname)s] %(name)s: %(message)s",
+    datefmt="%H:%M:%S",
+)
+
+import posthog
+
+
+def load_env_file():
+    """Load environment variables from .env file if it exists."""
+    env_path = os.path.join(os.path.dirname(__file__), ".env")
+    if os.path.exists(env_path):
+        with open(env_path, "r") as f:
+            for line in f:
+                line = line.strip()
+                if line and not line.startswith("#") and "=" in line:
+                    key, value = line.split("=", 1)
+                    os.environ.setdefault(key.strip(), value.strip())
+
+
+load_env_file()
+
+project_key = os.getenv("POSTHOG_PROJECT_API_KEY", "")
+personal_api_key = os.getenv("POSTHOG_PERSONAL_API_KEY", "")
+host = os.getenv("POSTHOG_HOST", "https://us.posthog.com")
+
+if not project_key or not personal_api_key:
+    print("Missing credentials. Set POSTHOG_PROJECT_API_KEY and POSTHOG_PERSONAL_API_KEY")
+    print("Or copy .env.example to .env and fill in your values")
+    exit(1)
+
+print("=" * 70)
+print("ETag Support Test")
+print("=" * 70)
+print(f"Host: {host}")
+print(f"Project Key: {project_key[:12]}...")
+print(f"Poll Interval: 5 seconds")
+print()
+print("Watch for these log messages:")
+print("  - First poll: 'GET ... completed successfully' (200 response)")
+print("  - Subsequent polls: 'Flags not modified (304)' (if flags unchanged)")
+print("  - 'GET ... returned 304 Not Modified' from request layer")
+print()
+print("Press Ctrl+C to stop")
+print("=" * 70)
+print()
+
+# Configure PostHog with short poll interval to see ETag behavior quickly
+posthog.api_key = project_key
+posthog.project_api_key = project_key
+posthog.personal_api_key = personal_api_key
+posthog.host = host
+posthog.debug = True
+posthog.poll_interval = 5  # Poll every 5 seconds to see ETag behavior
+
+try:
+    # This triggers initial flag load and starts the poller
+    print("Starting client and loading feature flags...")
+    flags = posthog.get_all_flags("test-user", only_evaluate_locally=True)
+    print(f"\nLoaded {len(flags)} flags: {list(flags.keys())[:5]}{'...' if len(flags) > 5 else ''}")
+    print()
+
+    # Keep running so we can observe the poller
+    poll_count = 0
+    while True:
+        poll_count += 1
+        time.sleep(6)  # Slightly longer than poll interval
+
+        # Show current flags after each poll
+        flags = posthog.get_all_flags("test-user", only_evaluate_locally=True)
+        print(f"\n--- After poll #{poll_count + 1} ---")
+        print(f"Flags ({len(flags)}): {flags}")
+        print()
+
+except KeyboardInterrupt:
+    print("\n\nShutting down...")
+    posthog.shutdown()
+    print("Done!")