@@ -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