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")
	}
}