wehub-resource-sync
84 lines
2.4 KiB
Go
84 lines
2.4 KiB
Go
package agent
|
|
|
|
import (
|
|
"errors"
|
|
"os"
|
|
"path/filepath"
|
|
"testing"
|
|
|
|
"reasonix/internal/provider"
|
|
"reasonix/internal/store"
|
|
)
|
|
|
|
func TestSessionRemovalGuardBlocksWhileLeaseHeld(t *testing.T) {
|
|
dir := t.TempDir()
|
|
path := filepath.Join(dir, "session.jsonl")
|
|
s := NewSession("sys")
|
|
s.Add(provider.Message{Role: provider.RoleUser, Content: "work"})
|
|
if err := s.SaveSnapshot(path); err != nil {
|
|
t.Fatalf("SaveSnapshot: %v", err)
|
|
}
|
|
|
|
lease, err := TryAcquireSessionLease(path)
|
|
if err != nil {
|
|
t.Fatalf("TryAcquireSessionLease: %v", err)
|
|
}
|
|
defer lease.Release()
|
|
|
|
if _, err := TryAcquireSessionRemovalGuard(path); !errors.Is(err, ErrSessionLeaseHeld) {
|
|
t.Fatalf("guard under live lease err = %v, want ErrSessionLeaseHeld", err)
|
|
}
|
|
if _, err := os.Stat(store.SessionLeaseLock(path)); err != nil {
|
|
t.Fatalf("lease lock disturbed by failed guard: %v", err)
|
|
}
|
|
|
|
lease.Release()
|
|
guard, err := TryAcquireSessionRemovalGuard(path)
|
|
if err != nil {
|
|
t.Fatalf("guard after release: %v", err)
|
|
}
|
|
// While the guard holds both locks, no new lease can be acquired: this is
|
|
// the window that used to allow probe-then-delete races.
|
|
if _, err := TryAcquireSessionLease(path); !errors.Is(err, ErrSessionLeaseHeld) {
|
|
guard.Release()
|
|
t.Fatalf("lease acquired while removal guard held, err = %v", err)
|
|
}
|
|
if err := guard.RemoveSidecarsAndRelease(); err != nil {
|
|
t.Fatalf("RemoveSidecarsAndRelease: %v", err)
|
|
}
|
|
for _, p := range []string{
|
|
store.SessionLockFile(path),
|
|
store.SessionLeaseLock(path),
|
|
store.SessionLeaseInfo(path),
|
|
} {
|
|
if _, err := os.Stat(p); !os.IsNotExist(err) {
|
|
t.Errorf("sidecar survived removal: %s (err=%v)", p, err)
|
|
}
|
|
}
|
|
// The path is free again for a normal acquire afterwards.
|
|
after, err := TryAcquireSessionLease(path)
|
|
if err != nil {
|
|
t.Fatalf("lease after removal: %v", err)
|
|
}
|
|
after.Release()
|
|
}
|
|
|
|
func TestSessionRemovalGuardReleaseKeepsSidecars(t *testing.T) {
|
|
dir := t.TempDir()
|
|
path := filepath.Join(dir, "session.jsonl")
|
|
guard, err := TryAcquireSessionRemovalGuard(path)
|
|
if err != nil {
|
|
t.Fatalf("guard: %v", err)
|
|
}
|
|
guard.Release()
|
|
// Abort path: locks released, files left in place for the next owner.
|
|
if _, err := os.Stat(store.SessionLeaseLock(path)); err != nil {
|
|
t.Fatalf("lease lock missing after abort: %v", err)
|
|
}
|
|
lease, err := TryAcquireSessionLease(path)
|
|
if err != nil {
|
|
t.Fatalf("lease after abort: %v", err)
|
|
}
|
|
lease.Release()
|
|
}
|