def test_auth_flow(self):
        """Ensures that the auth_flow signs the request correctly."""
        request = httpx.Request("POST", "https://example.com")
        prepared_request_mock = Mock()
        prepared_request_mock.headers = {"Authorization": "Signed"}
        with patch("requests.Request") as mock_requests_request:
            mock_requests_request.return_value = Mock()
            mock_requests_request.return_value.prepare.return_value = (
                prepared_request_mock
            )
            self.signer_mock.do_request_sign = Mock()

            list(self.oci_auth.auth_flow(request))

            self.signer_mock.do_request_sign.assert_called()
            assert request.headers.get("Authorization") == "Signed"
Ensures that the auth_flow signs the request correctly.
llama-index-integrations/llms/llama-index-llms-oci-data-science/tests/test_oci_data_science_client.py