ha-supervisor/tests/test_main.py

18 lines
394 B
Python

"""Testing handling with main."""
from pathlib import Path
import supervisor.__main__ as main
def test_write_state(tmp_path):
"""Test startup-marker file cleanup."""
test_file = Path(tmp_path, "test.file")
test_file.touch()
assert test_file.exists()
main.CONTAINER_OS_STARTUP_CHECK = test_file
main.run_os_startup_check_cleanup()
assert not test_file.exists()