feat: small refactor

tests/test_helpers_automation.py

@@ -0,0 +1,64 @@
+"""Tests for data_platform.helpers.automation — apply_automation."""
+
+from unittest.mock import MagicMock
+
+from dagster import AssetSpec
+
+from data_platform.helpers.automation import (
+    AUTOMATION_CONDITION,
+    _is_manual,
+    apply_automation,
+)
+
+
+def _make_asset_def(tags: dict | None = None):
+    """Create a minimal mock AssetsDefinition with the given tags."""
+    spec = MagicMock(spec=AssetSpec)
+    spec.tags = tags or {}
+
+    asset_def = MagicMock()
+    asset_def.specs = [spec]
+
+    updated = MagicMock()
+    asset_def.with_attributes.return_value = updated
+
+    return asset_def, updated
+
+
+class TestIsManual:
+    def test_manual_when_tagged(self):
+        asset_def, _ = _make_asset_def(tags={"manual": "true"})
+        assert _is_manual(asset_def) is True
+
+    def test_not_manual_when_no_tags(self):
+        asset_def, _ = _make_asset_def(tags={})
+        assert _is_manual(asset_def) is False
+
+    def test_not_manual_when_other_tags(self):
+        asset_def, _ = _make_asset_def(tags={"owner": "team-data"})
+        assert _is_manual(asset_def) is False
+
+
+class TestApplyAutomation:
+    def test_manual_asset_unchanged(self):
+        asset_def, _ = _make_asset_def(tags={"manual": "true"})
+        result = apply_automation([asset_def])
+        assert result == [asset_def]
+        asset_def.with_attributes.assert_not_called()
+
+    def test_non_manual_asset_gets_condition(self):
+        asset_def, updated = _make_asset_def(tags={})
+        result = apply_automation([asset_def])
+        assert result == [updated]
+        asset_def.with_attributes.assert_called_once_with(
+            automation_condition=AUTOMATION_CONDITION
+        )
+
+    def test_empty_list(self):
+        assert apply_automation([]) == []
+
+    def test_mixed_assets(self):
+        manual, _ = _make_asset_def(tags={"manual": "true"})
+        auto, auto_updated = _make_asset_def(tags={})
+        result = apply_automation([manual, auto])
+        assert result == [manual, auto_updated]