54 lines
1.7 KiB
Python
54 lines
1.7 KiB
Python
 
|
from __future__ import annotations
|
||
|
|
|
||
|
|
import os
|
||
|
|
import threading
|
||
|
|
import time
|
||
|
|
from pathlib import Path
|
||
|
|
|
||
|
|
import pytest
|
||
|
|
|
||
|
|
from packetstation import cli, worker
|
||
|
|
|
||
|
|
|
||
|
|
def test_version(capsys: pytest.CaptureFixture[str], monkeypatch: pytest.MonkeyPatch) -> None:
|
||
|
|
monkeypatch.setenv("PS_BUILD", "abc123")
|
||
|
|
assert cli.main(["version"]) == 0
|
||
|
|
assert "build abc123" in capsys.readouterr().out
|
||
|
|
|
||
|
|
|
||
|
|
def test_worker_healthcheck(tmp_path: Path, monkeypatch: pytest.MonkeyPatch) -> None:
|
||
|
|
beat = tmp_path / "beat"
|
||
|
|
monkeypatch.setenv("WORKER_BEAT_FILE", str(beat))
|
||
|
|
assert cli.main(["healthcheck", "worker"]) == 1
|
||
|
|
worker.touch(beat)
|
||
|
|
assert cli.main(["healthcheck", "worker"]) == 0
|
||
|
|
old = time.time() - 120
|
||
|
|
os.utime(beat, (old, old))
|
||
|
|
assert cli.main(["healthcheck", "worker"]) == 1
|
||
|
|
|
||
|
|
|
||
|
|
def test_app_healthcheck_down() -> None:
|
||
|
|
# Nothing listens on 8000 in tests (or something unrelated does): either way not a crash
|
||
|
|
assert cli.main(["healthcheck", "app"]) in (0, 1)
|
||
|
|
|
||
|
|
|
||
|
|
def test_worker_loop_touches_and_stops(tmp_path: Path) -> None:
|
||
|
|
beat = tmp_path / "sub" / "beat"
|
||
|
|
stop = threading.Event()
|
||
|
|
t = threading.Thread(target=worker.run, args=(beat, "t"), kwargs={"stop": stop, "interval": 0.05})
|
||
|
|
t.start()
|
||
|
|
deadline = time.time() + 5
|
||
|
|
while not beat.exists() and time.time() < deadline:
|
||
|
|
time.sleep(0.01)
|
||
|
|
stop.set()
|
||
|
|
t.join(5)
|
||
|
|
assert beat.exists() and not t.is_alive()
|
||
|
|
|
||
|
|
|
||
|
|
def test_migrate_reports_bad_secret(
|
||
|
|
tmp_path: Path, monkeypatch: pytest.MonkeyPatch, capsys: pytest.CaptureFixture[str]
|
||
|
|
) -> None:
|
||
|
|
monkeypatch.setenv("PG_MIGRATE_PASSWORD_FILE", str(tmp_path / "missing"))
|
||
|
|
assert cli.main(["migrate"]) == 2
|
||
|
|
assert "PG_MIGRATE_PASSWORD_FILE" in capsys.readouterr().err
|