feat: implement Layer 2 per-file transit, RAM session, mock buffer, and tests (v0.2.0)

tests/test_app_integration.py

@@ -0,0 +1,115 @@
+import importlib.util
+import json
+from pathlib import Path
+import pytest
+
+app_path = Path(__file__).resolve().parent.parent / "site" / "app.py"
+spec = importlib.util.spec_from_file_location("site_app", app_path)
+site_app = importlib.util.module_from_spec(spec)
+spec.loader.exec_module(site_app)
+
+app = site_app.app
+_mock_storage = site_app._mock_storage
+_mock_storage_lock = site_app._mock_storage_lock
+
+from upload.backend.session import cleanup, get_files
+import httpx
+
+
+@pytest.fixture
+def client():
+    app.config["TESTING"] = True
+    app.config["UPLOAD_HTTPX_TRANSPORT"] = httpx.WSGITransport(app=app)
+    with app.test_client() as client:
+        yield client
+
+
+def test_health(client):
+    res = client.get("/health")
+    assert res.status_code == 200
+    assert res.data == b"ok"
+
+
+def test_index_page(client):
+    res = client.get("/")
+    assert res.status_code == 200
+    assert b"upload-btn" in res.data
+    assert b"file-picker" in res.data
+
+
+def test_mock_buffer_crud(client):
+    key = "test_item_1"
+    # PUT
+    res = client.put(f"/mock-buffer/{key}", data=b"binary_payload_123")
+    assert res.status_code == 201
+
+    # GET
+    res = client.get(f"/mock-buffer/{key}")
+    assert res.status_code == 200
+    assert res.data == b"binary_payload_123"
+
+    # Status
+    res = client.get("/mock-buffer/status")
+    assert res.status_code == 200
+    data = res.get_json()
+    assert data["count"] >= 1
+    assert any(item["key"] == key for item in data["items"])
+
+    # DELETE
+    res = client.delete(f"/mock-buffer/{key}")
+    assert res.status_code == 204
+
+    # GET after delete -> 404
+    res = client.get(f"/mock-buffer/{key}")
+    assert res.status_code == 404
+
+
+def test_full_transit_flow_mock(client):
+    """Тестирует пофайловый транзит через mock-буфер и приём в сессию RAM."""
+    with _mock_storage_lock:
+        _mock_storage.clear()
+
+    # Шаг 1: Браузер кладёт файл в mock-буфер
+    key = "token_uuid_0"
+    file_bytes = b"%PDF-1.4 test document content for transit"
+    res = client.put(f"/mock-buffer/{key}", data=file_bytes)
+    assert res.status_code == 201
+
+    # Проверяем, что файл в буфере
+    with _mock_storage_lock:
+        assert key in _mock_storage
+
+    # Шаг 2: Браузер вызывает /api/upload_refs для этого файла
+    # Используем относительный URL пути /mock-buffer/key
+    res = client.post("/api/upload_refs", json={
+        "files": [
+            {"name": "contract.pdf", "size": len(file_bytes), "url": f"/mock-buffer/{key}"}
+        ]
+    })
+    assert res.status_code == 200
+    data = res.get_json()
+    assert data["ok"] is True
+    sid = data["session"]
+    assert data["count"] == 1
+    assert data["added"] == 1
+
+    # Шаг 3: Проверяем, что файл переместился в RAM сессии
+    files = get_files(sid)
+    assert len(files) == 1
+    assert files[0] == ("contract.pdf", file_bytes)
+
+    # Шаг 4: Проверяем, что файл удалился из mock-буфера (RAM буфера очищен)
+    with _mock_storage_lock:
+        assert key not in _mock_storage
+
+    # Шаг 5: Проверяем эндпоинт проверки сессии
+    res = client.get(f"/api/session/{sid}/files")
+    assert res.status_code == 200
+    s_data = res.get_json()
+    assert s_data["ok"] is True
+    assert s_data["session"] == sid
+    assert len(s_data["files"]) == 1
+    assert s_data["files"][0]["name"] == "contract.pdf"
+    assert s_data["files"][0]["size"] == len(file_bytes)
+
+    cleanup(sid)