Complete platform management workflows

run/runtime/worker_test.go

@@ -76,6 +76,51 @@ func TestWorkerClaimsAcksProgressAndCompletesJob(t *testing.T) {
 	}
 }
 
+func TestWorkerDispatchesSelfUpdateJob(t *testing.T) {
+	client := newFakeWorkerClient()
+	assignment := workerJobAssignment(protocol.RunCapabilityRunSelfUpdate)
+	assignment.TargetKey = "run/update"
+	assignment.InputRef = "artifact://artifact-run-latest"
+	client.claimJob = assignment
+	worker, err := NewWorker(workerTestConfig(t), client)
+	if err != nil {
+		t.Fatalf("new worker: %v", err)
+	}
+	if err := worker.Register(context.Background()); err != nil {
+		t.Fatalf("register: %v", err)
+	}
+
+	handled, err := worker.ClaimAndRunOnce(context.Background())
+	if err != nil || !handled {
+		t.Fatalf("claim/run handled=%v err=%v", handled, err)
+	}
+	if len(client.resultRequests) != 1 || client.resultRequests[0].State != "succeeded" || !strings.Contains(client.resultRequests[0].ResultRef, "run-update-staged") {
+		t.Fatalf("expected self-update result, got %+v", client.resultRequests)
+	}
+}
+
+func TestWorkerRegistersPackageIdentity(t *testing.T) {
+	client := newFakeWorkerClient()
+	cfg := workerTestConfig(t)
+	cfg.RegistrationToken = "current-run-key"
+	cfg.ServerInstanceID = "server-worker"
+	cfg.PluginID = "game.minecraft"
+	cfg.ComponentKind = "run"
+	cfg.KeyGeneration = 7
+	worker, err := NewWorker(cfg, client)
+	if err != nil {
+		t.Fatalf("new worker: %v", err)
+	}
+
+	if err := worker.Register(context.Background()); err != nil {
+		t.Fatalf("register: %v", err)
+	}
+	hello := client.helloRequests[0]
+	if hello.RegistrationToken != "current-run-key" || hello.ServerInstanceID != "server-worker" || hello.ComponentKind != "run" || hello.KeyGeneration != 7 {
+		t.Fatalf("expected package identity in hello request, got %+v", hello)
+	}
+}
+
 func TestWorkerHandlesCancellationAndReconcile(t *testing.T) {
 	client := newFakeWorkerClient()
 	client.claimJob = workerJobAssignment(protocol.RunCapabilityProcessStart)