fix the byte2int import in tests.

Author: naihe2010

pymysqlreplication/row_event.py

@@ -1,5 +1,6 @@
 # -*- coding: utf-8 -*-
 
+import sys
 import struct
 import decimal
 import datetime
@@ -531,6 +532,12 @@ def _dump(self):
                                       row["after_values"][key]))
 
 
+
+_need_byte2int = False
+if sys.version.startswith('2') or sys.version.startswith('3.4'):
+    _need_byte2int = True
+
+
 class TableMapEvent(BinLogEvent):
     """This event describes the structure of a table.
     It's sent before a change happens on a table.
@@ -594,6 +601,9 @@ def __init__(self, from_packet, event_size, table_map, ctl_connection, **kwargs)
             self.packet.read_length_coded_binary()
             for i in range(0, len(column_types)):
                 column_type = column_types[i]
+                if _need_byte2int:
+                    column_type = byte2int(column_types[i])
+
                 try:
                     column_schema = self.column_schemas[ordinal_pos_loc]
 
@@ -617,7 +627,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