mirror of
https://github.com/tursodatabase/limbo.git
synced 2025-08-04 10:08:20 +00:00
ext/python: Close connection after each test
This commit is contained in:
parent
3c2bb6c3a8
commit
2ee934577f
1 changed files with 16 additions and 4 deletions
|
@ -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
|
||||
|
||||
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue