Class TestAzureRouting

tests/extraction/llm/test_openai_provider.py:667–769

Azure-prefixed models should route through the OpenAI-compatible provider.

class TestAzureRouting(unittest.TestCase):
    """Azure-prefixed models should route through the OpenAI-compatible provider."""

    @patch.dict(
        "os.environ",
        {
            "AZURE_OPENAI_API_KEY": "azure-key",
            "AZURE_OPENAI_ENDPOINT": "https://example.openai.azure.com",
        },
    )
    @patch("explainshell.extraction.llm.providers.openai.OpenAI")
    def test_factory_routes_azure_models_to_openai_provider(
        self, _mock_openai_cls: MagicMock
    ) -> None:
        self.assertIsInstance(make_provider("azure/my-deployment"), OpenAIProvider)
        self.assertIsInstance(
            make_batch_provider("azure/my-deployment"), OpenAIProvider
        )

    @patch.dict(
        "os.environ",
        {
            "AZURE_OPENAI_API_KEY": "azure-key",
            "AZURE_OPENAI_ENDPOINT": "https://example.openai.azure.com",
        },
        clear=True,
    )
    @patch("explainshell.extraction.llm.providers.openai.OpenAI")
    def test_call_uses_azure_endpoint_configuration(
        self, mock_openai_cls: MagicMock
    ) -> None:
        response = SimpleNamespace(output_text='{"options":[]}', usage=None)
        client = MagicMock()
        client.responses.create.return_value = response
        mock_openai_cls.return_value = client

        provider = OpenAIProvider("azure/my-deployment")
        text, usage = provider.call("man page text")

        self.assertEqual(text, '{"options":[]}')
        self.assertEqual(usage.input_tokens, 0)
        mock_openai_cls.assert_called_once_with(
            api_key="azure-key",
            base_url="https://example.openai.azure.com/openai/v1/",
            timeout=LLM_TIMEOUT_SECONDS,
        )
        client.responses.create.assert_called_once()
        self.assertEqual(
            client.responses.create.call_args.kwargs["model"], "my-deployment"
        )

    @patch.dict(
        "os.environ",
        {
            "AZURE_OPENAI_API_KEY": "azure-key",
            "AZURE_OPENAI_BASE_URL": "https://example.openai.azure.com/openai/v1",
        },
        clear=True,