tests: remove unnecessary service test
This commit is contained in:
talksik
@@ -1,250 +0,0 @@
|
||||
package depot_test
|
||||
|
||||
import (
|
||||
"context"
|
||||
"os"
|
||||
"testing"
|
||||
"time"
|
||||
|
||||
"github.com/flowy-live/llink/internal/depot"
|
||||
"github.com/flowy-live/llink/internal/testhelper"
|
||||
"github.com/jackc/pgx/v5/pgxpool"
|
||||
"github.com/stretchr/testify/assert"
|
||||
)
|
||||
|
||||
var dbPool *pgxpool.Pool
|
||||
|
||||
func TestMain(m *testing.M) {
|
||||
dbPool = testhelper.SetupTestDB()
|
||||
defer testhelper.TeardownTestDB()
|
||||
|
||||
ret := m.Run()
|
||||
os.Exit(ret)
|
||||
}
|
||||
|
||||
// TestDepotRepository tests the repository layer directly
|
||||
// These tests can run without GCS since they only test database operations
|
||||
func TestDepotRepository_CreateAndGet(t *testing.T) {
|
||||
ctx := context.Background()
|
||||
|
||||
// Create a depot object directly in the database for testing
|
||||
var id string
|
||||
err := dbPool.QueryRow(ctx,
|
||||
`INSERT INTO depot_objects (id, name, content_type, content_length, bucket_name, object_key, contains_content)
|
||||
VALUES ($1, $2, $3, $4, $5, $6, $7)
|
||||
RETURNING id`,
|
||||
"dpo_test123", "test-file.txt", "text/plain", int64(1024), "test-bucket", "test-key-123", false,
|
||||
).Scan(&id)
|
||||
assert.NoError(t, err)
|
||||
assert.Equal(t, "dpo_test123", id)
|
||||
|
||||
// Query the object back
|
||||
var obj struct {
|
||||
ID string
|
||||
Name string
|
||||
ContentType string
|
||||
ContentLength int64
|
||||
BucketName string
|
||||
ObjectKey string
|
||||
ContainsContent bool
|
||||
CreatedAt time.Time
|
||||
}
|
||||
err = dbPool.QueryRow(ctx,
|
||||
`SELECT id, name, content_type, content_length, bucket_name, object_key, contains_content, created_at
|
||||
FROM depot_objects WHERE id = $1`,
|
||||
id,
|
||||
).Scan(&obj.ID, &obj.Name, &obj.ContentType, &obj.ContentLength,
|
||||
&obj.BucketName, &obj.ObjectKey, &obj.ContainsContent, &obj.CreatedAt)
|
||||
assert.NoError(t, err)
|
||||
assert.Equal(t, "test-file.txt", obj.Name)
|
||||
assert.Equal(t, "text/plain", obj.ContentType)
|
||||
assert.Equal(t, int64(1024), obj.ContentLength)
|
||||
assert.Equal(t, "test-bucket", obj.BucketName)
|
||||
assert.Equal(t, "test-key-123", obj.ObjectKey)
|
||||
assert.False(t, obj.ContainsContent)
|
||||
assert.False(t, obj.CreatedAt.IsZero())
|
||||
|
||||
// Clean up
|
||||
_, err = dbPool.Exec(ctx, `DELETE FROM depot_objects WHERE id = $1`, id)
|
||||
assert.NoError(t, err)
|
||||
}
|
||||
|
||||
func TestDepotRepository_ConfirmUpload(t *testing.T) {
|
||||
ctx := context.Background()
|
||||
|
||||
// Create a depot object
|
||||
var id string
|
||||
err := dbPool.QueryRow(ctx,
|
||||
`INSERT INTO depot_objects (id, name, content_type, content_length, bucket_name, object_key, contains_content)
|
||||
VALUES ($1, $2, $3, $4, $5, $6, $7)
|
||||
RETURNING id`,
|
||||
"dpo_confirm123", "confirm-file.txt", "text/plain", int64(2048), "test-bucket", "confirm-key", false,
|
||||
).Scan(&id)
|
||||
assert.NoError(t, err)
|
||||
|
||||
// Verify it starts with contains_content = false
|
||||
var containsContent bool
|
||||
err = dbPool.QueryRow(ctx,
|
||||
`SELECT contains_content FROM depot_objects WHERE id = $1`,
|
||||
id,
|
||||
).Scan(&containsContent)
|
||||
assert.NoError(t, err)
|
||||
assert.False(t, containsContent)
|
||||
|
||||
// Confirm upload
|
||||
_, err = dbPool.Exec(ctx,
|
||||
`UPDATE depot_objects SET contains_content = TRUE WHERE id = $1`,
|
||||
id,
|
||||
)
|
||||
assert.NoError(t, err)
|
||||
|
||||
// Verify it's now true
|
||||
err = dbPool.QueryRow(ctx,
|
||||
`SELECT contains_content FROM depot_objects WHERE id = $1`,
|
||||
id,
|
||||
).Scan(&containsContent)
|
||||
assert.NoError(t, err)
|
||||
assert.True(t, containsContent)
|
||||
|
||||
// Clean up
|
||||
_, err = dbPool.Exec(ctx, `DELETE FROM depot_objects WHERE id = $1`, id)
|
||||
assert.NoError(t, err)
|
||||
}
|
||||
|
||||
func TestDepotRepository_Delete(t *testing.T) {
|
||||
ctx := context.Background()
|
||||
|
||||
// Create a depot object
|
||||
var id string
|
||||
err := dbPool.QueryRow(ctx,
|
||||
`INSERT INTO depot_objects (id, name, content_type, content_length, bucket_name, object_key, contains_content)
|
||||
VALUES ($1, $2, $3, $4, $5, $6, $7)
|
||||
RETURNING id`,
|
||||
"dpo_delete123", "delete-file.txt", "text/plain", int64(512), "test-bucket", "delete-key", false,
|
||||
).Scan(&id)
|
||||
assert.NoError(t, err)
|
||||
|
||||
// Delete it
|
||||
result, err := dbPool.Exec(ctx, `DELETE FROM depot_objects WHERE id = $1`, id)
|
||||
assert.NoError(t, err)
|
||||
assert.Equal(t, int64(1), result.RowsAffected())
|
||||
|
||||
// Verify it's gone
|
||||
var count int
|
||||
err = dbPool.QueryRow(ctx,
|
||||
`SELECT COUNT(*) FROM depot_objects WHERE id = $1`,
|
||||
id,
|
||||
).Scan(&count)
|
||||
assert.NoError(t, err)
|
||||
assert.Equal(t, 0, count)
|
||||
}
|
||||
|
||||
func TestDepotRepository_Exists(t *testing.T) {
|
||||
ctx := context.Background()
|
||||
|
||||
// Create a depot object
|
||||
var id string
|
||||
err := dbPool.QueryRow(ctx,
|
||||
`INSERT INTO depot_objects (id, name, content_type, content_length, bucket_name, object_key, contains_content)
|
||||
VALUES ($1, $2, $3, $4, $5, $6, $7)
|
||||
RETURNING id`,
|
||||
"dpo_exists123", "exists-file.txt", "text/plain", int64(256), "test-bucket", "exists-key", true,
|
||||
).Scan(&id)
|
||||
assert.NoError(t, err)
|
||||
|
||||
// Check exists
|
||||
var exists bool
|
||||
err = dbPool.QueryRow(ctx,
|
||||
`SELECT EXISTS(SELECT 1 FROM depot_objects WHERE id = $1)`,
|
||||
id,
|
||||
).Scan(&exists)
|
||||
assert.NoError(t, err)
|
||||
assert.True(t, exists)
|
||||
|
||||
// Check non-existent
|
||||
err = dbPool.QueryRow(ctx,
|
||||
`SELECT EXISTS(SELECT 1 FROM depot_objects WHERE id = $1)`,
|
||||
"dpo_nonexistent",
|
||||
).Scan(&exists)
|
||||
assert.NoError(t, err)
|
||||
assert.False(t, exists)
|
||||
|
||||
// Clean up
|
||||
_, err = dbPool.Exec(ctx, `DELETE FROM depot_objects WHERE id = $1`, id)
|
||||
assert.NoError(t, err)
|
||||
}
|
||||
|
||||
func TestDepotRepository_OrphanedIndex(t *testing.T) {
|
||||
ctx := context.Background()
|
||||
|
||||
// Create an orphaned object (contains_content = false)
|
||||
var orphanID string
|
||||
err := dbPool.QueryRow(ctx,
|
||||
`INSERT INTO depot_objects (id, name, content_type, content_length, bucket_name, object_key, contains_content)
|
||||
VALUES ($1, $2, $3, $4, $5, $6, $7)
|
||||
RETURNING id`,
|
||||
"dpo_orphan123", "orphan-file.txt", "text/plain", int64(128), "test-bucket", "orphan-key", false,
|
||||
).Scan(&orphanID)
|
||||
assert.NoError(t, err)
|
||||
|
||||
// Create a confirmed object (contains_content = true)
|
||||
var confirmedID string
|
||||
err = dbPool.QueryRow(ctx,
|
||||
`INSERT INTO depot_objects (id, name, content_type, content_length, bucket_name, object_key, contains_content)
|
||||
VALUES ($1, $2, $3, $4, $5, $6, $7)
|
||||
RETURNING id`,
|
||||
"dpo_confirmed123", "confirmed-file.txt", "text/plain", int64(128), "test-bucket", "confirmed-key", true,
|
||||
).Scan(&confirmedID)
|
||||
assert.NoError(t, err)
|
||||
|
||||
// Query orphaned objects using the index
|
||||
rows, err := dbPool.Query(ctx,
|
||||
`SELECT id FROM depot_objects WHERE contains_content = FALSE`)
|
||||
assert.NoError(t, err)
|
||||
defer rows.Close()
|
||||
|
||||
var orphanedIDs []string
|
||||
for rows.Next() {
|
||||
var id string
|
||||
err := rows.Scan(&id)
|
||||
assert.NoError(t, err)
|
||||
orphanedIDs = append(orphanedIDs, id)
|
||||
}
|
||||
|
||||
// Our orphan should be in the list
|
||||
assert.Contains(t, orphanedIDs, orphanID)
|
||||
assert.NotContains(t, orphanedIDs, confirmedID)
|
||||
|
||||
// Clean up
|
||||
_, err = dbPool.Exec(ctx, `DELETE FROM depot_objects WHERE id IN ($1, $2)`, orphanID, confirmedID)
|
||||
assert.NoError(t, err)
|
||||
}
|
||||
|
||||
// TestDepotModels tests the model structures
|
||||
func TestDepotModels(t *testing.T) {
|
||||
// Test Config defaults
|
||||
config := depot.Config{
|
||||
BucketName: "test-bucket",
|
||||
}
|
||||
assert.Equal(t, "test-bucket", config.BucketName)
|
||||
assert.Equal(t, time.Duration(0), config.UploadURLExpiry)
|
||||
assert.Equal(t, time.Duration(0), config.DownloadURLExpiry)
|
||||
|
||||
// Test with explicit values
|
||||
config = depot.Config{
|
||||
BucketName: "custom-bucket",
|
||||
UploadURLExpiry: 10 * time.Minute,
|
||||
DownloadURLExpiry: 12 * time.Hour,
|
||||
}
|
||||
assert.Equal(t, "custom-bucket", config.BucketName)
|
||||
assert.Equal(t, 10*time.Minute, config.UploadURLExpiry)
|
||||
assert.Equal(t, 12*time.Hour, config.DownloadURLExpiry)
|
||||
}
|
||||
|
||||
// TestDepotErrors tests the error definitions
|
||||
func TestDepotErrors(t *testing.T) {
|
||||
assert.Error(t, depot.ErrNotFound)
|
||||
assert.Error(t, depot.ErrInvalidInput)
|
||||
assert.Equal(t, "object not found", depot.ErrNotFound.Error())
|
||||
assert.Equal(t, "invalid input", depot.ErrInvalidInput.Error())
|
||||
}
|
||||
Reference in New Issue
Block a user