Fix TypeError on get_attributes when result has no columns

MagicStack · Sep 13, 2017 · f29de23 · f29de23
1 parent b6fe018
commit f29de23
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/asyncpg/protocol/prepared_stmt.pyx b/asyncpg/protocol/prepared_stmt.pyx
@@ -40,6 +40,9 @@ cdef class PreparedStatementState:
     def _get_attributes(self):
         cdef Codec codec
 
+        if not self.row_desc:
+            return ()
+
         result = []
         for d in self.row_desc:
             name = d[0]

diff --git a/tests/test_codecs.py b/tests/test_codecs.py
@@ -1398,3 +1398,7 @@ async def test_enum_and_range(self):
                 DROP TABLE testtab;
                 DROP TYPE enum_t;
             ''')
+
+    async def test_no_result(self):
+        st = await self.con.prepare('rollback')
+        self.assertTupleEqual(st.get_attributes(), ())