Skip to content

Commit 2f22a81

Browse files
committed
Added tests for row_factory
Signed-off-by: chandr-andr (Kiselev Aleksandr) <chandr@chandr.net>
1 parent b2cf502 commit 2f22a81

File tree

1 file changed

+21
-9
lines changed

1 file changed

+21
-9
lines changed

python/tests/test_value_converter.py

Lines changed: 21 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -585,19 +585,31 @@ async def test_row_factory_query_result(
585585
def row_factory(db_result: dict[str, Any]) -> list[str]:
586586
return list(db_result.keys())
587587

588-
as_pydantic = select_result.row_factory(
588+
as_row_factory = select_result.row_factory(
589589
row_factory=row_factory,
590590
)
591-
assert len(as_pydantic) == number_database_records
591+
assert len(as_row_factory) == number_database_records
592592

593-
for single_record in as_pydantic:
594-
assert isinstance(single_record, DefaultPydanticModel)
593+
assert isinstance(as_row_factory[0], list)
595594

596-
as_py_class = select_result.as_class(
597-
as_class=DefaultPythonModelClass,
595+
596+
async def test_row_factory_single_query_result(
597+
psql_pool: ConnectionPool,
598+
table_name: str,
599+
number_database_records: int,
600+
) -> None:
601+
connection = await psql_pool.connection()
602+
select_result = await connection.fetch_row(
603+
f"SELECT * FROM {table_name} LIMIT 1",
598604
)
599605

600-
assert len(as_py_class) == number_database_records
606+
def row_factory(db_result: dict[str, Any]) -> list[str]:
607+
return list(db_result.keys())
601608

602-
for single_py_record in as_py_class:
603-
assert isinstance(single_py_record, DefaultPythonModelClass)
609+
as_row_factory = select_result.row_factory(
610+
row_factory=row_factory,
611+
)
612+
expected_number_of_elements_in_result = 2
613+
assert len(as_row_factory) == expected_number_of_elements_in_result
614+
615+
assert isinstance(as_row_factory, list)

0 commit comments

Comments
 (0)