rev 4

2026-04-07 13:42:45 +03:00
parent 7d2f444e1c
commit 259d3da89f
34 changed files with 2130 additions and 788 deletions
@@ -0,0 +1,137 @@
+from __future__ import annotations
+
+from types import SimpleNamespace
+from unittest.mock import AsyncMock, MagicMock, patch
+
+import pytest
+
+from cpv3.modules.media.service import apply_silence_cuts, remove_silence
+
+
+class _TempDownload:
+    def __init__(self, path: str) -> None:
+        self.path = path
+
+    def cleanup(self) -> None:
+        return None
+
+
+class _FakeProcess:
+    returncode = 0
+
+    async def communicate(self) -> tuple[bytes, bytes]:
+        return b"", b""
+
+
+@pytest.mark.asyncio
+async def test_remove_silence_uses_single_input_trim_filter(tmp_path) -> None:
+    input_path = tmp_path / "input.mp4"
+    input_path.write_bytes(b"video")
+
+    file_info = SimpleNamespace(
+        file_path="uploads/test-file.txt",
+        file_url="http://example.com/uploads/test-file.txt",
+        file_size=1024,
+        filename="test-file.txt",
+    )
+    storage = SimpleNamespace(
+        download_to_temp=AsyncMock(return_value=_TempDownload(str(input_path))),
+        upload_fileobj=AsyncMock(return_value="uploads/test-file.txt"),
+        get_file_info=AsyncMock(return_value=file_info),
+    )
+
+    captured: dict[str, tuple[str, ...]] = {}
+
+    async def fake_create_subprocess_exec(
+        *cmd: str, stdout=None, stderr=None
+    ) -> _FakeProcess:
+        captured["cmd"] = cmd
+        return _FakeProcess()
+
+    with (
+        patch(
+            "cpv3.modules.media.service._compute_non_silent_segments",
+            return_value=[(0, 1000), (2000, 3000)],
+        ),
+        patch(
+            "cpv3.modules.media.service.asyncio.create_subprocess_exec",
+            side_effect=fake_create_subprocess_exec,
+        ),
+    ):
+        await remove_silence(
+            storage,
+            file_key="uploads/input.mp4",
+            out_folder="processed",
+        )
+
+    cmd = list(captured["cmd"])
+    assert cmd.count("-i") == 1
+
+    filter_complex = cmd[cmd.index("-filter_complex") + 1]
+    assert "trim=start=0.000:end=1.000" in filter_complex
+    assert "atrim=start=0.000:end=1.000" in filter_complex
+    assert "trim=start=2.000:end=3.000" in filter_complex
+    assert "atrim=start=2.000:end=3.000" in filter_complex
+
+
+async def _run_sync_immediately(func):
+    return func()
+
+
+@pytest.mark.asyncio
+async def test_apply_silence_cuts_uses_single_input_trim_filter(tmp_path) -> None:
+    input_path = tmp_path / "input.mp4"
+    input_path.write_bytes(b"video")
+
+    file_info = SimpleNamespace(
+        file_path="uploads/test-file.txt",
+        file_url="http://example.com/uploads/test-file.txt",
+        file_size=1024,
+        filename="test-file.txt",
+    )
+    storage = SimpleNamespace(
+        download_to_temp=AsyncMock(return_value=_TempDownload(str(input_path))),
+        upload_fileobj=AsyncMock(return_value="uploads/test-file.txt"),
+        get_file_info=AsyncMock(return_value=file_info),
+    )
+
+    captured: dict[str, tuple[str, ...]] = {}
+
+    async def fake_create_subprocess_exec(
+        *cmd: str, stdout=None, stderr=None
+    ) -> _FakeProcess:
+        captured["cmd"] = cmd
+        return _FakeProcess()
+
+    fake_audio = MagicMock()
+    fake_audio.__len__.return_value = 5000
+
+    with (
+        patch(
+            "cpv3.modules.media.service.anyio.to_thread.run_sync",
+            side_effect=_run_sync_immediately,
+        ),
+        patch("pydub.AudioSegment.from_file", return_value=fake_audio),
+        patch(
+            "cpv3.modules.media.service.asyncio.create_subprocess_exec",
+            side_effect=fake_create_subprocess_exec,
+        ),
+    ):
+        await apply_silence_cuts(
+            storage,
+            file_key="uploads/input.mp4",
+            out_folder="processed",
+            cuts=[
+                {"start_ms": 1000, "end_ms": 2000},
+                {"start_ms": 3000, "end_ms": 3500},
+            ],
+        )
+
+    cmd = list(captured["cmd"])
+    assert cmd.count("-i") == 1
+
+    filter_complex = cmd[cmd.index("-filter_complex") + 1]
+    assert "trim=start=0.000:end=1.000" in filter_complex
+    assert "atrim=start=0.000:end=1.000" in filter_complex
+    assert "trim=start=2.000:end=3.000" in filter_complex
+    assert "atrim=start=3.500:end=5.000" in filter_complex