92 lines
2.5 KiB
Go
92 lines
2.5 KiB
Go
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},
|
|
{"[self-hosted, vm-dev]", 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)
|
|
}
|
|
}
|