1 files changed, 70 insertions, 0 deletions

diff --git a/internal/storage/storage_test.go b/internal/storage/storage_test.go
new file mode 100644
index 0000000..f2fe401
--- /dev/null
+++ b/internal/storage/storage_test.go
@@ -0,0 +1,70 @@
+package storage
+
+import (
+        "testing"
+)
+
+func TestNew(t *testing.T) {
+        // Test in-memory database
+        store, err := New(":memory:")
+        if err != nil {
+                t.Fatalf("failed to create storage: %v", err)
+        }
+        defer store.Close()
+
+        // Verify database is accessible
+        if store.DB() == nil {
+                t.Fatal("DB() returned nil")
+        }
+
+        // Verify schema was created by checking if tables exist
+        var count int
+        query := `SELECT COUNT(*) FROM sqlite_master WHERE type='table' AND name IN ('events', 'deletions', 'replaceable_events')`
+        err = store.DB().QueryRow(query).Scan(&count)
+        if err != nil {
+                t.Fatalf("failed to query tables: %v", err)
+        }
+
+        if count != 3 {
+                t.Errorf("expected 3 main tables, got %d", count)
+        }
+}
+
+func TestNewFileDatabase(t *testing.T) {
+        // Test file-based database
+        dbPath := t.TempDir() + "/test.db"
+        store, err := New(dbPath)
+        if err != nil {
+                t.Fatalf("failed to create file-based storage: %v", err)
+        }
+        defer store.Close()
+
+        // Verify WAL mode is enabled
+        var walMode string
+        err = store.DB().QueryRow("PRAGMA journal_mode").Scan(&walMode)
+        if err != nil {
+                t.Fatalf("failed to query journal mode: %v", err)
+        }
+
+        if walMode != "wal" {
+                t.Errorf("expected WAL mode, got %s", walMode)
+        }
+}
+
+func TestClose(t *testing.T) {
+        store, err := New(":memory:")
+        if err != nil {
+                t.Fatalf("failed to create storage: %v", err)
+        }
+
+        err = store.Close()
+        if err != nil {
+                t.Errorf("failed to close storage: %v", err)
+        }
+
+        // Verify database is closed by attempting a query
+        err = store.DB().Ping()
+        if err == nil {
+                t.Error("expected error when pinging closed database")
+        }
+}