Fixed tests

Author: bakwc

mysql_ch_replicator/db_optimizer.py

@@ -79,6 +79,7 @@ def optimize_table(self, db_name, table_name):
     def optimize_database(self, db_name):
         self.mysql_api.set_database(db_name)
         tables = self.mysql_api.get_tables()
+        self.mysql_api.close()
         tables = [table for table in tables if self.config.is_table_matches(table)]
 
         self.clickhouse_api.execute_command(f'USE {db_name}')

mysql_ch_replicator/mysql_api.py

@@ -57,6 +57,7 @@ def execute(self, command, commit=False, args=None):
             self.db.commit()
 
     def set_database(self, database):
+        self.reconnect_if_required()
         self.database = database
         self.cursor = self.db.cursor()
         self.cursor.execute(f'USE {self.database}')