test: нагрузочные/стресс-тесты (маркер load): массовый pipeline, apply_ops, конкуренция, HTTP-стресс

tests/load/test_load_apply_ops.py

@@ -0,0 +1,52 @@
+"""Нагрузочный тест: массовые apply_ops (ADD → UPDATE → DELETE).
+
+Прогоняет N операций каждого типа одним вызовом apply_ops, замеряет время,
+проверяет целостность spec_current/spec_events (ничего не потеряно, не продублировано).
+"""
+import os
+import time
+
+import pytest
+
+from db import spec_events, spec_current
+from db.connection import query
+
+pytestmark = pytest.mark.load
+
+N = int(os.getenv("LOAD_OPS", "5000"))
+
+CID = "load-cid-111"
+SID = "load-sid-111"
+DID = "load-did-111"
+
+
+def test_mass_apply_ops(db):
+    # ADD N строк
+    ops = [{"action": "ADD", "new_row": {"name": f"услуга {i}", "price": i, "qty": 1, "sum": i}} for i in range(N)]
+    t0 = time.time()
+    s = spec_events.apply_ops(CID, SID, DID, ops, "pid", {})
+    t_add = time.time() - t0
+    assert s["added"] == N
+    assert len(spec_current.list_by_contract(CID)) == N
+
+    # UPDATE всех N строк
+    hashes = [spec_events._hash(f"услуга {i}") for i in range(N)]
+    ups = [{"action": "UPDATE", "target_hash": h, "new_values": {"price": i + 1}} for i, h in enumerate(hashes)]
+    t0 = time.time()
+    s = spec_events.apply_ops(CID, SID, DID, ups, "pid", {})
+    t_upd = time.time() - t0
+    assert s["updated"] == N
+    assert spec_current.list_by_contract(CID)[0]["price"] == 1  # первая строка обновлена
+
+    # DELETE всех N строк
+    dels = [{"action": "DELETE", "target_hash": h} for h in hashes]
+    t0 = time.time()
+    s = spec_events.apply_ops(CID, SID, DID, dels, "pid", {})
+    t_del = time.time() - t0
+    assert s["deleted"] == N
+    assert spec_current.list_by_contract(CID) == []
+
+    total = query("SELECT count(*) AS c FROM spec_events WHERE contract_id = %s", (CID,))
+    assert total[0]["c"] == N * 3  # ADD + UPDATE + DELETE, ничего не потеряно
+
+    print(f"\n[load] ops={N} add={t_add:.1f}с upd={t_upd:.1f}с del={t_del:.1f}с")