test: fix to check in setUp method whether it is mariadb or not

Author: mikaniz

pymysqlreplication/tests/test_basic.py

@@ -1300,9 +1300,12 @@ def tearDown(self):
 
 
 class TestMariadbBinlogStreamReader(base.PyMySQLReplicationTestCase):
-    def test_binlog_checkpoint_event(self):
+    def setUp(self):
+        super().setUp()
         if not self.isMariaDB():
-            self.skipTest("Database is not mariadb")
+            self.skipTest("Skipping the entire class for MariaDB")
+
+    def test_binlog_checkpoint_event(self):
         self.stream.close()
         self.stream = BinLogStreamReader(
             self.database, server_id=1023, blocking=False, is_mariadb=True
@@ -1333,9 +1336,12 @@ def test_binlog_checkpoint_event(self):
 
 
 class TestMariadbBinlogStreamReader2(base.PyMySQLReplicationTestCase):
-    def test_annotate_rows_event(self):
+    def setUp(self):
+        super().setUp()
         if not self.isMariaDB():
-            self.skipTest("Database is not mariadb")
+            self.skipTest("Skipping the entire class for MariaDB")
+
+    def test_annotate_rows_event(self):
         query = "CREATE TABLE test (id INT NOT NULL AUTO_INCREMENT, data VARCHAR (50) NOT NULL, PRIMARY KEY (id))"
         self.execute(query)
         # Insert first event
@@ -1364,8 +1370,6 @@ def test_annotate_rows_event(self):
         self.assertIsInstance(event, MariadbAnnotateRowsEvent)
 
     def test_start_encryption_event(self):
-        if not self.isMariaDB():
-            self.skipTest("Database is not mariadb")
         query = "CREATE TABLE test (id INT NOT NULL AUTO_INCREMENT, data VARCHAR (50) NOT NULL, PRIMARY KEY (id))"
         self.execute(query)
         query = "INSERT INTO test (data) VALUES('Hello World')"
@@ -1404,8 +1408,6 @@ def test_start_encryption_event(self):
         self.assertEqual(len(nonce), 12)
 
     def test_gtid_list_event(self):
-        if not self.isMariaDB():
-            self.skipTest("Database is not mariadb")
         # set max_binlog_size to create new binlog file
         query = "SET GLOBAL max_binlog_size=4096"
         self.execute(query)