weichen
2de4eda16d
Track workspace size and age, expose an admin list and cleanup route, and refuse TTL or quota deletion until a frozen manifest matches verified receipts.
101 lines
2.8 KiB
Go
101 lines
2.8 KiB
Go
package workspace
|
|
|
|
import (
|
|
"context"
|
|
"os"
|
|
"path/filepath"
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
func sampleFrozen(id string) (FrozenManifest, []RetentionReceipt) {
|
|
frozen := FrozenManifest{
|
|
SHA256: "abc",
|
|
StorageIdentity: "disk",
|
|
WorkspaceID: id,
|
|
Members: []FrozenMember{{LogicalID: "doc", Digest: "d1", Kind: "document"}},
|
|
}
|
|
receipts := []RetentionReceipt{{LogicalID: "doc", Digest: "d1", StorageIdentity: "disk", Verified: true}}
|
|
return frozen, receipts
|
|
}
|
|
|
|
func TestGCDeletesExpiredWhenReceiptsMatch(t *testing.T) {
|
|
root := t.TempDir()
|
|
id := "ws-old"
|
|
dir := filepath.Join(root, "workspaces", id)
|
|
if err := os.MkdirAll(dir, 0o755); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if err := os.WriteFile(filepath.Join(dir, "output.md"), []byte("kept-until-gc"), 0o644); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
reg := NewFileRegistry(root)
|
|
frozen, receipts := sampleFrozen(id)
|
|
if err := reg.Upsert(Record{ID: id, CreatedAt: time.Now().Add(-8 * 24 * time.Hour), LastSeen: time.Now().Add(-8 * 24 * time.Hour)}); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if err := reg.SetRetention(id, frozen, receipts); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
report, err := reg.CollectGarbage(context.Background(), time.Now(), 7*24*time.Hour, 0, nil)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if len(report.Deleted) != 1 || report.Deleted[0] != id {
|
|
t.Fatalf("report %#v", report)
|
|
}
|
|
if _, err := os.Stat(dir); !os.IsNotExist(err) {
|
|
t.Fatal("workspace directory still exists")
|
|
}
|
|
}
|
|
|
|
func TestGCBlocksWithoutReceipts(t *testing.T) {
|
|
root := t.TempDir()
|
|
id := "ws-blocked"
|
|
dir := filepath.Join(root, "workspaces", id)
|
|
if err := os.MkdirAll(dir, 0o755); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
reg := NewFileRegistry(root)
|
|
if err := reg.Upsert(Record{ID: id, CreatedAt: time.Now().Add(-30 * 24 * time.Hour)}); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
report, err := reg.CollectGarbage(context.Background(), time.Now(), time.Hour, 1, nil)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if len(report.Deleted) != 0 || len(report.Blocked) != 1 {
|
|
t.Fatalf("report %#v", report)
|
|
}
|
|
if _, err := os.Stat(dir); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|
|
|
|
func TestDiscoverAndNote(t *testing.T) {
|
|
root := t.TempDir()
|
|
id := "ws-scan"
|
|
if err := os.MkdirAll(filepath.Join(root, "workspaces", id), 0o755); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
reg := NewFileRegistry(root)
|
|
if err := reg.Discover(); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
rec, ok, err := reg.Get(id)
|
|
if err != nil || !ok || rec.ID != id {
|
|
t.Fatalf("discovered %#v %v %v", rec, ok, err)
|
|
}
|
|
nested := filepath.Join(root, "workspaces", "ws-note", "tasks")
|
|
if err := os.MkdirAll(nested, 0o755); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if err := NoteWorkspace(root, nested, "run-1"); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
noted, ok, err := reg.Get("ws-note")
|
|
if err != nil || !ok || noted.RunID != "run-1" {
|
|
t.Fatalf("noted %#v %v", noted, err)
|
|
}
|
|
}
|