67 lines
2.5 KiB
Python
67 lines
2.5 KiB
Python
|
"""Tests for stub_writer (binary file -> .md frontmatter stub)."""
|
||
|
|
|
||
|
|
import tempfile
|
||
|
|
import unittest
|
||
|
|
from pathlib import Path
|
||
|
|
|
||
|
|
from src.ingest.stub_writer import write_stub, _infer_kind
|
||
|
|
|
||
|
|
|
||
|
|
class StubWriterTestCase(unittest.TestCase):
|
||
|
|
def test_infer_kind_video(self):
|
||
|
|
self.assertEqual(_infer_kind(Path("meeting.mp4")), "video")
|
||
|
|
self.assertEqual(_infer_kind(Path("recording.MKV")), "video")
|
||
|
|
|
||
|
|
def test_infer_kind_audio(self):
|
||
|
|
self.assertEqual(_infer_kind(Path("track.wav")), "audio")
|
||
|
|
|
||
|
|
def test_infer_kind_image(self):
|
||
|
|
self.assertEqual(_infer_kind(Path("photo.png")), "image")
|
||
|
|
|
||
|
|
def test_infer_kind_archive(self):
|
||
|
|
self.assertEqual(_infer_kind(Path("backup.zip")), "archive")
|
||
|
|
|
||
|
|
def test_infer_kind_other(self):
|
||
|
|
self.assertEqual(_infer_kind(Path("unknown.xyz")), "other")
|
||
|
|
|
||
|
|
def test_write_stub_creates_md(self):
|
||
|
|
with tempfile.TemporaryDirectory() as tmp:
|
||
|
|
mp4 = Path(tmp) / "video.mp4"
|
||
|
|
mp4.write_bytes(b"fake-mp4-content")
|
||
|
|
stub = write_stub(mp4, project="2026")
|
||
|
|
self.assertEqual(stub, mp4.with_suffix(".mp4.md"))
|
||
|
|
self.assertTrue(stub.exists())
|
||
|
|
text = stub.read_text(encoding="utf-8")
|
||
|
|
self.assertIn("source: video.mp4", text)
|
||
|
|
self.assertIn("kind: video", text)
|
||
|
|
self.assertIn("project: 2026", text)
|
||
|
|
self.assertIn("size: 16", text)
|
||
|
|
self.assertIn("Бинарный файл", text)
|
||
|
|
|
||
|
|
def test_write_stub_overwrites(self):
|
||
|
|
with tempfile.TemporaryDirectory() as tmp:
|
||
|
|
mp4 = Path(tmp) / "video.mp4"
|
||
|
|
mp4.write_bytes(b"a" * 10)
|
||
|
|
stub = write_stub(mp4, project="p1")
|
||
|
|
stub.write_text("OLD", encoding="utf-8")
|
||
|
|
stub2 = write_stub(mp4, project="p1")
|
||
|
|
self.assertEqual(stub, stub2)
|
||
|
|
self.assertNotIn("OLD", stub.read_text(encoding="utf-8"))
|
||
|
|
|
||
|
|
def test_write_stub_missing_file_raises(self):
|
||
|
|
with tempfile.TemporaryDirectory() as tmp:
|
||
|
|
missing = Path(tmp) / "nope.mp4"
|
||
|
|
with self.assertRaises(FileNotFoundError):
|
||
|
|
write_stub(missing, project="x")
|
||
|
|
|
||
|
|
def test_explicit_kind_overrides_inference(self):
|
||
|
|
with tempfile.TemporaryDirectory() as tmp:
|
||
|
|
f = Path(tmp) / "weird.bin"
|
||
|
|
f.write_bytes(b"x")
|
||
|
|
stub = write_stub(f, project="p", kind="custom")
|
||
|
|
self.assertIn("kind: custom", stub.read_text(encoding="utf-8"))
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
unittest.main()
|