mirror of
https://github.com/hanzoai/orm.git
synced 2026-10-30 15:01:22 +00:00
zeekay
b5ddfc9185
Wrap an already-open *sql.DB (pragma'd, encrypted at rest, single-writer) as an orm.DB/SQLiteDB instead of opening one from a path. The ORM manages records in a file whose lifecycle the caller owns: one connection serves reads and writes (serialized by writeMu), initSchema ensures _entities, and Close is a no-op so the caller closes the connection it opened. Additive; NewSQLiteDB is unchanged.
45 lines
1.1 KiB
Go
45 lines
1.1 KiB
Go
package orm
|
|
|
|
import (
|
|
"context"
|
|
"database/sql"
|
|
"path/filepath"
|
|
"testing"
|
|
|
|
_ "github.com/hanzoai/sqlite"
|
|
)
|
|
|
|
// TestAdaptSQLite_RootWrapsConn: the root orm.DB works over a caller-owned *sql.DB,
|
|
// and closing it leaves the caller's connection open (borrow, not own).
|
|
func TestAdaptSQLite_RootWrapsConn(t *testing.T) {
|
|
path := filepath.Join(t.TempDir(), "root.db")
|
|
conn, err := sql.Open("sqlite", path)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
conn.SetMaxOpenConns(1)
|
|
defer func() { _ = conn.Close() }()
|
|
|
|
db, err := AdaptSQLite(conn)
|
|
if err != nil {
|
|
t.Fatalf("AdaptSQLite: %v", err)
|
|
}
|
|
ctx := context.Background()
|
|
created, err := db.CreateIfAbsent(ctx, db.NewKey("k", "1", 0, nil), map[string]any{"v": 1})
|
|
if err != nil || !created {
|
|
t.Fatalf("CreateIfAbsent created=%v err=%v, want true nil", created, err)
|
|
}
|
|
if err := db.Close(); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if err := conn.PingContext(ctx); err != nil {
|
|
t.Fatalf("AdaptSQLite Close closed the caller's conn: %v", err)
|
|
}
|
|
}
|
|
|
|
func TestAdaptSQLite_NilConn(t *testing.T) {
|
|
if _, err := AdaptSQLite(nil); err == nil {
|
|
t.Fatal("AdaptSQLite(nil) = nil error, want error")
|
|
}
|
|
}
|