Макет библиотечного метода 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: Как выполнить модульное тестирование модуля, который содержит операции с базой данных? , который, как я думал, решит мою проблему, но mysql. connect.connect продолжает вызываться вместо макета).

Я не знаю, что я могу сделать, чтобы проверить этот класс