Base class get_context returns an empty dict

Author: LucasG0

backend/infrahub/database/__init__.py

@@ -183,9 +183,12 @@ def is_transaction(self) -> bool:
         return False
 
     def get_context(self) -> dict[str, Any]:
-        return {
-            "queries_names_to_config": self.queries_names_to_config,
-        }
+        """
+        This method is meant to be overridden by subclasses in order to fill in subclass attributes
+        to methods returning a copy of this object using self.__class__ constructor.
+        """
+
+        return {}
 
     def add_schema(self, schema: SchemaBranch, name: Optional[str] = None) -> None:
         self._schemas[name or schema.name] = schema
@@ -205,6 +208,7 @@ def start_session(self, read_only: bool = False, schemas: Optional[list[SchemaBr
             db_manager=self.manager,
             driver=self._driver,
             session_mode=session_mode,
+            queries_names_to_config=self.queries_names_to_config,
             **context,
         )
 
@@ -219,6 +223,7 @@ def start_transaction(self, schemas: Optional[list[SchemaBranch]] = None) -> Inf
             driver=self._driver,
             session=self._session,
             session_mode=self._session_mode,
+            queries_names_to_config=self.queries_names_to_config,
             **context,
         )
 

backend/tests/helpers/query_benchmark/db_query_profiler.py

@@ -171,6 +171,7 @@ class InfrahubDatabaseProfiler(InfrahubDatabase):
     def __init__(self, query_analyzer: QueryAnalyzer, **kwargs: Any) -> None:
         super().__init__(**kwargs)
         self.query_analyzer = query_analyzer
+        # Note that any attribute added here should be added to get_context method.
 
     def get_context(self) -> dict[str, Any]:
         ctx = super().get_context()