Question

因此，我编写了一个类，该类带有与现有数据库的连接，如下所示：

class DatabaseHandler:
    def __init__(self):
        self.connection = sqlite3.connect('database.db)
    def insert (self):
        # inserts stuff

我要测试insert函数是否按预期工作。但是我不想弄乱database.db。我想到的是模拟类，使其不连接到database.db，而是连接到内存中的表（':memory:'）。

不幸的是，我不知道该怎么做，到目前为止，互联网对我没有帮助。如果可能的话，我不想更改类并提供数据库路径作为参数，因为它应该始终连接到同一数据库。

我知道如何创建一个内存数据库，但是当我调用我的类时，它将连接到database.db，这根本不是我的意图。

我正在使用python内置的unittest-module。

感谢您的帮助。