sneak
86b533d6ee
- Changed blob table to use ID (UUID) as primary key instead of hash - Blob records are now created at packing start, enabling immediate chunk associations - Implemented streaming chunking to process large files without memory exhaustion - Fixed blob manifest generation to include all referenced blobs - Updated all foreign key references from blob_hash to blob_id - Added progress reporting and improved error handling - Enforced encryption requirement for all blob packing - Updated tests to use test encryption keys - Added Cyrillic transliteration to README
132 lines
3.2 KiB
Go
132 lines
3.2 KiB
Go
package database
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
func TestBlobRepository(t *testing.T) {
|
|
db, cleanup := setupTestDB(t)
|
|
defer cleanup()
|
|
|
|
ctx := context.Background()
|
|
repo := NewBlobRepository(db)
|
|
|
|
// Test Create
|
|
blob := &Blob{
|
|
ID: "test-blob-id-123",
|
|
Hash: "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)
|
|
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)
|
|
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: "test-blob-id-456",
|
|
Hash: "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, blob.Hash, 1000, 500)
|
|
if err != nil {
|
|
t.Fatalf("failed to update blob as finished: %v", err)
|
|
}
|
|
|
|
// Verify update
|
|
updated, err := repo.GetByID(ctx, blob.ID)
|
|
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)
|
|
if err != nil {
|
|
t.Fatalf("failed to update blob as uploaded: %v", err)
|
|
}
|
|
|
|
// Verify upload update
|
|
uploaded, err := repo.GetByID(ctx, blob.ID)
|
|
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: "duplicate-test-id",
|
|
Hash: "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")
|
|
}
|
|
}
|