@@ -1228,3 +1228,63 @@ def test_postgres_list_conversations_for_one_user_large_db(
12281228 None
12291229 """
12301230 benchmark_list_conversations_for_one_user (benchmark , LARGE_DB_RECORDS_COUNT )
1231+
1232+
1233+ def test_postgres_retrieve_conversation_empty_db (
1234+ postgres_database : None , benchmark : BenchmarkFixture
1235+ ) -> None :
1236+ """Benchmark retrieving conversations on an empty database.
1237+
1238+ Parameters:
1239+ postgres_database: Fixture that prepares a temporary PostgreSQL DB.
1240+ benchmark (BenchmarkFixture): pytest-benchmark fixture.
1241+
1242+ Returns:
1243+ None
1244+ """
1245+ benchmark_retrieve_conversation (benchmark , 0 )
1246+
1247+
1248+ def test_postgres_retrieve_conversation_small_db (
1249+ postgres_database : None , benchmark : BenchmarkFixture
1250+ ) -> None :
1251+ """Benchmark retrieving conversations on a small database.
1252+
1253+ Parameters:
1254+ postgres_database: Fixture that prepares a temporary PostgreSQL DB.
1255+ benchmark (BenchmarkFixture): pytest-benchmark fixture.
1256+
1257+ Returns:
1258+ None
1259+ """
1260+ benchmark_retrieve_conversation (benchmark , SMALL_DB_RECORDS_COUNT )
1261+
1262+
1263+ def test_postgres_retrieve_conversation_middle_db (
1264+ postgres_database : None , benchmark : BenchmarkFixture
1265+ ) -> None :
1266+ """Benchmark retrieving conversations on a medium-sized database.
1267+
1268+ Parameters:
1269+ postgres_database: Fixture that prepares a temporary PostgreSQL DB.
1270+ benchmark (BenchmarkFixture): pytest-benchmark fixture.
1271+
1272+ Returns:
1273+ None
1274+ """
1275+ benchmark_retrieve_conversation (benchmark , MIDDLE_DB_RECORDS_COUNT )
1276+
1277+
1278+ def test_postgres_retrieve_conversation_large_db (
1279+ postgres_database : None , benchmark : BenchmarkFixture
1280+ ) -> None :
1281+ """Benchmark retrieving conversations on a large database.
1282+
1283+ Parameters:
1284+ postgres_database: Fixture that prepares a temporary PostgreSQL DB.
1285+ benchmark (BenchmarkFixture): pytest-benchmark fixture.
1286+
1287+ Returns:
1288+ None
1289+ """
1290+ benchmark_retrieve_conversation (benchmark , LARGE_DB_RECORDS_COUNT )
0 commit comments