#!/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)
