fixed unittest.testcase to PyMySQLReplicationTestCase

Author: roy0424

pymysqlreplication/tests/test_util_bytes.py

@@ -1,8 +1,8 @@
-import unittest
 from pymysqlreplication.util import *
+from pymysqlreplication.tests.base import PyMySQLReplicationTestCase
 
 
-class TestIsDataShort(unittest.TestCase):
+class TestIsDataShort(PyMySQLReplicationTestCase):
     def test_data_is_shorter(self):
         data = bytearray([0x01])
         expected_length = 2
@@ -24,7 +24,7 @@ def test_data_is_empty(self):
         self.assertTrue(is_data_short(data, expected_length))
 
 
-class TestDecodeCount(unittest.TestCase):
+class TestDecodeCount(PyMySQLReplicationTestCase):
     def test_small_format(self):
         data = bytearray([0x01, 0x00])
         is_small = True
@@ -38,7 +38,7 @@ def test_large_format(self):
         self.assertEqual(result, 1)
 
 
-class TestDecodeUint(unittest.TestCase):
+class TestDecodeUint(PyMySQLReplicationTestCase):
     def test_valid_input(self):
         data = bytearray([0x01, 0x00])
         result = decode_uint(data)
@@ -55,7 +55,7 @@ def test_empty_data(self):
         self.assertEqual(result, 0)
 
 
-class TestDecodeVariableLength(unittest.TestCase):
+class TestDecodeVariableLength(PyMySQLReplicationTestCase):
     def test_single_byte(self):
         data = bytearray([0x05])
         length, pos = decode_variable_length(data)
@@ -75,7 +75,7 @@ def test_max_length(self):
         self.assertEqual(pos, 5)
 
 
-class TestParseUint16(unittest.TestCase):
+class TestParseUint16(PyMySQLReplicationTestCase):
     def test_valid_input(self):
         data = bytearray([0x01, 0x00])
         result = parse_uint16(data)
@@ -87,7 +87,7 @@ def test_different_input(self):
         self.assertEqual(result, 255)
 
 
-class TestLengthEncodedInt(unittest.TestCase):
+class TestLengthEncodedInt(PyMySQLReplicationTestCase):
     def test_single_byte(self):
         data = bytearray([0x05])
         result, _, _ = length_encoded_int(data)
@@ -104,7 +104,7 @@ def test_three_bytes(self):
         self.assertEqual(result, 0x030201)
 
 
-class TestDecodeTime(unittest.TestCase):
+class TestDecodeTime(PyMySQLReplicationTestCase):
     def test_midnight(self):
         data = bytearray([0x00] * 8)
         result = decode_time(data)
@@ -116,7 +116,7 @@ def test_valid_time(self):
         self.assertEqual(result, datetime.time(17, 35, 0))
 
 
-class TestDecodeDatetime(unittest.TestCase):
+class TestDecodeDatetime(PyMySQLReplicationTestCase):
     def test_zero_datetime(self):
         data = bytearray([0x00] * 8)
         result = decode_datetime(data)
@@ -129,7 +129,3 @@ def test_valid_datetime(self):
         )
         result = decode_datetime(data)
         self.assertEqual(result, expected_datetime)
-
-
-if __name__ == "__main__":
-    unittest.main()