如何在Python unittest中重置:memory:中的SQLite数据库？

嘿，这个问题我之前也碰到过——内存里的SQLite数据库没法像磁盘文件那样直接删掉，得换个思路来处理。其实核心在于SQLite的:memory:数据库是和单个连接绑定的：每个新的连接都会创建一个全新的空内存数据库，而关闭连接后对应的内存数据库就会被销毁。

针对你的unittest场景，最简洁可靠的方案是在每个测试用例运行前重新创建数据库连接，具体操作如下：

方法一：利用unittest的setUp方法新建连接 #

在unittest的setUp方法里初始化数据库连接（以及必要的表结构），这个方法会在每个测试用例执行前自动运行，确保每次测试都从全新的空数据库开始。

示例代码：

import unittest
import sqlite3

class TestDBOperations(unittest.TestCase):
    def setUp(self):
        # 每个测试前新建内存数据库连接
        self.db_conn = sqlite3.connect(":memory:")
        cursor = self.db_conn.cursor()
        
        # 这里可以放你的数据库初始化代码，比如创建表
        cursor.execute("""
            CREATE TABLE products (
                id INTEGER PRIMARY KEY AUTOINCREMENT,
                name TEXT NOT NULL,
                price REAL
            )
        """)
        self.db_conn.commit()

    def tearDown(self):
        # 测试结束后关闭连接，释放内存
        self.db_conn.close()

    def test_insert_product(self):
        cursor = self.db_conn.cursor()
        cursor.execute("INSERT INTO products (name, price) VALUES ('Apple', 5.99)")
        self.db_conn.commit()
        
        cursor.execute("SELECT COUNT(*) FROM products")
        self.assertEqual(cursor.fetchone()[0], 1)

    def test_empty_db_initial_state(self):
        cursor = self.db_conn.cursor()
        cursor.execute("SELECT COUNT(*) FROM products")
        # 初始状态下数据库是空的
        self.assertEqual(cursor.fetchone()[0], 0)

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

这个方案的优势在于完全避免了手动清空数据的麻烦，每次测试都是独立的全新环境，不会有数据残留的问题。

方法二：手动清空现有数据库（不推荐，仅作补充） #

如果因为某些原因你不想重新创建连接，也可以手动删除所有表的数据甚至表本身。不过这种方法需要处理所有用户创建的表，还要注意重置自增序列，相对繁琐：

def clear_memory_db(conn):
    cursor = conn.cursor()
    # 获取所有用户创建的表（排除SQLite系统表）
    cursor.execute("SELECT name FROM sqlite_master WHERE type='table' AND name NOT LIKE 'sqlite_%';")
    tables = cursor.fetchall()
    
    for table in tables:
        table_name = table[0]
        # 清空表数据
        cursor.execute(f"DELETE FROM {table_name};")
        # 重置自增ID（针对使用AUTOINCREMENT的主键）
        cursor.execute(f"DELETE FROM sqlite_sequence WHERE name='{table_name}';")
    
    conn.commit()

然后在setUp里调用这个函数（前提是你已经创建了连接和表结构）：

def setUp(self):
    self.db_conn = sqlite3.connect(":memory:")
    # 初始化表结构...
    clear_memory_db(self.db_conn)

不过还是强烈推荐第一种方法，既简单又不容易出错，完全符合你“每次测试从空数据库开始”的需求。

内容的提问来源于stack exchange，提问作者sunyata