work: Добваил нагрев в автоматизацию

tests/test_schedule_auto_temperature_parser.py

@@ -0,0 +1,217 @@
+from pathlib import Path
+from tempfile import TemporaryDirectory
+
+from schedule import (
+    ScheduleActionType,
+    load_schedule,
+)
+
+
+VALID_YAML = """
+version: 1
+
+enabled: true
+timezone: local
+
+templates:
+
+  test:
+
+    - time: "00:00"
+      action:
+        type: auto
+        speed: 3
+        target_temp: 20
+
+days:
+  mon: test
+  tue: test
+  wed: test
+  thu: test
+  fri: test
+  sat: test
+  sun: test
+"""
+
+
+def load_config(text: str):
+
+    temp_dir = TemporaryDirectory()
+
+    path = (
+        Path(temp_dir.name)
+        / "schedule.yaml"
+    )
+
+    path.write_text(
+        text,
+        encoding="utf-8",
+    )
+
+    config = load_schedule(path)
+
+    return temp_dir, config
+
+
+def test_auto_target_temperature():
+
+    temp_dir, config = load_config(
+        VALID_YAML
+    )
+
+    try:
+        point = (
+            config.templates["test"][0]
+        )
+
+        action = point.action
+
+        assert (
+            action.type
+            == ScheduleActionType.AUTO
+        )
+
+        assert (
+            action.settings.speed
+            == 3
+        )
+
+        assert (
+            action.settings.target_temp
+            == 20
+        )
+
+    finally:
+        temp_dir.cleanup()
+
+
+def test_auto_rejects_heater():
+
+    config_text = """
+version: 1
+
+enabled: true
+timezone: local
+
+templates:
+
+  test:
+
+    - time: "00:00"
+      action:
+        type: auto
+        speed: 3
+        target_temp: 20
+        heater: on
+
+days:
+  mon: test
+  tue: test
+  wed: test
+  thu: test
+  fri: test
+  sat: test
+  sun: test
+"""
+
+    try:
+        temp_dir, _ = load_config(
+            config_text
+        )
+
+    except ValueError as exc:
+
+        print()
+        print(
+            "Expected error:",
+            exc,
+        )
+
+        assert "heater" in str(exc)
+
+        return
+
+    else:
+        temp_dir.cleanup()
+
+    raise AssertionError(
+        "AUTO heater field must "
+        "raise ValueError"
+    )
+
+
+def test_auto_speed_is_required():
+
+    config_text = """
+version: 1
+
+enabled: true
+timezone: local
+
+templates:
+
+  test:
+
+    - time: "00:00"
+      action:
+        type: auto
+        target_temp: 20
+
+days:
+  mon: test
+  tue: test
+  wed: test
+  thu: test
+  fri: test
+  sat: test
+  sun: test
+"""
+
+    try:
+        temp_dir, _ = load_config(
+            config_text
+        )
+
+    except ValueError as exc:
+
+        print()
+        print(
+            "Expected error:",
+            exc,
+        )
+
+        assert (
+            str(exc)
+            == "AUTO action requires 'speed'"
+        )
+
+        return
+
+    else:
+        temp_dir.cleanup()
+
+    raise AssertionError(
+        "AUTO without speed must "
+        "raise ValueError"
+    )
+
+
+def main():
+
+    test_auto_target_temperature()
+
+    test_auto_rejects_heater()
+
+    test_auto_speed_is_required()
+
+    print()
+    print("=" * 70)
+    print(
+        "ALL AUTO TEMPERATURE PARSER "
+        "TESTS PASSED"
+    )
+    print("=" * 70)
+
+
+if __name__ == "__main__":
+    main()