mirror of
https://github.com/aljazceru/turso.git
synced 2025-12-18 17:14:20 +01:00
ext/python: Add support for Context Manager
This commit is contained in:
@@ -144,6 +144,17 @@ def test_commit(provider):
|
||||
conn.close()
|
||||
assert record
|
||||
|
||||
@pytest.mark.parametrize("provider", ["sqlite3", "limbo"])
|
||||
def test_with_statement(provider):
|
||||
with connect(provider, "tests/database.db") as conn:
|
||||
conn = connect(provider, "tests/database.db")
|
||||
cursor = conn.cursor()
|
||||
cursor.execute("SELECT MAX(id) FROM users")
|
||||
|
||||
max_id = cursor.fetchone()
|
||||
|
||||
assert max_id
|
||||
assert max_id == (2,)
|
||||
|
||||
def connect(provider, database):
|
||||
if provider == "limbo":
|
||||
|
||||
Reference in New Issue
Block a user