fix the byte2int import in tests.

Author: naihe2010

pymysqlreplication/row_event.py

@@ -5,7 +5,7 @@
 import datetime
 # import json
 
-from six import byte2int
+from six import byte2int, PY2
 from pymysql.charset import charset_by_name
 
 from .event import BinLogEvent
@@ -593,7 +593,12 @@ def __init__(self, from_packet, event_size, table_map, ctl_connection, **kwargs)
             column_types = list(self.packet.read(self.column_count))
             self.packet.read_length_coded_binary()
             for i in range(0, len(column_types)):
-                column_type = column_types[i]
+                if PY2:
+                    column_type = byte2int(column_types[i])
+
+                else:
+                    column_type = column_types[i]
+
                 try:
                     column_schema = self.column_schemas[ordinal_pos_loc]
 
@@ -617,7 +622,7 @@ def __init__(self, from_packet, event_size, table_map, ctl_connection, **kwargs)
                         'COLUMN_TYPE': 'BLOB',  # we don't know what it is, so let's not do anything with it.
                         'COLUMN_KEY': '',
                     }
-                col = Column(int(column_type), column_schema, from_packet)
+                col = Column(column_type, column_schema, from_packet)
                 self.columns.append(col)
 
         self.table_obj = Table(self.column_schemas, self.table_id, self.schema,

pymysqlreplication/tests/binlogfilereader.py

@@ -1,7 +1,7 @@
 '''Read binlog files'''
 import struct
 
-from pymysql.util import byte2int
+from six import byte2int
 from pymysqlreplication import constants
 from pymysqlreplication.event import FormatDescriptionEvent
 from pymysqlreplication.event import QueryEvent

pymysqlreplication/tests/test_data_type.py

@@ -8,6 +8,7 @@
 else:
     import unittest
 
+import json
 from decimal import Decimal
 
 from pymysqlreplication.tests import base