Initial commit

tests/test_config.py

@@ -0,0 +1,207 @@
+"""
+Unit tests for configuration management
+"""
+import pytest
+from pathlib import Path
+from drt.config.models import (
+    DatabaseConnection,
+    DatabasePair,
+    ComparisonSettings,
+    RowCountSettings,
+    SchemaSettings,
+    AggregateSettings,
+    ReportingSettings,
+    LoggingSettings,
+    Config
+)
+
+
+class TestDatabaseConnection:
+    """Test DatabaseConnection model"""
+    
+    def test_database_connection_minimal(self):
+        """Test creating a minimal database connection"""
+        conn = DatabaseConnection(
+            server="SQLSERVER01",
+            database="TestDB"
+        )
+        assert conn.server == "SQLSERVER01"
+        assert conn.database == "TestDB"
+        assert conn.timeout.connection == 30
+        assert conn.timeout.query == 300
+    
+    def test_database_connection_with_timeout(self):
+        """Test database connection with custom timeout"""
+        conn = DatabaseConnection(
+            server="SQLSERVER01",
+            database="TestDB",
+            timeout={"connection": 60, "query": 600}
+        )
+        assert conn.timeout.connection == 60
+        assert conn.timeout.query == 600
+
+
+class TestDatabasePair:
+    """Test DatabasePair model"""
+    
+    def test_database_pair_creation(self):
+        """Test creating a database pair"""
+        pair = DatabasePair(
+            name="Test_Pair",
+            enabled=True,
+            baseline=DatabaseConnection(
+                server="SQLSERVER01",
+                database="PROD_DB"
+            ),
+            target=DatabaseConnection(
+                server="SQLSERVER01",
+                database="TEST_DB"
+            )
+        )
+        assert pair.name == "Test_Pair"
+        assert pair.enabled is True
+        assert pair.baseline.database == "PROD_DB"
+        assert pair.target.database == "TEST_DB"
+
+
+class TestComparisonSettings:
+    """Test ComparisonSettings model"""
+    
+    def test_comparison_settings_health_check(self):
+        """Test health check mode settings"""
+        settings = ComparisonSettings(
+            mode="health_check",
+            row_count=RowCountSettings(enabled=True, tolerance_percent=0.0),
+            schema=SchemaSettings(
+                enabled=True,
+                checks={
+                    "column_names": True,
+                    "data_types": True
+                }
+            ),
+            aggregates=AggregateSettings(enabled=False)
+        )
+        assert settings.mode == "health_check"
+        assert settings.row_count.enabled is True
+        assert settings.aggregates.enabled is False
+    
+    def test_comparison_settings_full_mode(self):
+        """Test full mode settings"""
+        settings = ComparisonSettings(
+            mode="full",
+            row_count=RowCountSettings(enabled=True, tolerance_percent=0.0),
+            schema=SchemaSettings(enabled=True),
+            aggregates=AggregateSettings(enabled=True, tolerance_percent=0.01)
+        )
+        assert settings.mode == "full"
+        assert settings.aggregates.enabled is True
+        assert settings.aggregates.tolerance_percent == 0.01
+
+
+class TestReportingSettings:
+    """Test ReportingSettings model"""
+    
+    def test_reporting_settings_defaults(self):
+        """Test default reporting settings"""
+        settings = ReportingSettings()
+        assert settings.output_dir == "./reports"
+        assert settings.formats.html is True
+        assert settings.formats.csv is True
+        assert settings.formats.pdf is False
+        assert settings.include_timestamp is True
+    
+    def test_reporting_settings_custom(self):
+        """Test custom reporting settings"""
+        settings = ReportingSettings(
+            output_dir="./custom_reports",
+            filename_prefix="custom_test",
+            formats={"html": True, "csv": False, "pdf": True}
+        )
+        assert settings.output_dir == "./custom_reports"
+        assert settings.filename_prefix == "custom_test"
+        assert settings.formats.pdf is True
+
+
+class TestLoggingSettings:
+    """Test LoggingSettings model"""
+    
+    def test_logging_settings_defaults(self):
+        """Test default logging settings"""
+        settings = LoggingSettings()
+        assert settings.level == "INFO"
+        assert settings.output_dir == "./logs"
+        assert settings.console.enabled is True
+        assert settings.file.enabled is True
+    
+    def test_logging_settings_custom(self):
+        """Test custom logging settings"""
+        settings = LoggingSettings(
+            level="DEBUG",
+            console={"enabled": True, "level": "WARNING"}
+        )
+        assert settings.level == "DEBUG"
+        assert settings.console.level == "WARNING"
+
+
+class TestConfig:
+    """Test Config model"""
+    
+    def test_config_minimal(self):
+        """Test creating a minimal config"""
+        config = Config(
+            database_pairs=[
+                DatabasePair(
+                    name="Test",
+                    enabled=True,
+                    baseline=DatabaseConnection(
+                        server="SERVER01",
+                        database="PROD"
+                    ),
+                    target=DatabaseConnection(
+                        server="SERVER01",
+                        database="TEST"
+                    )
+                )
+            ],
+            comparison=ComparisonSettings(
+                mode="health_check",
+                row_count=RowCountSettings(enabled=True),
+                schema=SchemaSettings(enabled=True),
+                aggregates=AggregateSettings(enabled=False)
+            ),
+            tables=[]
+        )
+        assert len(config.database_pairs) == 1
+        assert config.comparison.mode == "health_check"
+        assert len(config.tables) == 0
+    
+    def test_config_with_tables(self):
+        """Test config with table definitions"""
+        from drt.models.table import TableInfo
+        
+        config = Config(
+            database_pairs=[
+                DatabasePair(
+                    name="Test",
+                    enabled=True,
+                    baseline=DatabaseConnection(server="S1", database="D1"),
+                    target=DatabaseConnection(server="S1", database="D2")
+                )
+            ],
+            comparison=ComparisonSettings(
+                mode="health_check",
+                row_count=RowCountSettings(enabled=True),
+                schema=SchemaSettings(enabled=True),
+                aggregates=AggregateSettings(enabled=False)
+            ),
+            tables=[
+                TableInfo(
+                    schema="dbo",
+                    name="TestTable",
+                    enabled=True,
+                    expected_in_target=True
+                )
+            ]
+        )
+        assert len(config.tables) == 1
+        assert config.tables[0].name == "TestTable"