test_wizard.py

import os
import json
from server import set_secret, list_secrets, save_connection, list_connections

def test_wizard():
    print("Testing API Wizard Tools...")
    
    # Test set_secret
    res = set_secret("TEST_KEY", "test_value_123")
    print(f"set_secret: {res}")
    
    # Test list_secrets
    secrets = list_secrets()
    print(f"list_secrets: {secrets}")
    assert "TEST_KEY" in secrets
    
    # Test save_connection
    res = save_connection(
        name="httpbin",
        base_url="https://httpbin.org",
        secret_key_name="TEST_KEY",
        auth_type="Bearer"
    )
    print(f"save_connection: {res}")
    
    # Test list_connections
    connections = list_connections()
    print(f"list_connections: {list(connections.keys())}")
    assert "httpbin" in connections
    
    print("\nAll local tests passed!")

if __name__ == "__main__":
    test_wizard()