From eff5596bdd14f4c9cd83b8b8a8c6e2906118110c Mon Sep 17 00:00:00 2001
From: heehehe <heeseon0315@gmail.com>
Date: Tue, 19 Sep 2023 22:45:41 +0900
Subject: [PATCH] feat: add version8 test in test_data_type.py related PR:
 https://github.com/julien-duponchelle/python-mysql-replication/pull/355

---
 pymysqlreplication/tests/base.py           |  3 ++-
 pymysqlreplication/tests/test_data_type.py | 20 +++++++++++++++++++-
 2 files changed, 21 insertions(+), 2 deletions(-)

diff --git a/pymysqlreplication/tests/base.py b/pymysqlreplication/tests/base.py
index eec03f65..2cd7e2cf 100644
--- a/pymysqlreplication/tests/base.py
+++ b/pymysqlreplication/tests/base.py
@@ -161,9 +161,10 @@ def bin_log_basename(self):
         bin_log_basename = bin_log_basename.split("/")[-1]
         return bin_log_basename
 
+
 class PyMySQLReplicationPercona8TestCase(PyMySQLReplicationTestCase):
     def setUp(self):
-        super().setUp()        
+        super().setUp()
         # default
         self.database = {
             "host": os.environ.get("MYSQL_8_0") or "localhost",
diff --git a/pymysqlreplication/tests/test_data_type.py b/pymysqlreplication/tests/test_data_type.py
index ed30cf9c..c4f74bb3 100644
--- a/pymysqlreplication/tests/test_data_type.py
+++ b/pymysqlreplication/tests/test_data_type.py
@@ -19,7 +19,7 @@
 from pymysqlreplication._compat import text_type
 
 
-__all__ = ["TestDataType"]
+__all__ = ["TestDataType", "TestDataTypeVersion8"]
 
 
 def to_binary_dict(d):
@@ -894,5 +894,23 @@ def test_mariadb_only_status_vars(self):
         self.assertEqual(event.query, create_query)
 
 
+class TestDataTypeVersion8(base.PyMySQLReplicationPercona8TestCase):
+    def test_partition_id(self):
+        if not self.isMySQL80AndMore():
+            self.skipTest("Not supported in this version of MySQL")
+        create_query = "CREATE TABLE test (id INTEGER) \
+            PARTITION BY RANGE (id) ( \
+                PARTITION p0 VALUES LESS THAN (1),   \
+                PARTITION p1 VALUES LESS THAN (2),   \
+                PARTITION p2 VALUES LESS THAN (3),   \
+                PARTITION p3 VALUES LESS THAN (4),   \
+                PARTITION p4 VALUES LESS THAN (5)    \
+            )"
+        insert_query = "INSERT INTO test (id) VALUES(3)"
+        event = self.create_and_insert_value(create_query, insert_query)
+        self.assertEqual(event.extra_data_type, 1)
+        self.assertEqual(event.partition_id, 3)
+
+
 if __name__ == "__main__":
     unittest.main()