#!/usr/bin/env python3
# Copyright (c) 2024- Datalayer, Inc.
#
# BSD 3-Clause License
"""
Simple test script to verify the configuration system works correctly.
"""
import os
from jupyter_mcp_server.config import get_config, set_config, reset_config
def test_config():
"""Test the configuration singleton."""
print("Testing Jupyter MCP Configuration System")
print("=" * 50)
# Test default configuration
config = get_config()
print(f"Default runtime_url: {config.runtime_url}")
print(f"Default document_id: {config.document_id}")
print(f"Default provider: {config.provider}")
# Test setting configuration
new_config = set_config(
runtime_url="http://localhost:9999",
document_id="test_notebooks.ipynb",
provider="datalayer",
runtime_token="test_token"
)
print(f"\nUpdated runtime_url: {new_config.runtime_url}")
print(f"Updated document_id: {new_config.document_id}")
print(f"Updated provider: {new_config.provider}")
print(f"Updated runtime_token: {'***' if new_config.runtime_token else 'None'}")
# Test that singleton works - getting config again should return same values
config2 = get_config()
print(f"\nSingleton test - runtime_url: {config2.runtime_url}")
print(f"Singleton test - document_id: {config2.document_id}")
# Test reset
reset_config()
config3 = get_config()
print(f"\nAfter reset - runtime_url: {config3.runtime_url}")
print(f"After reset - document_id: {config3.document_id}")
print(f"After reset - provider: {config3.provider}")
print("\n✅ Configuration system test completed successfully!")
def test_allowed_jupyter_mcp_tools_config():
"""Test the allowed_jupyter_mcp_tools configuration."""
reset_config()
# Test default configuration
config = get_config()
default_tools = config.get_allowed_jupyter_mcp_tools()
assert "notebook_run-all-cells" in default_tools
assert "notebook_get-selected-cell" in default_tools
print(f"Default allowed tools: {default_tools}")
# Test setting custom tools via set_config
new_config = set_config(allowed_jupyter_mcp_tools="custom_tool1,custom_tool2")
custom_tools = new_config.get_allowed_jupyter_mcp_tools()
assert custom_tools == ["custom_tool1", "custom_tool2"]
print(f"Custom tools: {custom_tools}")
# Test configuration via set_config (simulates how CLI sets the value)
set_config_result = set_config(allowed_jupyter_mcp_tools="env_tool1,env_tool2")
env_tools = set_config_result.get_allowed_jupyter_mcp_tools()
assert env_tools == ["env_tool1", "env_tool2"]
print(f"CLI-style configuration: {env_tools}")
# Reset to clean state
reset_config()
# Test comma-separated parsing with spaces
config_with_spaces = set_config(allowed_jupyter_mcp_tools=" tool1 , tool2 , tool3 ")
tools_with_spaces = config_with_spaces.get_allowed_jupyter_mcp_tools()
assert tools_with_spaces == ["tool1", "tool2", "tool3"]
print(f"Tools with spaces parsed: {tools_with_spaces}")
# Test empty entries filtering
config_empty = set_config(allowed_jupyter_mcp_tools="tool1,,tool2,")
tools_filtered = config_empty.get_allowed_jupyter_mcp_tools()
assert tools_filtered == ["tool1", "tool2"]
print(f"Empty entries filtered: {tools_filtered}")
print("✅ Allowed jupyter mcp tools configuration test completed successfully!")
def test_jupyter_extension_trait():
"""Test the Jupyter Server Extension trait configuration."""
from jupyter_mcp_server.jupyter_extension.extension import JupyterMCPServerExtensionApp
# Test default configuration
app = JupyterMCPServerExtensionApp()
assert hasattr(app, 'allowed_jupyter_mcp_tools')
assert app.allowed_jupyter_mcp_tools == "notebook_run-all-cells,notebook_get-selected-cell"
print(f"Extension default tools: {app.allowed_jupyter_mcp_tools}")
# Test custom configuration
app.allowed_jupyter_mcp_tools = "custom_ext_tool1,custom_ext_tool2"
assert app.allowed_jupyter_mcp_tools == "custom_ext_tool1,custom_ext_tool2"
print(f"Extension custom tools: {app.allowed_jupyter_mcp_tools}")
print("✅ Jupyter extension trait test completed successfully!")
if __name__ == "__main__":
test_config()
test_allowed_jupyter_mcp_tools_config()
test_jupyter_extension_trait()
