Method test_validate_headers

tests/test_client.py:448–524

(self)

        test_client2.close()

    def test_validate_headers(self) -> None:
        client = OpenAI(
            base_url=base_url, api_key=api_key, admin_api_key=admin_api_key, _strict_response_validation=True
        )
        options = client._prepare_options(FinalRequestOptions(method="get", url="/foo"))
        request = client._build_request(options)

        assert request.headers.get("Authorization") == f"Bearer {api_key}"

        admin_request = client._build_request(
            FinalRequestOptions(
                method="get",
                url="/organization/projects",
                security={"admin_api_key_auth": True},
            )
        )
        assert admin_request.headers.get("Authorization") == f"Bearer {admin_api_key}"

        with update_env(**{"OPENAI_API_KEY": Omit()}):
            admin_only = OpenAI(
                base_url=base_url,
                api_key=None,
                admin_api_key=admin_api_key,
                _strict_response_validation=True,
            )
            admin_only_request = admin_only._build_request(
                FinalRequestOptions(
                    method="get",
                    url="/organization/projects",
                    security={"admin_api_key_auth": True},
                )
            )
            assert admin_only_request.headers.get("Authorization") == f"Bearer {admin_api_key}"

            with pytest.raises(
                TypeError,
                match="Could not resolve authentication method",
            ):
                admin_only._build_request(
                    FinalRequestOptions(
                        method="post",
                        url="/responses",
                        security={"bearer_auth": True},
                    )
                )

        with update_env(
            **{
                "OPENAI_API_KEY": Omit(),
                "OPENAI_ADMIN_KEY": Omit(),
            }
        ):
            no_credentials = OpenAI(
                base_url=base_url,
                api_key=None,
                admin_api_key=None,
                _enforce_credentials=False,
                _strict_response_validation=True,