v0.0.2: split api_bp into upload/process/download modules, Connection: close, upload tests (13 new, 106 total)

tests/test_upload.py

@@ -0,0 +1,283 @@
+"""
+Тесты POST /api/upload — загрузка файлов в сессию.
+
+Запуск:
+    python3 tests/test_upload.py
+"""
+
+import io
+import os
+import sys
+import time
+import json
+import gc
+
+# Настраиваем пути
+_site = os.path.join(os.path.dirname(__file__), "..", "site")
+sys.path.insert(0, _site)
+sys.path.insert(0, os.path.join(os.path.dirname(__file__), ".."))
+
+from flask import Flask
+
+# Импортируем напрямую, минуя routes/__init__.py (process_bp тянет drhider)
+import importlib.util
+_spec = importlib.util.spec_from_file_location(
+    "upload_bp", os.path.join(_site, "routes", "upload_bp.py")
+)
+_upload_mod = importlib.util.module_from_spec(_spec)
+_spec.loader.exec_module(_upload_mod)
+upload_bp = _upload_mod.upload_bp
+
+
+# ═══════════════════════════════════════════════════════════════════════════
+# Helpers
+# ═══════════════════════════════════════════════════════════════════════════
+
+def _new_client():
+    """Создать свежий тестовый клиент."""
+    app = Flask(__name__)
+    app.register_blueprint(upload_bp)
+    return app.test_client()
+
+
+def _get_rss():
+    """RSS процесса в байтах (Linux)."""
+    try:
+        with open("/proc/self/statm", "r") as f:
+            parts = f.read().split()
+            return int(parts[1]) * 4096
+    except Exception:
+        return 0
+
+
+# ═══════════════════════════════════════════════════════════════════════════
+# Базовые тесты
+# ═══════════════════════════════════════════════════════════════════════════
+
+def test_no_file():
+    """POST без файла → 400."""
+    c = _new_client()
+    rv = c.post("/api/upload")
+    assert rv.status_code == 400
+    data = json.loads(rv.data)
+    assert data["ok"] is False
+    assert "No file" in data["error"]
+
+
+def test_no_filename():
+    """POST с файлом без имени → 400."""
+    c = _new_client()
+    data = {"files": (io.BytesIO(b"hello"), "")}
+    rv = c.post("/api/upload", data=data, content_type="multipart/form-data")
+    assert rv.status_code == 400
+    resp = json.loads(rv.data)
+    assert resp["ok"] is False
+    assert "filename" in resp["error"].lower()
+
+
+def test_single_file():
+    """POST с одним файлом → 200, сессия + count=1."""
+    c = _new_client()
+    data = {"files": (io.BytesIO(b"hello world"), "test.txt")}
+    rv = c.post("/api/upload", data=data, content_type="multipart/form-data")
+    assert rv.status_code == 200
+    resp = json.loads(rv.data)
+    assert resp["ok"] is True
+    assert len(resp["session"]) == 12
+    assert resp["count"] == 1
+
+
+def test_two_files_same_session():
+    """Два файла в одну сессию → count=2."""
+    c = _new_client()
+    data1 = {"files": (io.BytesIO(b"first"), "file1.txt")}
+    rv1 = c.post("/api/upload", data=data1, content_type="multipart/form-data")
+    assert rv1.status_code == 200
+    sid = json.loads(rv1.data)["session"]
+
+    data2 = {
+        "files": (io.BytesIO(b"second"), "file2.txt"),
+        "session": sid,
+    }
+    rv2 = c.post("/api/upload", data=data2, content_type="multipart/form-data")
+    assert rv2.status_code == 200
+    resp2 = json.loads(rv2.data)
+    assert resp2["session"] == sid
+    assert resp2["count"] == 2
+
+
+def test_fake_session_404():
+    """Несуществующая сессия → 404."""
+    c = _new_client()
+    data = {
+        "files": (io.BytesIO(b"data"), "f.txt"),
+        "session": "deadbeef1234",
+    }
+    rv = c.post("/api/upload", data=data, content_type="multipart/form-data")
+    assert rv.status_code == 404
+    resp = json.loads(rv.data)
+    assert resp["ok"] is False
+
+
+def test_empty_file():
+    """Пустой файл (0 байт) → 200."""
+    c = _new_client()
+    data = {"files": (io.BytesIO(b""), "empty.txt")}
+    rv = c.post("/api/upload", data=data, content_type="multipart/form-data")
+    assert rv.status_code == 200
+    resp = json.loads(rv.data)
+    assert resp["ok"] is True
+    assert resp["count"] == 1
+
+
+def test_cyrillic_filename():
+    """Кириллическое имя файла → 200."""
+    c = _new_client()
+    content = "кириллица".encode("utf-8")
+    data = {"files": (io.BytesIO(content), "договор_№123.txt")}
+    rv = c.post("/api/upload", data=data, content_type="multipart/form-data")
+    assert rv.status_code == 200
+    resp = json.loads(rv.data)
+    assert resp["ok"] is True
+    assert resp["count"] == 1
+
+
+def test_content_type_json():
+    """Ответ — application/json."""
+    c = _new_client()
+    data = {"files": (io.BytesIO(b"x"), "f.txt")}
+    rv = c.post("/api/upload", data=data, content_type="multipart/form-data")
+    assert rv.status_code == 200
+    assert "application/json" in rv.content_type
+
+
+# ═══════════════════════════════════════════════════════════════════════════
+# Большие файлы
+# ═══════════════════════════════════════════════════════════════════════════
+
+def test_large_10mb():
+    """Файл 10 MB — загружается без ошибок."""
+    c = _new_client()
+    size = 10 * 1024 * 1024
+    big_data = b"A" * size
+    data = {"files": (io.BytesIO(big_data), "big_10mb.bin")}
+    rv = c.post("/api/upload", data=data, content_type="multipart/form-data")
+    assert rv.status_code == 200
+    resp = json.loads(rv.data)
+    assert resp["ok"] is True
+    assert resp["count"] == 1
+
+
+def test_large_50mb():
+    """Файл 50 MB — загружается без ошибок."""
+    c = _new_client()
+    size = 50 * 1024 * 1024
+    big_data = b"B" * size
+    data = {"files": (io.BytesIO(big_data), "big_50mb.bin")}
+    rv = c.post("/api/upload", data=data, content_type="multipart/form-data")
+    assert rv.status_code == 200
+    resp = json.loads(rv.data)
+    assert resp["ok"] is True
+    assert resp["count"] == 1
+
+
+# ═══════════════════════════════════════════════════════════════════════════
+# Много мелких файлов
+# ═══════════════════════════════════════════════════════════════════════════
+
+def test_many_small_files():
+    """100 файлов по 1 KB в одну сессию."""
+    c = _new_client()
+    data1 = {"files": (io.BytesIO(b"x" * 1024), "file_000.txt")}
+    rv1 = c.post("/api/upload", data=data1, content_type="multipart/form-data")
+    sid = json.loads(rv1.data)["session"]
+
+    for i in range(1, 100):
+        data = {
+            "files": (io.BytesIO(b"y" * 1024), f"file_{i:03d}.txt"),
+            "session": sid,
+        }
+        rv = c.post("/api/upload", data=data, content_type="multipart/form-data")
+        assert rv.status_code == 200
+        resp = json.loads(rv.data)
+        assert resp["session"] == sid
+        assert resp["count"] == i + 1
+
+    assert resp["count"] == 100
+
+
+# ═══════════════════════════════════════════════════════════════════════════
+# Утечка памяти
+# ═══════════════════════════════════════════════════════════════════════════
+
+def test_memory_after_large_files():
+    """После загрузки больших файлов память не течёт."""
+    c = _new_client()
+    gc.collect()
+    mem_before = _get_rss()
+
+    for i in range(10):
+        data = {"files": (io.BytesIO(b"M" * (1024 * 1024)), f"memtest_{i}.bin")}
+        rv = c.post("/api/upload", data=data, content_type="multipart/form-data")
+        assert rv.status_code == 200
+
+    gc.collect()
+    mem_after = _get_rss()
+    diff_mb = (mem_after - mem_before) / (1024 * 1024)
+    assert diff_mb < 50, f"Memory grew {diff_mb:.1f} MB, expected < 50 MB"
+
+
+# ═══════════════════════════════════════════════════════════════════════════
+# Бенчмарк скорости
+# ═══════════════════════════════════════════════════════════════════════════
+
+def test_upload_speed():
+    """1 MB файл — загрузка < 500 ms."""
+    c = _new_client()
+    data = {"files": (io.BytesIO(b"S" * (1024 * 1024)), "speed.bin")}
+
+    start = time.time()
+    rv = c.post("/api/upload", data=data, content_type="multipart/form-data")
+    elapsed = time.time() - start
+
+    assert rv.status_code == 200
+    assert elapsed < 0.5, f"Upload took {elapsed:.3f}s, expected < 0.5s"
+
+
+# ═══════════════════════════════════════════════════════════════════════════
+# Основной блок
+# ═══════════════════════════════════════════════════════════════════════════
+
+if __name__ == "__main__":
+    passed = 0
+    failed = 0
+    tests = [
+        test_no_file,
+        test_no_filename,
+        test_single_file,
+        test_two_files_same_session,
+        test_fake_session_404,
+        test_empty_file,
+        test_cyrillic_filename,
+        test_content_type_json,
+        test_large_10mb,
+        test_large_50mb,
+        test_many_small_files,
+        test_memory_after_large_files,
+        test_upload_speed,
+    ]
+    for test in tests:
+        try:
+            test()
+            print(f"  ✅ {test.__name__}")
+            passed += 1
+        except Exception as e:
+            print(f"  ❌ {test.__name__}: {e}")
+            failed += 1
+    print(f"\n{'='*50}")
+    print(f"Результат: {passed} OK, {failed} FAIL из {len(tests)}")
+    if failed == 0:
+        print("✅ ВСЕ ТЕСТЫ ПРОЙДЕНЫ")
+    else:
+        print("❌ ЕСТЬ ПРОВАЛЫ")
+        sys.exit(1)