59 lines
1.8 KiB
Go
59 lines
1.8 KiB
Go
|
package boxstore
|
||
|
|
|
||
|
|
import (
|
||
|
|
"os"
|
||
|
|
"path/filepath"
|
||
|
|
"testing"
|
||
|
|
"time"
|
||
|
|
|
||
|
|
"warpbox/lib/models"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestCleanupExpiredBoxesDeletesOnlyExpiredManifestBoxes(t *testing.T) {
|
||
|
|
root := filepath.Join(t.TempDir(), "uploads")
|
||
|
|
previousRoot := UploadRoot()
|
||
|
|
t.Cleanup(func() { SetUploadRoot(previousRoot) })
|
||
|
|
SetUploadRoot(root)
|
||
|
|
|
||
|
|
expiredID := "aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa"
|
||
|
|
activeID := "bbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbb"
|
||
|
|
legacyID := "cccccccccccccccccccccccccccccccc"
|
||
|
|
|
||
|
|
if err := os.MkdirAll(BoxPath(expiredID), 0755); err != nil {
|
||
|
|
t.Fatalf("mkdir expired: %v", err)
|
||
|
|
}
|
||
|
|
if err := os.MkdirAll(BoxPath(activeID), 0755); err != nil {
|
||
|
|
t.Fatalf("mkdir active: %v", err)
|
||
|
|
}
|
||
|
|
if err := os.MkdirAll(BoxPath(legacyID), 0755); err != nil {
|
||
|
|
t.Fatalf("mkdir legacy: %v", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
if err := WriteManifest(expiredID, models.BoxManifest{CreatedAt: time.Now().UTC().Add(-2 * time.Hour), ExpiresAt: time.Now().UTC().Add(-time.Minute)}); err != nil {
|
||
|
|
t.Fatalf("write expired manifest: %v", err)
|
||
|
|
}
|
||
|
|
if err := WriteManifest(activeID, models.BoxManifest{CreatedAt: time.Now().UTC(), ExpiresAt: time.Now().UTC().Add(time.Hour)}); err != nil {
|
||
|
|
t.Fatalf("write active manifest: %v", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
result, err := CleanupExpiredBoxes()
|
||
|
|
if err != nil {
|
||
|
|
t.Fatalf("cleanup failed: %v", err)
|
||
|
|
}
|
||
|
|
if result.Deleted != 1 {
|
||
|
|
t.Fatalf("expected 1 deleted box, got %d", result.Deleted)
|
||
|
|
}
|
||
|
|
if len(result.DeletedIDs) != 1 || result.DeletedIDs[0] != expiredID {
|
||
|
|
t.Fatalf("expected deleted id %s, got %#v", expiredID, result.DeletedIDs)
|
||
|
|
}
|
||
|
|
if _, err := os.Stat(BoxPath(expiredID)); !os.IsNotExist(err) {
|
||
|
|
t.Fatalf("expected expired box dir removed, stat err=%v", err)
|
||
|
|
}
|
||
|
|
if _, err := os.Stat(BoxPath(activeID)); err != nil {
|
||
|
|
t.Fatalf("expected active box to remain, stat err=%v", err)
|
||
|
|
}
|
||
|
|
if _, err := os.Stat(BoxPath(legacyID)); err != nil {
|
||
|
|
t.Fatalf("expected legacy box to remain, stat err=%v", err)
|
||
|
|
}
|
||
|
|
}
|