Refactor test environment setup by introducing setup_duckdb_env function for improved clarity and reusability

Author: luisggc

tests/cli_helpers.py

@@ -29,3 +29,9 @@ def set_cli_env(
     monkeypatch.setenv("SQLCOMPARE_CONN_DEFAULT", connection_name)
     monkeypatch.setenv(f"SQLCOMPARE_CONN_{connection_name.upper()}", connection_url)
     monkeypatch.setenv("SQLCOMPARE_COMPARISON_SCHEMA", schema)
+
+
+def setup_duckdb_env(monkeypatch, config_dir: Path, schema: str = "sqlcompare") -> None:
+    """Setup environment for DuckDB tests without a specific connection."""
+    monkeypatch.setenv("SQLCOMPARE_CONFIG_DIR", str(config_dir))
+    monkeypatch.setenv("SQLCOMPARE_COMPARISON_SCHEMA", schema)

tests/test_cli_dataset.py

@@ -9,6 +9,7 @@
 from sqlcompare.cli import app
 from sqlcompare.config import load_test_runs
 from sqlcompare.db import DBConnection
+from tests.cli_helpers import setup_duckdb_env
 
 
 def _seed_duckdb(db_path: Path) -> None:
@@ -124,7 +125,7 @@ def test_dataset_command_with_yaml_dataset(tmp_path: Path, monkeypatch) -> None:
 def test_dataset_command_file_name_without_connection(
     tmp_path: Path, monkeypatch
 ) -> None:
-    monkeypatch.setenv("SQLCOMPARE_CONFIG_DIR", str(tmp_path / "config"))
+    setup_duckdb_env(monkeypatch, tmp_path / "config")
 
     runner = CliRunner()
     dataset_path = Path(__file__).parent / "datasets" / "row_compare" / "dataset.yaml"

tests/test_cli_table.py

@@ -6,7 +6,7 @@
 
 from sqlcompare.cli import app
 from sqlcompare.config import load_test_runs
-from tests.cli_helpers import seed_duckdb, set_cli_env
+from tests.cli_helpers import seed_duckdb, set_cli_env, setup_duckdb_env
 
 
 def test_table_command_with_options(tmp_path, monkeypatch) -> None:
@@ -43,8 +43,7 @@ def test_table_command_with_options(tmp_path, monkeypatch) -> None:
 
 
 def test_table_command_with_files(tmp_path, monkeypatch) -> None:
-    config_dir = tmp_path / "config"
-    monkeypatch.setenv("SQLCOMPARE_CONFIG_DIR", str(config_dir))
+    setup_duckdb_env(monkeypatch, tmp_path / "config")
     runner = CliRunner()
 
     dataset_dir = Path(__file__).parent / "datasets" / "row_compare"