Chore: Fix usage of cached properties in the Context

Author: izeigerman

sqlmesh/core/context.py

@@ -367,11 +367,7 @@ def __init__(
         self._requirements: t.Dict[str, str] = {}
         self._environment_statements: t.List[EnvironmentStatements] = []
         self._excluded_requirements: t.Set[str] = set()
-        self._default_catalog: t.Optional[str] = None
-        self._default_catalog_per_gateway: t.Optional[t.Dict[str, str]] = None
         self._engine_adapter: t.Optional[EngineAdapter] = None
-        self._connection_config: t.Optional[ConnectionConfig] = None
-        self._test_connection_config: t.Optional[ConnectionConfig] = None
         self._linters: t.Dict[str, Linter] = {}
         self._loaded: bool = False
 
@@ -967,11 +963,9 @@ def requirements(self) -> t.Dict[str, str]:
         """Returns the Python dependencies of the project loaded in this context."""
         return self._requirements.copy()
 
-    @property
+    @cached_property
     def default_catalog(self) -> t.Optional[str]:
-        if self._default_catalog is None and self.default_catalog_per_gateway:
-            self._default_catalog = self.default_catalog_per_gateway[self.selected_gateway]
-        return self._default_catalog
+        return self.default_catalog_per_gateway.get(self.selected_gateway)
 
     @python_api_analytics
     def render(
@@ -2516,33 +2510,25 @@ def engine_adapters(self) -> t.Dict[str, EngineAdapter]:
     @cached_property
     def default_catalog_per_gateway(self) -> t.Dict[str, str]:
         """Returns the default catalogs for each engine adapter."""
-        if self._default_catalog_per_gateway is None:
-            self._default_catalog_per_gateway = self._scheduler.get_default_catalog_per_gateway(
-                self
-            )
-        return self._default_catalog_per_gateway
+        return self._scheduler.get_default_catalog_per_gateway(self)
 
-    @cached_property
+    @property
     def concurrent_tasks(self) -> int:
         if self._concurrent_tasks is None:
             self._concurrent_tasks = self.connection_config.concurrent_tasks
         return self._concurrent_tasks
 
     @cached_property
     def connection_config(self) -> ConnectionConfig:
-        if self._connection_config is None:
-            self._connection_config = self.config.get_connection(self.selected_gateway)
-        return self._connection_config
+        return self.config.get_connection(self.selected_gateway)
 
     @cached_property
     def test_connection_config(self) -> ConnectionConfig:
-        if self._test_connection_config is None:
-            self._test_connection_config = self.config.get_test_connection(
-                self.gateway,
-                self.default_catalog,
-                default_catalog_dialect=self.config.dialect,
-            )
-        return self._test_connection_config
+        return self.config.get_test_connection(
+            self.gateway,
+            self.default_catalog,
+            default_catalog_dialect=self.config.dialect,
+        )
 
     @cached_property
     def environment_catalog_mapping(self) -> RegexKeyDict: