mirror of
https://github.com/simonw/sqlite-utils.git
synced 2026-07-23 09:24:31 +02:00
72 lines
2 KiB
Python
72 lines
2 KiB
Python
from sqlite_utils import Database
|
|
from sqlite_utils.utils import sqlite3
|
|
import pytest
|
|
|
|
|
|
def test_recursive_triggers():
|
|
db = Database(memory=True)
|
|
assert db.execute("PRAGMA recursive_triggers").fetchone()[0]
|
|
|
|
|
|
def test_recursive_triggers_off():
|
|
db = Database(memory=True, recursive_triggers=False)
|
|
assert not db.execute("PRAGMA recursive_triggers").fetchone()[0]
|
|
|
|
|
|
def test_memory_name():
|
|
db1 = Database(memory_name="shared")
|
|
db2 = Database(memory_name="shared")
|
|
db1["dogs"].insert({"name": "Cleo"})
|
|
assert list(db2["dogs"].rows) == [{"name": "Cleo"}]
|
|
|
|
|
|
def test_sqlite_version():
|
|
db = Database(memory=True)
|
|
version = db.sqlite_version
|
|
assert isinstance(version, tuple)
|
|
as_string = ".".join(map(str, version))
|
|
actual = next(db.query("select sqlite_version() as v"))["v"]
|
|
assert actual == as_string
|
|
|
|
|
|
@pytest.mark.parametrize("memory", [True, False])
|
|
def test_database_close(tmpdir, memory):
|
|
if memory:
|
|
db = Database(memory=True)
|
|
else:
|
|
db = Database(str(tmpdir / "test.db"))
|
|
assert db.execute("select 1 + 1").fetchone()[0] == 2
|
|
db.close()
|
|
with pytest.raises(sqlite3.ProgrammingError):
|
|
db.execute("select 1 + 1")
|
|
|
|
|
|
def test_memory_attribute_for_memory_true():
|
|
db = Database(memory=True)
|
|
assert db.memory is True
|
|
assert db.memory_name is None
|
|
|
|
|
|
def test_memory_attribute_for_memory_name():
|
|
db = Database(memory_name="shared_attr")
|
|
assert db.memory is True
|
|
assert db.memory_name == "shared_attr"
|
|
|
|
|
|
def test_memory_attribute_for_memory_string_path():
|
|
db = Database(":memory:")
|
|
assert db.memory is True
|
|
assert db.memory_name is None
|
|
|
|
|
|
def test_memory_attribute_for_file_path(tmpdir):
|
|
db = Database(str(tmpdir / "file.db"))
|
|
assert db.memory is False
|
|
assert db.memory_name is None
|
|
|
|
|
|
def test_memory_attribute_for_existing_connection():
|
|
conn = sqlite3.connect(":memory:")
|
|
db = Database(conn)
|
|
assert db.memory is False
|
|
assert db.memory_name is None
|