Make sure the correct values are returned from a postgres record (#173)

Co-authored-by: Tom Christie <[email protected]>

Author: rzane

databases/backends/postgres.py

@@ -94,6 +94,9 @@ def __init__(
             self._column_map_int[idx] = (idx, datatype)
             self._column_map_full[str(column[0])] = (idx, datatype)
 
+    def values(self) -> typing.ValuesView:
+        return self._row.values()
+
     def __getitem__(self, key: typing.Any) -> typing.Any:
         if len(self._column_map) == 0:  # raw query
             return self._row[tuple(self._row.keys()).index(key)]

tests/test_databases.py

@@ -278,6 +278,22 @@ async def test_results_support_column_reference(database_url):
             assert results[0][custom_date.c.published] == today
 
 
+@pytest.mark.parametrize("database_url", DATABASE_URLS)
+@async_adapter
+async def test_result_values_allow_duplicate_names(database_url):
+    """
+    The values of a result should respect when two columns are selected
+    with the same name.
+    """
+    async with Database(database_url) as database:
+        async with database.transaction(force_rollback=True):
+            query = "SELECT 1 AS id, 2 AS id"
+            row = await database.fetch_one(query=query)
+
+            assert list(row.keys()) == ["id", "id"]
+            assert list(row.values()) == [1, 2]
+
+
 @pytest.mark.parametrize("database_url", DATABASE_URLS)
 @async_adapter
 async def test_fetch_one_returning_no_results(database_url):