sneak
d3afa65420
- Add unified compression/encryption package in internal/blobgen - Update DATAMODEL.md to reflect current schema implementation - Refactor snapshot cleanup into well-named methods for clarity - Add snapshot_id to uploads table to track new blobs per snapshot - Fix blob count reporting for incremental backups - Add DeleteOrphaned method to BlobChunkRepository - Fix cleanup order to respect foreign key constraints - Update tests to reflect schema changes
102 lines
2.2 KiB
Go
102 lines
2.2 KiB
Go
package database
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"path/filepath"
|
|
"testing"
|
|
)
|
|
|
|
func TestDatabase(t *testing.T) {
|
|
ctx := context.Background()
|
|
dbPath := filepath.Join(t.TempDir(), "test.db")
|
|
|
|
db, err := New(ctx, dbPath)
|
|
if err != nil {
|
|
t.Fatalf("failed to create database: %v", err)
|
|
}
|
|
defer func() {
|
|
if err := db.Close(); err != nil {
|
|
t.Errorf("failed to close database: %v", err)
|
|
}
|
|
}()
|
|
|
|
// Test connection
|
|
if db.Conn() == nil {
|
|
t.Fatal("database connection is nil")
|
|
}
|
|
|
|
// Test schema creation (already done in New)
|
|
// Verify tables exist
|
|
tables := []string{
|
|
"files", "file_chunks", "chunks", "blobs",
|
|
"blob_chunks", "chunk_files", "snapshots",
|
|
}
|
|
|
|
for _, table := range tables {
|
|
var name string
|
|
err := db.conn.QueryRow("SELECT name FROM sqlite_master WHERE type='table' AND name=?", table).Scan(&name)
|
|
if err != nil {
|
|
t.Errorf("table %s does not exist: %v", table, err)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestDatabaseInvalidPath(t *testing.T) {
|
|
ctx := context.Background()
|
|
|
|
// Test with invalid path
|
|
_, err := New(ctx, "/invalid/path/that/does/not/exist/test.db")
|
|
if err == nil {
|
|
t.Fatal("expected error for invalid path")
|
|
}
|
|
}
|
|
|
|
func TestDatabaseConcurrentAccess(t *testing.T) {
|
|
ctx := context.Background()
|
|
dbPath := filepath.Join(t.TempDir(), "test.db")
|
|
|
|
db, err := New(ctx, dbPath)
|
|
if err != nil {
|
|
t.Fatalf("failed to create database: %v", err)
|
|
}
|
|
defer func() {
|
|
if err := db.Close(); err != nil {
|
|
t.Errorf("failed to close database: %v", err)
|
|
}
|
|
}()
|
|
|
|
// Test concurrent writes
|
|
type result struct {
|
|
index int
|
|
err error
|
|
}
|
|
results := make(chan result, 10)
|
|
|
|
for i := 0; i < 10; i++ {
|
|
go func(i int) {
|
|
_, err := db.ExecWithLog(ctx, "INSERT INTO chunks (chunk_hash, size) VALUES (?, ?)",
|
|
fmt.Sprintf("hash%d", i), i*1024)
|
|
results <- result{index: i, err: err}
|
|
}(i)
|
|
}
|
|
|
|
// Wait for all goroutines and check results
|
|
for i := 0; i < 10; i++ {
|
|
r := <-results
|
|
if r.err != nil {
|
|
t.Fatalf("concurrent insert %d failed: %v", r.index, r.err)
|
|
}
|
|
}
|
|
|
|
// Verify all inserts succeeded
|
|
var count int
|
|
err = db.conn.QueryRowContext(ctx, "SELECT COUNT(*) FROM chunks").Scan(&count)
|
|
if err != nil {
|
|
t.Fatalf("failed to count chunks: %v", err)
|
|
}
|
|
if count != 10 {
|
|
t.Errorf("expected 10 chunks, got %d", count)
|
|
}
|
|
}
|