cf62e65c55
Community edition generated from private repo via sync pipeline. Includes 9 plugins (WordPress, WooCommerce, WP Advanced, Gitea, n8n, Supabase, OpenPanel, Appwrite, Directus) with ~587 tools. Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
61 lines
1.8 KiB
Python
61 lines
1.8 KiB
Python
#!/usr/bin/env python3
|
|
"""
|
|
Test context-based API key isolation after fixing circular import.
|
|
"""
|
|
|
|
from core.context import clear_api_key_context, get_api_key_context, set_api_key_context
|
|
|
|
print("=" * 60)
|
|
print("Testing Context API")
|
|
print("=" * 60)
|
|
|
|
# Test 1: Set and get context
|
|
print("\nTest 1: Set and get context")
|
|
set_api_key_context(
|
|
key_id="key_test123", project_id="wordpress_site4", scope="admin", is_global=False
|
|
)
|
|
ctx = get_api_key_context()
|
|
assert ctx["key_id"] == "key_test123"
|
|
assert ctx["project_id"] == "wordpress_site4"
|
|
assert not ctx["is_global"]
|
|
print(" ✓ PASS: Context set and retrieved correctly")
|
|
|
|
# Test 2: Check access logic
|
|
print("\nTest 2: Access logic for per-project key")
|
|
allowed_project = ctx["project_id"]
|
|
target_project = "wordpress_site4"
|
|
if allowed_project == target_project:
|
|
print(" ✓ PASS: Access allowed (same project)")
|
|
else:
|
|
print(" ✗ FAIL: Should allow access")
|
|
|
|
# Test 3: Different project access
|
|
print("\nTest 3: Access logic for different project")
|
|
target_project = "wordpress_site1"
|
|
if allowed_project != target_project:
|
|
print(" ✓ PASS: Access denied (different project)")
|
|
else:
|
|
print(" ✗ FAIL: Should deny access")
|
|
|
|
# Test 4: Global key
|
|
print("\nTest 4: Global key access")
|
|
set_api_key_context(key_id="key_global", project_id="*", scope="admin", is_global=True)
|
|
ctx = get_api_key_context()
|
|
if ctx["is_global"]:
|
|
print(" ✓ PASS: Global key bypasses project check")
|
|
else:
|
|
print(" ✗ FAIL: Should be global")
|
|
|
|
# Test 5: Clear context
|
|
print("\nTest 5: Clear context")
|
|
clear_api_key_context()
|
|
ctx = get_api_key_context()
|
|
if ctx is None:
|
|
print(" ✓ PASS: Context cleared")
|
|
else:
|
|
print(" ✗ FAIL: Context should be None")
|
|
|
|
print("\n" + "=" * 60)
|
|
print("✅ All context tests passed!")
|
|
print("=" * 60)
|