fix unit tests

Author: darynaishchenko

airbyte_cdk/sources/declarative/parsers/model_to_component_factory.py

@@ -3146,6 +3146,24 @@ def create_legacy_session_token_authenticator(
             parameters=model.parameters or {},
         )
 
+    def _get_log_formatter(
+            self, log_formatter: Callable[[Response], Any] | None, name: str
+    ) -> Callable[[Response], Any] | None:
+        if self._should_limit_slices_fetched():
+            return (
+                (
+                    lambda response: format_http_message(
+                        response,
+                        f"Stream '{name}' request",
+                        f"Request performed in order to extract records for stream '{name}'",
+                        name,
+                    )
+                )
+                if not log_formatter
+                else log_formatter
+            )
+        return None
+
     def _should_limit_slices_fetched(self) -> bool:
         """
         Returns True if the number of slices fetched should be limited, False otherwise.

airbyte_cdk/sources/declarative/retrievers/simple_retriever.py

@@ -132,10 +132,9 @@ class SimpleRetriever(Retriever, ComponentConstructor[SimpleRetrieverModel]):
     ignore_stream_slicer_parameters_on_paginated_requests: bool = False
     additional_query_properties: Optional[QueryProperties] = None
     log_formatter: Optional[Callable[[requests.Response], Any]] = None
-    _should_use_lazy_simple_retriever: bool = False
 
     @classmethod
-    def _validate_if_lazy_simple_retriever_should_be_applied(
+    def _should_use_lazy_simple_retriever(
         cls,
         name: str,
         model: SimpleRetrieverModel,
@@ -170,7 +169,6 @@ def _validate_if_lazy_simple_retriever_should_be_applied(
                 raise ValueError(
                     f"LazySimpleRetriever only supports JsonDecoder. Found: {model.decoder.type}."
                 )
-            cls._should_use_lazy_simple_retriever = True
             return True
 
         return False
@@ -312,17 +310,13 @@ def _get_log_formatter(
             query_properties = dependency_constructor(
                 model=query_properties_definition,
                 config=config,
-                dependency_constructor=dependency_constructor,
-                additional_flags=additional_flags,
                 **kwargs,
             )
 
         elif hasattr(model.requester, "query_properties") and model.requester.query_properties:
             query_properties = dependency_constructor(
                 model=model.requester.query_properties,
                 config=config,
-                dependency_constructor=dependency_constructor,
-                additional_flags=additional_flags,
                 **kwargs,
             )
 
@@ -392,7 +386,7 @@ def _get_log_formatter(
             "parameters": model.parameters or {},
         }
 
-        if cls._validate_if_lazy_simple_retriever_should_be_applied(
+        if cls._should_use_lazy_simple_retriever(
             name, model, additional_flags, incremental_sync
         ):
             return resolved_dependencies
@@ -427,8 +421,13 @@ def build(
             additional_flags=additional_flags,
             **kwargs,
         )
-        if cls._should_use_lazy_simple_retriever:
-            LazySimpleRetriever(**resolved_dependencies)
+        if cls._should_use_lazy_simple_retriever(
+                name=kwargs.get("name"),
+                model=model,
+                additional_flags=additional_flags,
+                incremental_sync=kwargs.get("incremental_sync"),
+        ):
+            return LazySimpleRetriever(**resolved_dependencies)
         return SimpleRetriever(**resolved_dependencies)
 
     def __post_init__(self, parameters: Mapping[str, Any]) -> None: