stack/tests/aco/test_lake_quality.py

"""Tests for aco.lake.quality — Lakehouse Monitoring setup."""

from __future__ import annotations

from unittest.mock import MagicMock

from aco.lake.quality import MonitorProfile, MonitorResult, setup_monitors


class TestMonitorModels:
    def test_profile_defaults(self):
        p = MonitorProfile(table="aco.core")
        assert p.output_schema == "_monitoring"
        assert p.slicing_exprs == []

    def test_result_model(self):
        r = MonitorResult(table="aco.core.encounter", status="active")
        assert r.table == "aco.core.encounter"
        assert r.dashboard_id is None


class TestSetupMonitors:
    def test_dry_run(self):
        client = MagicMock()
        table = MagicMock()
        table.full_name = "aco.core.encounter"
        table.name = "encounter"
        client._ws.tables.list.return_value = [table]

        actions = setup_monitors(client, catalog="aco", schemas=["core"], dry_run=True)
        assert len(actions) == 1
        assert "[dry-run]" in actions[0]
        assert "aco.core.encounter" in actions[0]
        # Should not call create in dry_run
        client._ws.quality_monitors.create.assert_not_called()

    def test_creates_output_schema(self):
        client = MagicMock()
        client._ws.schemas.get.side_effect = Exception("not found")
        client._ws.tables.list.return_value = []

        setup_monitors(client, catalog="aco", schemas=["core"], dry_run=False)
        client._ws.schemas.create.assert_called_once()

    def test_skips_existing_monitor(self):
        client = MagicMock()
        client._ws.schemas.get.return_value = MagicMock()
        table = MagicMock()
        table.full_name = "aco.core.encounter"
        table.name = "encounter"
        client._ws.tables.list.return_value = [table]
        # Monitor already exists
        client._ws.quality_monitors.get.return_value = MagicMock()

        actions = setup_monitors(client, catalog="aco", schemas=["core"], dry_run=False)
        assert any("[exists]" in a for a in actions)
        client._ws.quality_monitors.create.assert_not_called()