sneak
417b25a5f5
- Add internal/types package with type-safe wrappers for IDs, hashes, paths, and credentials (FileID, BlobID, ChunkHash, etc.) - Implement driver.Valuer and sql.Scanner for UUID-based types - Add `vaultik version` command showing version, commit, go version - Add `--verify` flag to restore command that checksums all restored files against expected chunk hashes with progress bar - Remove fetch.go (dead code, functionality in restore) - Clean up TODO.md, remove completed items - Update all database and snapshot code to use new custom types
134 lines
3.3 KiB
Go
134 lines
3.3 KiB
Go
package database
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
"time"
|
|
|
|
"git.eeqj.de/sneak/vaultik/internal/types"
|
|
)
|
|
|
|
func TestBlobRepository(t *testing.T) {
|
|
db, cleanup := setupTestDB(t)
|
|
defer cleanup()
|
|
|
|
ctx := context.Background()
|
|
repo := NewBlobRepository(db)
|
|
|
|
// Test Create
|
|
blob := &Blob{
|
|
ID: types.NewBlobID(),
|
|
Hash: types.BlobHash("blobhash123"),
|
|
CreatedTS: time.Now().Truncate(time.Second),
|
|
}
|
|
|
|
err := repo.Create(ctx, nil, blob)
|
|
if err != nil {
|
|
t.Fatalf("failed to create blob: %v", err)
|
|
}
|
|
|
|
// Test GetByHash
|
|
retrieved, err := repo.GetByHash(ctx, blob.Hash.String())
|
|
if err != nil {
|
|
t.Fatalf("failed to get blob: %v", err)
|
|
}
|
|
if retrieved == nil {
|
|
t.Fatal("expected blob, got nil")
|
|
}
|
|
if retrieved.Hash != blob.Hash {
|
|
t.Errorf("blob hash mismatch: got %s, want %s", retrieved.Hash, blob.Hash)
|
|
}
|
|
if !retrieved.CreatedTS.Equal(blob.CreatedTS) {
|
|
t.Errorf("created timestamp mismatch: got %v, want %v", retrieved.CreatedTS, blob.CreatedTS)
|
|
}
|
|
|
|
// Test GetByID
|
|
retrievedByID, err := repo.GetByID(ctx, blob.ID.String())
|
|
if err != nil {
|
|
t.Fatalf("failed to get blob by ID: %v", err)
|
|
}
|
|
if retrievedByID == nil {
|
|
t.Fatal("expected blob, got nil")
|
|
}
|
|
if retrievedByID.ID != blob.ID {
|
|
t.Errorf("blob ID mismatch: got %s, want %s", retrievedByID.ID, blob.ID)
|
|
}
|
|
|
|
// Test with second blob
|
|
blob2 := &Blob{
|
|
ID: types.NewBlobID(),
|
|
Hash: types.BlobHash("blobhash456"),
|
|
CreatedTS: time.Now().Truncate(time.Second),
|
|
}
|
|
err = repo.Create(ctx, nil, blob2)
|
|
if err != nil {
|
|
t.Fatalf("failed to create second blob: %v", err)
|
|
}
|
|
|
|
// Test UpdateFinished
|
|
now := time.Now()
|
|
err = repo.UpdateFinished(ctx, nil, blob.ID.String(), blob.Hash.String(), 1000, 500)
|
|
if err != nil {
|
|
t.Fatalf("failed to update blob as finished: %v", err)
|
|
}
|
|
|
|
// Verify update
|
|
updated, err := repo.GetByID(ctx, blob.ID.String())
|
|
if err != nil {
|
|
t.Fatalf("failed to get updated blob: %v", err)
|
|
}
|
|
if updated.FinishedTS == nil {
|
|
t.Fatal("expected finished timestamp to be set")
|
|
}
|
|
if updated.UncompressedSize != 1000 {
|
|
t.Errorf("expected uncompressed size 1000, got %d", updated.UncompressedSize)
|
|
}
|
|
if updated.CompressedSize != 500 {
|
|
t.Errorf("expected compressed size 500, got %d", updated.CompressedSize)
|
|
}
|
|
|
|
// Test UpdateUploaded
|
|
err = repo.UpdateUploaded(ctx, nil, blob.ID.String())
|
|
if err != nil {
|
|
t.Fatalf("failed to update blob as uploaded: %v", err)
|
|
}
|
|
|
|
// Verify upload update
|
|
uploaded, err := repo.GetByID(ctx, blob.ID.String())
|
|
if err != nil {
|
|
t.Fatalf("failed to get uploaded blob: %v", err)
|
|
}
|
|
if uploaded.UploadedTS == nil {
|
|
t.Fatal("expected uploaded timestamp to be set")
|
|
}
|
|
// Allow 1 second tolerance for timestamp comparison
|
|
if uploaded.UploadedTS.Before(now.Add(-1 * time.Second)) {
|
|
t.Error("uploaded timestamp should be around test time")
|
|
}
|
|
}
|
|
|
|
func TestBlobRepositoryDuplicate(t *testing.T) {
|
|
db, cleanup := setupTestDB(t)
|
|
defer cleanup()
|
|
|
|
ctx := context.Background()
|
|
repo := NewBlobRepository(db)
|
|
|
|
blob := &Blob{
|
|
ID: types.NewBlobID(),
|
|
Hash: types.BlobHash("duplicate_blob"),
|
|
CreatedTS: time.Now().Truncate(time.Second),
|
|
}
|
|
|
|
err := repo.Create(ctx, nil, blob)
|
|
if err != nil {
|
|
t.Fatalf("failed to create blob: %v", err)
|
|
}
|
|
|
|
// Try to create duplicate - should fail due to unique constraint
|
|
err = repo.Create(ctx, nil, blob)
|
|
if err == nil {
|
|
t.Error("expected error for duplicate blob")
|
|
}
|
|
}
|