How to Mock SQLite executemany When Batch Inserting in Python

import sqlite3
from unittest.mock import Mock, patch

def insert_users(conn, users):
    """Insert multiple user records using executemany."""
    cursor = conn.cursor()
    cursor.executemany(
        "INSERT INTO users (name, age) VALUES (?, ?)",
        users
    )
    conn.commit()
    return cursor.rowcount

if __name__ == "__main__":
    # Create a real in-memory database for demonstration
    conn = sqlite3.connect(":memory:")
    conn.execute("CREATE TABLE users (name TEXT, age INTEGER)")
    
    users = [
        ("Alice", 30),
        ("Bob", 25),
        ("Charlie", 35)
    ]
    
    count = insert_users(conn, users)
    print(f"Inserted {count} rows")
    
    # Verify the data
    rows = conn.execute("SELECT * FROM users ORDER BY name").fetchall()
    print("Database contents:", rows)
    
    # Now demonstrate mocking for testing
    mock_conn = Mock()
    mock_conn.cursor.return_value.executemany.return_value = Mock(rowcount=0)
    
    with patch("__main__.insert_users", wraps=insert_users) as mocked_insert:
        # Test with mock connection
        result = mocked_insert(mock_conn, [("Test", 1)])
        print(f"Mock result: {result}")
    
    conn.close()