如何使用mock对象测试将数据插入到mysql的插入函数？

import unittest
from unittest import mock
from your_module import insert_data_to_mysql

class TestInsertData(unittest.TestCase):
    def test_insert_data_to_mysql(self):
        # 创建一个mock的MySQL连接对象
        mock_mysql_conn = mock.MagicMock()
        
        # 创建一个mock的游标对象
        mock_cursor = mock_mysql_conn.cursor.return_value
        
        # 设置游标对象的execute方法返回值
        mock_cursor.execute.return_value = None
        
        # 调用插入函数
        insert_data_to_mysql(mock_mysql_conn, data)
        
        # 断言游标对象的execute方法被调用了一次
        mock_cursor.execute.assert_called_once()
        
        # 断言游标对象的commit方法被调用了一次
        mock_mysql_conn.commit.assert_called_once()