- Fix connection.autocommit. (#21)

- Fix `_clear_result`. (#22)

Author: long2ice

CHANGELOG.md

@@ -5,6 +5,8 @@
 ### 0.2.3
 
 - Fix `escape_sequence`. (#20)
+- Fix `connection.autocommit`. (#21)
+- Fix `_clear_result`. (#22)
 
 ### 0.2.2
 

asyncmy/connection.pyx

@@ -636,7 +636,7 @@ class Connection:
             await result.read()
         self._result = result
         self._affected_rows = result.affected_rows
-        if result.server_status is not None:
+        if result.server_status != 0:
             self.server_status = result.server_status
 
     def insert_id(self):
@@ -704,7 +704,7 @@ class Connection:
 
         if self._ssl_context:
             # capablities, max packet, charset
-            data = IIB.pack( self._client_flag, 16777216, 33)
+            data = IIB.pack(self._client_flag, 16777216, 33)
             data += b'\x00' * (32 - len(data))
 
             self.write_packet(data)
@@ -1141,7 +1141,7 @@ cdef class MySQLResult:
         self.affected_rows = len(rows)
         self.rows = tuple(rows)
 
-    def _read_row_from_packet(self, packet):
+    cpdef _read_row_from_packet(self, packet: MysqlPacket):
         row = []
         for encoding, converter in self.converters:
             try:

asyncmy/cursors.pyx

@@ -306,7 +306,7 @@ cdef class Cursor:
         self._executed = q
         return args
 
-    cpdef  fetchone(self):
+    cpdef fetchone(self):
         """Fetch the next row."""
         self._check_executed()
         fut = self._loop.create_future()
@@ -366,7 +366,7 @@ cdef class Cursor:
         await self._do_get_result()
         return self.rowcount
 
-    cdef _clear_result(self):
+    cpdef _clear_result(self):
         self.rownumber = 0
         self._result = None
         self.rowcount = 0

tests/test_autocommit.py

@@ -0,0 +1,14 @@
+import pytest
+
+
+@pytest.mark.asyncio
+async def test_autocommit(connection):
+    await connection.autocommit(True)
+    cursor = connection.cursor()
+    await cursor.execute("SELECT @@autocommit;")
+    assert await cursor.fetchone() == (1,)
+    await cursor.close()
+    await connection.autocommit(False)
+    cursor = connection.cursor()
+    await cursor.execute("SELECT @@autocommit;")
+    assert await cursor.fetchone() == (0,)