Add benchmarks

2025-12-12 19:44:11 +09:00
parent 84d9627f99
commit 97c228816e
14 changed files with 820 additions and 16 deletions
--- a/tests/benchmark/test_benchmark_user_balance.py
+++ b/tests/benchmark/test_benchmark_user_balance.py
@@ -0,0 +1,81 @@
+import pytest
+from sqlalchemy.orm import Session
+
+from dibbler.models import User
+from dibbler.queries import update_cache, user_balance
+from tests.benchmark.benchmark_settings import BENCHMARK_ITERATIONS, BENCHMARK_ROUNDS
+from tests.benchmark.helpers import generate_random_transactions, insert_users_and_products
+
+
+@pytest.mark.benchmark(group="user_balance")
+@pytest.mark.parametrize(
+    "transaction_count",
+    [
+        100,
+        500,
+        1000,
+        1500,
+        2000,
+    ],
+)
+def test_benchmark_user_balance(
+    benchmark,
+    sql_session: Session,
+    transaction_count: int,
+):
+    users, _products = insert_users_and_products(sql_session)
+
+    transactions = generate_random_transactions(
+        sql_session,
+        transaction_count,
+    )
+
+    sql_session.add_all(transactions)
+    sql_session.commit()
+
+    benchmark.pedantic(
+        query_all_users_balance,
+        args=(sql_session, users),
+        iterations=BENCHMARK_ITERATIONS,
+        rounds=BENCHMARK_ROUNDS,
+    )
+
+
+@pytest.mark.benchmark(group="user_balance")
+@pytest.mark.parametrize(
+    "transaction_count",
+    [
+        1000,
+        1500,
+        2000,
+    ],
+)
+def test_benchmark_user_balance_cache_every_500(
+    benchmark,
+    sql_session: Session,
+    transaction_count: int,
+):
+    users, _products = insert_users_and_products(sql_session)
+
+    transactions = generate_random_transactions(
+        sql_session,
+        transaction_count,
+    )
+
+    for i in range(0, len(transactions), 500):
+        update_cache(sql_session)
+
+        sql_session.add_all(transactions[i : i + 500])
+        sql_session.commit()
+
+    benchmark.pedantic(
+        query_all_users_balance,
+        args=(sql_session, users),
+        iterations=BENCHMARK_ITERATIONS,
+        rounds=BENCHMARK_ROUNDS,
+    )
+
+
+def query_all_users_balance(sql_session: Session, users: list[User]) -> None:
+    for user in users:
+        user_balance(sql_session, user, use_cache=False)