test updates

Author: thejchap

pyproject.toml

@@ -40,9 +40,13 @@ indent-width = 4
 target-version = "py312"
 
 [tool.ruff.lint.per-file-ignores]
+# tryke's Depends() pattern requires function-call defaults on test params
 "tests/*.py" = ["ANN201", "PLR2004"]
 "src/sapling/backends/sqlite.py" = ["S608"]
 
+[tool.ruff.lint.flake8-bugbear]
+extend-immutable-calls = ["tryke.Depends"]
+
 [tool.ruff.lint.pycodestyle]
 max-line-length = 88
 

src/sapling/database.py

@@ -1,15 +1,10 @@
 from __future__ import annotations
 
-import tempfile
-from pathlib import Path
 from typing import TYPE_CHECKING
 
 from pydantic import BaseModel
-from tryke import describe, expect, test
 
-from sapling import MemoryBackend, SQLiteBackend
-from sapling.errors import NotFoundError
-from sapling.settings import SaplingSettings
+from sapling import SQLiteBackend
 
 if TYPE_CHECKING:
     from collections.abc import Generator, Iterator
@@ -247,148 +242,3 @@ def put_many[T: BaseModel](
         """
         with self.transaction() as txn:
             return txn.put_many(model_class, models)
-
-
-class _TestModel(BaseModel):
-    hello: str = "world"
-
-
-@test
-def test_basic() -> None:
-    db = Database()
-    hello = _TestModel()
-    with db.transaction() as txn:
-        pk = "hello"
-        record = txn.put(_TestModel, pk, hello)
-        expect(record.model_id).to_equal(pk)
-        maybe_record = txn.get(_TestModel, pk)
-        expect(maybe_record).to_be_truthy()
-        record = txn.fetch(_TestModel, pk)
-        txn.delete(_TestModel, pk)
-        expect(lambda: txn.fetch(_TestModel, pk)).to_raise(NotFoundError)
-
-
-@test
-def test_all_method() -> None:
-    db = Database()
-    with db.transaction() as txn:
-        txn.put(_TestModel, "1", _TestModel(hello="one"))
-        txn.put(_TestModel, "2", _TestModel(hello="two"))
-        txn.put(_TestModel, "3", _TestModel(hello="three"))
-
-        all_hellos = txn.all(_TestModel)
-        expect(all_hellos).to_have_length(3)
-        expect({h.model_id for h in all_hellos}).to_equal({"1", "2", "3"})
-        expect({h.model.hello for h in all_hellos}).to_equal({"one", "two", "three"})
-
-
-@test
-def test_all_empty() -> None:
-    db = Database()
-    with db.transaction() as txn:
-        all_hellos = txn.all(_TestModel)
-        expect(all_hellos).to_equal([])
-
-
-@test
-def test_backend_all_method() -> None:
-    backend = SQLiteBackend()
-    db = Database(backend=backend)
-
-    with db.transaction() as txn:
-        txn.put(_TestModel, "a", _TestModel(hello="alpha"))
-        txn.put(_TestModel, "b", _TestModel(hello="beta"))
-
-        all_docs = txn.all(_TestModel)
-        expect(all_docs).to_have_length(2)
-        expect({d.model_id for d in all_docs}).to_equal({"a", "b"})
-
-
-@test
-def test_memory_backend() -> None:
-    backend = MemoryBackend()
-    db = Database(backend=backend)
-
-    with db.transaction() as txn:
-        txn.put(_TestModel, "test", _TestModel(hello="world"))
-        doc = txn.fetch(_TestModel, "test")
-        expect(doc.model.hello).to_equal("world")
-
-        txn.put(_TestModel, "1", _TestModel(hello="one"))
-        txn.put(_TestModel, "2", _TestModel(hello="two"))
-
-        all_docs = txn.all(_TestModel)
-        expect(all_docs).to_have_length(3)
-        expect({d.model_id for d in all_docs}).to_equal({"test", "1", "2"})
-
-        txn.delete(_TestModel, "test")
-        expect(txn.get(_TestModel, "test")).to_be_none()
-
-        expect(lambda: txn.fetch(_TestModel, "test")).to_raise(NotFoundError)
-
-
-with describe("sqlite"):
-
-    @test
-    def test_sqlite_backend_memory() -> None:
-        backend = SQLiteBackend()
-        db = Database(backend=backend)
-        with db.transaction() as txn:
-            txn.put(_TestModel, "test", _TestModel(hello="world"))
-            doc = txn.fetch(_TestModel, "test")
-            expect(doc.model.hello).to_equal("world")
-
-    @test
-    def test_sqlite_backend_file() -> None:
-        with tempfile.TemporaryDirectory() as tmp_dir:
-            db_path = Path(tmp_dir) / "test.db"
-            settings = SaplingSettings(sqlite_path=str(db_path))
-            backend = SQLiteBackend(settings=settings)
-            db = Database(backend=backend)
-
-            with db.transaction() as txn:
-                txn.put(_TestModel, "persistent", _TestModel(hello="saved"))
-
-            settings2 = SaplingSettings(sqlite_path=str(db_path))
-            db2 = Database(backend=SQLiteBackend(settings=settings2))
-            with db2.transaction() as txn:
-                doc = txn.fetch(_TestModel, "persistent")
-                expect(doc.model.hello).to_equal("saved")
-
-
-with describe("initialization"):
-
-    @test
-    def test_deferred_initialization() -> None:
-        backend = SQLiteBackend()
-        db = Database(backend=backend, initialize=False)
-
-        db.initialize()
-
-        with db.transaction() as txn:
-            txn.put(_TestModel, "test", _TestModel(hello="world"))
-            doc = txn.fetch(_TestModel, "test")
-            expect(doc.model.hello).to_equal("world")
-
-    @test
-    def test_idempotent_initialization() -> None:
-        backend = SQLiteBackend()
-        db = Database(backend=backend, initialize=False)
-
-        db.initialize()
-        db.initialize()
-        db.initialize()
-
-        with db.transaction() as txn:
-            txn.put(_TestModel, "test", _TestModel(hello="world"))
-
-    @test
-    def test_uninitialized_error() -> None:
-        backend = SQLiteBackend()
-        db = Database(backend=backend, initialize=False)
-
-        def try_uninitialized() -> None:
-            with db.transaction() as txn:
-                txn.put(_TestModel, "test", _TestModel(hello="world"))
-
-        expect(try_uninitialized).to_raise(ValueError, match="not initialized")