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