chore: fix dbt test flakiness

Author: eakmanrq

tests/dbt/conftest.py

@@ -127,3 +127,19 @@ def dbt_dummy_postgres_config() -> PostgresConfig:
         port=5432,
         schema="schema",
     )
+
+
+@pytest.fixture(scope="function", autouse=True)
+def reset_dbt_globals():
+    # This fixture is used to clear the memoized cache for _get_package_with_retries
+    # in dbt.clients.registry. This is necessary because the cache is shared across
+    # tests and can cause unexpected behavior if not cleared as some tests depend on
+    # the deprecation warning that _get_package_with_retries fires
+    yield
+    from dbt.clients.registry import _get_cached
+    from dbt_common.events.functions import reset_metadata_vars
+
+    # https://github.com/dbt-labs/dbt-core/blob/main/core/dbt/tests/util.py#L82
+    reset_metadata_vars()
+    # https://github.com/dbt-labs/dbt-core/blob/main/tests/functional/conftest.py#L9
+    _get_cached.cache = {}

tests/dbt/test_model.py

@@ -6,6 +6,7 @@
 from sqlglot import exp
 from sqlglot.errors import SchemaError
 from sqlmesh import Context
+from sqlmesh.core.console import NoopConsole, get_console
 from sqlmesh.core.model import TimeColumn, IncrementalByTimeRangeKind
 from sqlmesh.core.model.kind import OnDestructiveChange, OnAdditiveChange
 from sqlmesh.core.state_sync.db.snapshot import _snapshot_to_json
@@ -537,6 +538,7 @@ def test_load_deprecated_incremental_time_column(
         f.write(incremental_time_range_contents)
 
     snapshot_fqn = '"local"."main"."incremental_time_range"'
+    assert isinstance(get_console(), NoopConsole)
     context = Context(paths=project_dir)
     model = context.snapshots[snapshot_fqn].model
     # Validate model-level attributes