217 lines
2.9 KiB
Python
217 lines
2.9 KiB
Python
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() |