package taskassignment

import (
	"bytes"
	"testing"

	runnerv1 "gitea.dev/actionslib/runner/v1"
	"google.golang.org/protobuf/types/known/structpb"
)

func task(t *testing.T, labels string) *runnerv1.Task {
	t.Helper()
	context, err := structpb.NewStruct(map[string]any{"repository": "owner/repo"})
	if err != nil {
		t.Fatal(err)
	}
	return &runnerv1.Task{
		Id:              42,
		Context:         context,
		WorkflowPayload: []byte("jobs:\n  publish:\n    runs-on: " + labels + "\n    steps: []\n"),
	}
}

func TestNewSelectsBackendFromRunsOn(t *testing.T) {
	for _, test := range []struct {
		labels  string
		backend Backend
	}{
		{"[self-hosted, pod]", BackendPod},
		{"[self-hosted, vm]", BackendVM},
	} {
		assignment, err := New(task(t, test.labels), "ddupan.top")
		if err != nil {
			t.Fatal(err)
		}
		if assignment.Backend != test.backend || assignment.ID != "gitea-task-42" {
			t.Fatalf("assignment = %#v", assignment)
		}
	}
}

func TestNewRejectsAmbiguousBackend(t *testing.T) {
	for _, labels := range []string{
		"[self-hosted]",
		"[self-hosted, pod, vm]",
		"[pod]",
	} {
		if _, err := New(task(t, labels), "ddupan.top"); err == nil {
			t.Fatalf("expected labels %s to fail", labels)
		}
	}
}

func TestAssignmentWireRoundTripAndValidation(t *testing.T) {
	want, err := New(task(t, "[self-hosted, pod]"), "ddupan.top")
	if err != nil {
		t.Fatal(err)
	}
	data, err := Marshal(want)
	if err != nil {
		t.Fatal(err)
	}
	got, err := Unmarshal(data, "ddupan.top")
	if err != nil {
		t.Fatal(err)
	}
	if got.ID != want.ID || got.Backend != want.Backend || got.Identity != want.Identity || !bytes.Equal(got.Task.WorkflowPayload, want.Task.WorkflowPayload) {
		t.Fatalf("round trip = %#v, want %#v", got, want)
	}

	tampered := bytes.Replace(data, []byte(`"backend":"pod"`), []byte(`"backend":"vm"`), 1)
	if _, err := Unmarshal(tampered, "ddupan.top"); err == nil {
		t.Fatal("expected tampered backend to fail")
	}
}

func TestFromMetadataRecoversMinimalAssignment(t *testing.T) {
	assignment, err := FromMetadata(map[string]string{
		"ci.ddupan.top/assignment-id": "gitea-task-42",
		"ci.ddupan.top/task-id":       "42",
		"ci.ddupan.top/backend":       "vm",
	}, map[string]string{
		"ci.ddupan.top/repository": "owner/repo",
		"ci.ddupan.top/job-key":    "publish",
		"ci.ddupan.top/spiffe-id":  "spiffe://ddupan.top/ci/owner/repo/publish",
	}, "ddupan.top")
	if err != nil || assignment.Task.GetId() != 42 || assignment.Backend != BackendVM {
		t.Fatalf("assignment=%#v err=%v", assignment, err)
	}
}