Generalized MCP Server

import json import pytest from unittest import mock import grpc from src.server import MCPServicer from src.generated import mcp_pb2 @pytest.fixture def servicer(): return MCPServicer() def test_azure_get_token(servicer): """Test Azure token retrieval with mock.""" with mock.patch('azure.identity.DefaultAzureCredential') as mock_cred: mock_instance = mock_cred.return_value mock_token = mock.MagicMock() mock_token.token = "fake-token" mock_instance.get_token.return_value = mock_token req = mcp_pb2.InvokeRequest( fq_name="azure_identity.DefaultAzureCredential.get_token", json_args='{"args": ["https://management.azure.com/.default"]}' ) resp = servicer.Invoke(req, grpc.ServicerContext()) result = json.loads(resp.json_result) assert "fake-token" in str(result) def test_azure_credential_creation(servicer): """Test Azure credential creation with mock.""" with mock.patch('azure.identity.DefaultAzureCredential') as mock_cred: mock_instance = mock_cred.return_value # Test that the credential can be instantiated assert mock_instance is not None mock_cred.assert_called_once()