mysql 单元测试

import mysql.connector
from mysql.connector import Error
import unittest

class TestMySQL(unittest.TestCase):
    def setUp(self):
        try:
            self.connection = mysql.connector.connect(host='localhost',
                                                     database='testdb',
                                                     user='testuser',
                                                     password='testpass')
            self.cursor = self.connection.cursor()
        except Error as e:
            print(f"Error connecting to MySQL: {e}")

    def test_insert_data(self):
        insert_query = "INSERT INTO test_table (name, age) VALUES (%s, %s)"
        insert_data = ("John Doe", 30)
        self.cursor.execute(insert_query, insert_data)
        self.connection.commit()
        select_query = "SELECT * FROM test_table WHERE name = %s"
        self.cursor.execute(select_query, ("John Doe",))
        result = self.cursor.fetchone()
        self.assertIsNotNone(result)
        self.assertEqual(result[1], 30)

    def tearDown(self):
        if self.connection.is_connected():
            self.cursor.close()
            self.connection.close()

if __name__ == '__main__':
    unittest.main()