quorate/services/project-delivery/controller/checkpoint_test.go

70 lines
2.5 KiB
Go
Raw Permalink Normal View History

package main
import (
"context"
"errors"
"testing"
"time"
apierrors "k8s.io/apimachinery/pkg/api/errors"
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
"k8s.io/apimachinery/pkg/apis/meta/v1/unstructured"
"k8s.io/apimachinery/pkg/runtime"
dynamicfake "k8s.io/client-go/dynamic/fake"
k8stesting "k8s.io/client-go/testing"
)
func TestCaptureCollected(t *testing.T) {
ctx := context.Background()
now := time.Date(2026, 10, 2, 0, 0, 0, 0, time.UTC)
revision := "0123456789abcdef0123456789abcdef01234567"
instance := testController(t, pointDelivery(revision), &mutableRefs{values: map[string]string{"refs/heads/main": revision}}, &now)
client := instance.client.(*dynamicfake.FakeDynamicClient)
changed := false
client.PrependReactor("update", "projectdeliveries", func(action k8stesting.Action) (bool, runtime.Object, error) {
if changed || action.GetSubresource() != "status" {
return false, nil, nil
}
changed = true
raw, err := client.Tracker().Get(projectDeliveryGVR, "delivery", "point")
if err != nil {
t.Fatal(err)
}
object := raw.(*unstructured.Unstructured).DeepCopy()
name := workflowName("build", string(object.GetUID()), "", revision)
accepted, err := client.Tracker().Get(workflowGVR, "delivery", name)
if err != nil {
t.Fatal(err)
}
workflow := accepted.(*unstructured.Unstructured).DeepCopy()
workflow.Object["status"] = map[string]any{"phase": "Succeeded"}
if len(workflow.GetFinalizers()) == 0 {
if err := client.Tracker().Delete(workflowGVR, "delivery", name); err != nil {
t.Fatal(err)
}
} else {
workflow.SetDeletionTimestamp(&metav1.Time{Time: now})
if err := client.Tracker().Update(workflowGVR, workflow, "delivery"); err != nil {
t.Fatal(err)
}
}
unstructured.RemoveNestedField(object.Object, "spec", "source", "point")
if err := unstructured.SetNestedMap(object.Object, map[string]any{"ref": "refs/heads/main"}, "spec", "source", "continuous"); err != nil {
t.Fatal(err)
}
if err := client.Tracker().Update(projectDeliveryGVR, object, "delivery"); err != nil {
t.Fatal(err)
}
return true, nil, apierrors.NewConflict(projectDeliveryGVR.GroupResource(), "point", errors.New("synthetic source update"))
})
if err := instance.reconcileKey(ctx, "delivery", "point"); err != nil {
t.Fatal(err)
}
points, _, err := unstructured.NestedSlice(getDelivery(t, instance, "point").Object, "status", "points")
if err != nil {
t.Fatal(err)
}
if len(points) != 1 {
t.Fatalf("accepted execution was collected before its status checkpoint: %d records", len(points))
}
}