模拟库方法python

Question

我正在尝试对与数据库的类连接进行单元测试。为了避免对数据库进行硬编码，我想断言使用足够的值调用了mysql.connector.connect方法。

from mysql.connector import connect
from mysql.connector import Error

from discovery.database import Database


class MariaDatabase(Database):

    def connect(self, username, password):
        """
        Don't forget to close the connection !

        :return: connection to the database
        """
        try:
            return connect(host=str(self.target),
                           database=self.db_name,
                           user=username,
                           password=password)

我已经阅读了有关模拟和类似问题的文档（特别是这个Python Unit Test : How to unit test the module which contains database operations?，我以为可以解决我的问题，但是mysql.connector.connect一直被调用而不是模拟）。

我不知道该怎么做才能对该课程进行UTest