Skip to content

Commit 2994a03

Browse files
committed
Benchmark for list conversations for retrieve conversation for one user operation
1 parent 9b01736 commit 2994a03

1 file changed

Lines changed: 60 additions & 0 deletions

File tree

tests/benchmarks/test_app_database.py

Lines changed: 60 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1288,3 +1288,63 @@ def test_postgres_retrieve_conversation_large_db(
12881288
None
12891289
"""
12901290
benchmark_retrieve_conversation(benchmark, LARGE_DB_RECORDS_COUNT)
1291+
1292+
1293+
def test_postgres_retrieve_conversation_for_one_user_empty_db(
1294+
postgres_database: None, benchmark: BenchmarkFixture
1295+
) -> None:
1296+
"""Benchmark retrieving conversations on an empty database.
1297+
1298+
Parameters:
1299+
postgres_database: Fixture that prepares a temporary PostgreSQL DB.
1300+
benchmark (BenchmarkFixture): pytest-benchmark fixture.
1301+
1302+
Returns:
1303+
None
1304+
"""
1305+
benchmark_retrieve_conversation_for_one_user(benchmark, 0)
1306+
1307+
1308+
def test_postgres_retrieve_conversation_for_one_user_small_db(
1309+
postgres_database: None, benchmark: BenchmarkFixture
1310+
) -> None:
1311+
"""Benchmark retrieving conversations on a small database.
1312+
1313+
Parameters:
1314+
postgres_database: Fixture that prepares a temporary PostgreSQL DB.
1315+
benchmark (BenchmarkFixture): pytest-benchmark fixture.
1316+
1317+
Returns:
1318+
None
1319+
"""
1320+
benchmark_retrieve_conversation_for_one_user(benchmark, SMALL_DB_RECORDS_COUNT)
1321+
1322+
1323+
def test_postgres_retrieve_conversation_for_one_user_middle_db(
1324+
postgres_database: None, benchmark: BenchmarkFixture
1325+
) -> None:
1326+
"""Benchmark retrieving conversations on a medium-sized database.
1327+
1328+
Parameters:
1329+
postgres_database: Fixture that prepares a temporary PostgreSQL DB.
1330+
benchmark (BenchmarkFixture): pytest-benchmark fixture.
1331+
1332+
Returns:
1333+
None
1334+
"""
1335+
benchmark_retrieve_conversation_for_one_user(benchmark, MIDDLE_DB_RECORDS_COUNT)
1336+
1337+
1338+
def test_postgres_retrieve_conversation_for_one_user_large_db(
1339+
postgres_database: None, benchmark: BenchmarkFixture
1340+
) -> None:
1341+
"""Benchmark retrieving conversations on a large database.
1342+
1343+
Parameters:
1344+
postgres_database: Fixture that prepares a temporary PostgreSQL DB.
1345+
benchmark (BenchmarkFixture): pytest-benchmark fixture.
1346+
1347+
Returns:
1348+
None
1349+
"""
1350+
benchmark_retrieve_conversation_for_one_user(benchmark, LARGE_DB_RECORDS_COUNT)

0 commit comments

Comments
 (0)