ext/python: Close connection after each test

This commit is contained in:
Diego Reis 2025-03-21 11:52:58 -03:00
parent 3c2bb6c3a8
commit 2ee934577f

View file

@ -42,6 +42,8 @@ def test_fetchall_select_all_users(provider, setup_database):
cursor.execute("SELECT * FROM users")
users = cursor.fetchall()
conn.close()
assert users
assert users == [(1, "alice"), (2, "bob")]
@ -53,6 +55,8 @@ def test_fetchall_select_user_ids(provider):
cursor.execute("SELECT id FROM users")
user_ids = cursor.fetchall()
conn.close()
assert user_ids
assert user_ids == [(1,), (2,)]
@ -67,6 +71,8 @@ def test_in_memory_fetchone_select_all_users(provider):
cursor.execute("SELECT * FROM users")
alice = cursor.fetchone()
conn.close()
assert alice
assert alice == (1, "alice")
@ -82,6 +88,8 @@ def test_fetchone_select_all_users(provider):
assert alice == (1, "alice")
bob = cursor.fetchone()
conn.close()
assert bob
assert bob == (2, "bob")
@ -93,6 +101,8 @@ def test_fetchone_select_max_user_id(provider):
cursor.execute("SELECT MAX(id) FROM users")
max_id = cursor.fetchone()
conn.close()
assert max_id
assert max_id == (2,)
@ -100,8 +110,8 @@ def test_fetchone_select_max_user_id(provider):
# Test case for: https://github.com/tursodatabase/limbo/issues/494
@pytest.mark.parametrize("provider", ["sqlite3", "limbo"])
def test_commit(provider):
con = connect(provider, "tests/database.db")
cur = con.cursor()
conn = connect(provider, "tests/database.db")
cur = conn.cursor()
cur.execute("""
CREATE TABLE IF NOT EXISTS users_b (
@ -113,7 +123,7 @@ def test_commit(provider):
)
""")
con.commit()
conn.commit()
sample_users = [
("alice", "alice@example.com", "admin"),
@ -125,11 +135,13 @@ def test_commit(provider):
for username, email, role in sample_users:
cur.execute("INSERT INTO users_b (username, email, role) VALUES (?, ?, ?)", (username, email, role))
con.commit()
conn.commit()
# Now query the table
res = cur.execute("SELECT * FROM users_b")
record = res.fetchone()
conn.close()
assert record