mirror of
https://github.com/simonw/datasette.git
synced 2026-10-19 15:56:58 +02:00
70 lines
2.4 KiB
Python
70 lines
2.4 KiB
Python
|
"""Error recovery must also work without Python 3.11's SQLite error metadata."""
|
||
|
|
|
||
|
|
import errno
|
||
|
|
from types import SimpleNamespace
|
||
|
|
from unittest.mock import Mock
|
||
|
|
|
||
|
|
import pytest
|
||
|
|
|
||
|
|
from datasette.schema_watcher import _is_transient
|
||
|
|
from datasette.utils import sqlite3
|
||
|
|
from datasette.write_budget import DatabaseResourceError, connect_with_retry
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.fixture
|
||
|
|
def old_sqlite_errors(monkeypatch):
|
||
|
|
for name in ("SQLITE_CANTOPEN", "SQLITE_NOMEM"):
|
||
|
|
monkeypatch.delattr(sqlite3, name, raising=False)
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.mark.parametrize(
|
||
|
|
"error,expected",
|
||
|
|
[
|
||
|
|
(sqlite3.OperationalError("unable to open database file"), True),
|
||
|
|
(sqlite3.OperationalError("out of memory"), True),
|
||
|
|
(sqlite3.OperationalError("database is locked"), True),
|
||
|
|
(sqlite3.OperationalError("no such table: missing"), False),
|
||
|
|
(OSError(errno.EMFILE, "too many open files"), True),
|
||
|
|
],
|
||
|
|
)
|
||
|
|
def test_transient_errors_without_sqlite_constants(old_sqlite_errors, error, expected):
|
||
|
|
assert _is_transient(error) is expected
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.mark.parametrize("os_error", [False, True])
|
||
|
|
def test_open_recovers_without_sqlite_constants(old_sqlite_errors, tmp_path, os_error):
|
||
|
|
path = tmp_path / "db.sqlite"
|
||
|
|
path.touch()
|
||
|
|
pool = Mock()
|
||
|
|
db = SimpleNamespace(path=str(path), ds=SimpleNamespace(_read_pool_or_none=pool))
|
||
|
|
error = (
|
||
|
|
OSError(errno.EMFILE, "too many open files")
|
||
|
|
if os_error
|
||
|
|
else sqlite3.OperationalError("unable to open database file")
|
||
|
|
)
|
||
|
|
connection = object()
|
||
|
|
connect = Mock(side_effect=[error, connection])
|
||
|
|
assert connect_with_retry(db, connect) is connection
|
||
|
|
assert connect.call_count == 2
|
||
|
|
pool.evict_idle.assert_called_once_with()
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.mark.parametrize("os_error", [False, True])
|
||
|
|
def test_open_failure_preserved_without_sqlite_constants(
|
||
|
|
old_sqlite_errors, tmp_path, os_error
|
||
|
|
):
|
||
|
|
path = tmp_path / "db.sqlite"
|
||
|
|
path.touch()
|
||
|
|
db = SimpleNamespace(path=str(path), ds=SimpleNamespace(_read_pool_or_none=None))
|
||
|
|
error = (
|
||
|
|
OSError(errno.EMFILE, "too many open files")
|
||
|
|
if os_error
|
||
|
|
else sqlite3.OperationalError("unable to open database file")
|
||
|
|
)
|
||
|
|
connect = Mock(side_effect=error)
|
||
|
|
with pytest.raises(
|
||
|
|
DatabaseResourceError if os_error else sqlite3.OperationalError
|
||
|
|
) as exc:
|
||
|
|
connect_with_retry(db, connect)
|
||
|
|
assert (exc.value.__cause__ if os_error else exc.value) is error
|
||
|
|
assert connect.call_count == 2
|