Update Heicode sub-mode runtime changes

test_phase1.py

@@ -0,0 +1,149 @@
+#!/usr/bin/env python3
+"""Test script for Phase 1 implementation."""
+import sys
+import os
+
+# Add current directory to path
+sys.path.insert(0, os.path.dirname(__file__))
+
+def test_imports():
+    """Test that all modules can be imported."""
+    print("Testing Phase 1 imports...")
+
+    try:
+        from config.error_codes import ErrorCode
+        print("✅ config.error_codes imported")
+        print(f"   Error codes: {[e.value for e in ErrorCode][:3]}...")
+    except Exception as e:
+        print(f"❌ config.error_codes failed: {e}")
+        return False
+
+    try:
+        from config.settings import settings
+        print("✅ config.settings imported")
+        print(f"   Service token configured: {bool(settings.HEICODE_SERVICE_TOKEN)}")
+    except Exception as e:
+        print(f"❌ config.settings failed: {e}")
+        return False
+
+    try:
+        from api.agnet.models import HealthCheckResponse, HealthCheckData, ErrorResponse
+        print("✅ api.agnet.models imported")
+    except Exception as e:
+        print(f"❌ api.agnet.models failed: {e}")
+        return False
+
+    try:
+        from api.agnet.validators import scan_for_sensitive_fields, validate_no_sensitive_fields
+        print("✅ api.agnet.validators imported")
+    except Exception as e:
+        print(f"❌ api.agnet.validators failed: {e}")
+        return False
+
+    try:
+        from api.agnet.auth import verify_service_token, extract_headers
+        print("✅ api.agnet.auth imported")
+    except Exception as e:
+        print(f"❌ api.agnet.auth failed: {e}")
+        return False
+
+    try:
+        from api.agnet.idempotency import idempotency_cache
+        print("✅ api.agnet.idempotency imported")
+        print(f"   Redis available: {idempotency_cache.redis_client is not None}")
+    except Exception as e:
+        print(f"❌ api.agnet.idempotency failed: {e}")
+        return False
+
+    try:
+        from api.agnet.router import router
+        print("✅ api.agnet.router imported")
+        print(f"   Router prefix: {router.prefix}")
+    except Exception as e:
+        print(f"❌ api.agnet.router failed: {e}")
+        return False
+
+    return True
+
+
+def test_validators():
+    """Test sensitive field scanner."""
+    print("\nTesting validators...")
+
+    from api.agnet.validators import scan_for_sensitive_fields
+
+    # Test case 1: Simple sensitive field
+    result = scan_for_sensitive_fields({"password": "test123"})
+    assert result == ["password"], f"Expected ['password'], got {result}"
+    print("✅ Simple sensitive field detected")
+
+    # Test case 2: Nested sensitive field
+    result = scan_for_sensitive_fields({"user": {"token": "abc"}})
+    assert result == ["user.token"], f"Expected ['user.token'], got {result}"
+    print("✅ Nested sensitive field detected")
+
+    # Test case 3: Multiple sensitive fields
+    result = scan_for_sensitive_fields({
+        "name": "john",
+        "password": "test",
+        "config": {"api_key": "secret"}
+    })
+    assert "password" in result and "config.api_key" in result
+    print("✅ Multiple sensitive fields detected")
+
+    # Test case 4: No sensitive fields
+    result = scan_for_sensitive_fields({"name": "john", "age": 30})
+    assert result == [], f"Expected [], got {result}"
+    print("✅ Clean payload passed")
+
+    return True
+
+
+def test_app_integration():
+    """Test that app.py includes the new router."""
+    print("\nTesting app.py integration...")
+
+    try:
+        from app import app
+
+        # Check if agnet router is registered
+        agnet_routes = [route for route in app.routes if '/agnet' in str(route.path)]
+
+        if agnet_routes:
+            print(f"✅ Agnet router registered with {len(agnet_routes)} routes")
+            for route in agnet_routes[:3]:
+                print(f"   - {route.path}")
+        else:
+            print("❌ Agnet router not found in app")
+            return False
+
+        return True
+    except Exception as e:
+        print(f"❌ App integration test failed: {e}")
+        return False
+
+
+if __name__ == "__main__":
+    print("=" * 60)
+    print("Phase 1 Implementation Test")
+    print("=" * 60)
+
+    success = True
+
+    if not test_imports():
+        success = False
+
+    if not test_validators():
+        success = False
+
+    if not test_app_integration():
+        success = False
+
+    print("\n" + "=" * 60)
+    if success:
+        print("✅ All Phase 1 tests passed!")
+    else:
+        print("❌ Some tests failed")
+    print("=" * 60)
+
+    sys.exit(0 if success else 1)