test_trace_verifier.py 1005 B

1234567891011121314151617181920212223242526
  1. from __future__ import annotations
  2. import pytest
  3. from agent import FileSystemTraceStore
  4. from script_build_host.domain.errors import InvalidConfiguration
  5. from script_build_host.infrastructure.trace_verifier import FileSystemTraceStoreVerifier
  6. @pytest.mark.asyncio
  7. async def test_filesystem_trace_verifier_reopens_message_event_and_attachment(tmp_path) -> None:
  8. root = tmp_path / "traces"
  9. verifier = FileSystemTraceStoreVerifier()
  10. await verifier.verify(FileSystemTraceStore(str(root)), require_attachments=True)
  11. probe_directories = list(root.glob("script-build-startup-probe-*"))
  12. assert len(probe_directories) == 1
  13. assert (probe_directories[0] / "meta.json").is_file()
  14. assert list((probe_directories[0] / "attachments").rglob("*.bin"))
  15. @pytest.mark.asyncio
  16. async def test_filesystem_trace_verifier_rejects_an_external_store() -> None:
  17. with pytest.raises(InvalidConfiguration, match="another store type"):
  18. await FileSystemTraceStoreVerifier().verify(object())