mcp/tests/test_scoped_clients.py

import unittest

from services.scoped_api_client import ScopedApiClient


class FakeTransport:
    def __init__(self):
        self.calls = []

    def post_json(self, url, payload, headers, timeout):
        self.calls.append((url, payload, headers, timeout))
        return {'code': 'MCP_0000', 'data': {'ok': True}}


class ScopedApiClientTest(unittest.TestCase):
    def test_call_tool_uses_explicit_token_not_global_store(self):
        transport = FakeTransport()
        client = ScopedApiClient('https://tools.example.com', transport=transport, timeout=7)

        response = client.call_tool(
            token='MT_employee_a',
            tool_code='query_order',
            route_path='/mcp/tools/queryOrder',
            payload={'keyword': 'USC'},
            request_id='rq_public_1',
        )

        self.assertEqual('MCP_0000', response['code'])
        url, payload, headers, timeout = transport.calls[0]
        self.assertEqual('https://tools.example.com/mcp/tools/queryOrder', url)
        self.assertEqual('Bearer MT_employee_a', headers['Authorization'])
        self.assertEqual('query_order', headers['X-MCP-Tool-Code'])
        self.assertEqual('rq_public_1', headers['X-Request-Id'])
        self.assertEqual(7, timeout)


if __name__ == '__main__':
    unittest.main()