summaryrefslogtreecommitdiff
path: root/internal/storage
diff options
context:
space:
mode:
authorPaul Buetow <paul@buetow.org>2026-04-14 10:40:44 +0300
committerPaul Buetow <paul@buetow.org>2026-04-14 10:40:44 +0300
commitd49376f71dd8defebf80f901ecb60ab99f0f4906 (patch)
tree749322d2bf5c33d28ff95b77ebc674d69435c31c /internal/storage
parentff749457e392288e29dbf553bf8e0e64cc8b6401 (diff)
n3: validate SQLite with PingContext after Open
Call db.PingContext in storage.Open and authkeys.OpenStore; close DB and return wrapped errors on failure. Add tests for canceled context and invalid directory path. Made-with: Cursor
Diffstat (limited to 'internal/storage')
-rw-r--r--internal/storage/db.go8
-rw-r--r--internal/storage/db_test.go34
2 files changed, 40 insertions, 2 deletions
diff --git a/internal/storage/db.go b/internal/storage/db.go
index 14d7050..7b577e7 100644
--- a/internal/storage/db.go
+++ b/internal/storage/db.go
@@ -40,11 +40,15 @@ type Record struct {
func Open(ctx context.Context, path string) (*sql.DB, error) {
db, err := sql.Open("sqlite", path)
if err != nil {
- return nil, err
+ return nil, fmt.Errorf("open sqlite: %w", err)
+ }
+ if err := db.PingContext(ctx); err != nil {
+ db.Close()
+ return nil, fmt.Errorf("ping sqlite: %w", err)
}
if _, err := db.ExecContext(ctx, "PRAGMA foreign_keys = OFF"); err != nil {
db.Close()
- return nil, err
+ return nil, fmt.Errorf("pragma foreign_keys: %w", err)
}
return db, nil
}
diff --git a/internal/storage/db_test.go b/internal/storage/db_test.go
new file mode 100644
index 0000000..5415235
--- /dev/null
+++ b/internal/storage/db_test.go
@@ -0,0 +1,34 @@
+package storage
+
+import (
+ "context"
+ "errors"
+ "os"
+ "path/filepath"
+ "testing"
+)
+
+func TestOpen_ContextCanceled(t *testing.T) {
+ ctx, cancel := context.WithCancel(context.Background())
+ cancel()
+ path := filepath.Join(t.TempDir(), "records.db")
+ _, err := Open(ctx, path)
+ if err == nil {
+ t.Fatal("expected error")
+ }
+ if !errors.Is(err, context.Canceled) {
+ t.Fatalf("expected context.Canceled, got %v", err)
+ }
+}
+
+func TestOpen_PingFailsOnDirectoryPath(t *testing.T) {
+ ctx := context.Background()
+ dir := filepath.Join(t.TempDir(), "isdir")
+ if err := os.Mkdir(dir, 0o755); err != nil {
+ t.Fatal(err)
+ }
+ _, err := Open(ctx, dir)
+ if err == nil {
+ t.Fatal("expected error opening sqlite at directory path")
+ }
+}