lease_test.go RawHistoryFolder

1package subagents
2
3import (
4	"errors"
5	"testing"
6)
7
8func TestAgentLeaseIsExclusive(t *testing.T) {
9	stateDir := t.TempDir()
10	first, err := acquireAgentLease(stateDir)
11	if err != nil {
12		t.Fatal(err)
13	}
14	defer first.Close()
15	second, err := acquireAgentLease(stateDir)
16	if !errors.Is(err, ErrAgentLeaseHeld) {
17		if second != nil {
18			_ = second.Close()
19		}
20		t.Fatalf("second lease = %v, want ErrAgentLeaseHeld", err)
21	}
22	if err := first.Close(); err != nil {
23		t.Fatal(err)
24	}
25	third, err := acquireAgentLease(stateDir)
26	if err != nil {
27		t.Fatal(err)
28	}
29	if err := third.Close(); err != nil {
30		t.Fatal(err)
31	}
32}