33 lines
966 B
Go
33 lines
966 B
Go
package store
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
)
|
|
|
|
func TestTryMarkTaskRunningIsExclusive(t *testing.T) {
|
|
s := testStore(t)
|
|
ctx := context.Background()
|
|
epSrc, _ := s.CreateEndpoint(ctx, Endpoint{RoleLabel: "src", Host: "a", Port: 993, TLSMode: "ssl"})
|
|
epDst, _ := s.CreateEndpoint(ctx, Endpoint{RoleLabel: "dst", Host: "b", Port: 993, TLSMode: "ssl"})
|
|
taskID, _ := s.CreateTask(ctx, Task{Name: "t", SrcEndpointID: epSrc, DstEndpointID: epDst})
|
|
|
|
first, err := s.TryMarkTaskRunning(ctx, taskID)
|
|
if err != nil || !first {
|
|
t.Fatalf("first acquire must succeed: ok=%v err=%v", first, err)
|
|
}
|
|
second, err := s.TryMarkTaskRunning(ctx, taskID)
|
|
if err != nil {
|
|
t.Fatalf("err: %v", err)
|
|
}
|
|
if second {
|
|
t.Fatal("second acquire must fail while running")
|
|
}
|
|
// after completion, a re-run may acquire again
|
|
_ = s.SetTaskStatus(ctx, taskID, "done")
|
|
third, _ := s.TryMarkTaskRunning(ctx, taskID)
|
|
if !third {
|
|
t.Fatal("acquire after completion must succeed")
|
|
}
|
|
}
|