test/e2e/scheduling/workload.go - master - githubmirror/kubernetes

test/e2e/scheduling/workload.go
112 строк4 KB
Antoni Zawodny
Update the code paths to use v1beta1
21 июл 2026, 15:56
21 июл 2026, 15:56108a615
/*
Copyright 2025 The Kubernetes Authors.

Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at

    http://www.apache.org/licenses/LICENSE-2.0

Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.
*/

package scheduling

import (
	"context"

	schedulingv1beta1 "k8s.io/api/scheduling/v1beta1"
	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
	"k8s.io/kubernetes/pkg/apis/scheduling"
	"k8s.io/kubernetes/pkg/features"
	"k8s.io/kubernetes/test/e2e/framework"
	e2econformance "k8s.io/kubernetes/test/e2e/framework/conformance"
	admissionapi "k8s.io/pod-security-admission/api"
)

var _ = SIGDescribe("Workload", framework.WithFeatureGate(features.GenericWorkload), func() {
	f := framework.NewDefaultFramework("workload-test")
	f.NamespacePodSecurityLevel = admissionapi.LevelPrivileged

	f.Context("CRUD Tests", func() {
		/*
			Testname: CRUD operations for Workloads
			Description: kube-apiserver must support create/get/list/update/patch/delete
			operations for scheduling.k8s.io/v1beta1 Workload.
		*/
		framework.It("Workload API availability", func(ctx context.Context) {
			e2econformance.TestResource(ctx, f,
				&e2econformance.ResourceTestcase[*schedulingv1beta1.Workload]{
					GVR:        schedulingv1beta1.SchemeGroupVersion.WithResource("workloads"),
					Namespaced: new(true),
					InitialSpec: &schedulingv1beta1.Workload{
						Spec: schedulingv1beta1.WorkloadSpec{
							ControllerRef: &schedulingv1beta1.TypedLocalObjectReference{
								APIGroup: "batch",
								Kind:     "Job",
								Name:     "foo",
							},
							PodGroupTemplates: []schedulingv1beta1.PodGroupTemplate{
								{
									Name: "pg1",
									SchedulingPolicy: schedulingv1beta1.PodGroupSchedulingPolicy{
										Gang: &schedulingv1beta1.GangSchedulingPolicy{
											MinCount: 5,
										},
									},
								},
							},
						},
					},
					UpdateSpec: func(obj *schedulingv1beta1.Workload) *schedulingv1beta1.Workload {
						obj.Labels["foo"] = "bar"
						return obj
					},
					StrategicMergePatchSpec: `{"metadata": {"labels": {"baz": "qux"}}}`,
				},
			)
		})

		/*
			Testname: CRUD operations for PodGroups
			Description: kube-apiserver must support create/get/list/update/patch/delete
			operations for scheduling.k8s.io/v1beta1 PodGroup.
		*/
		framework.It("PodGroup API availability", func(ctx context.Context) {
			e2econformance.TestResource(ctx, f,
				&e2econformance.ResourceTestcase[*schedulingv1beta1.PodGroup]{
					GVR:        schedulingv1beta1.SchemeGroupVersion.WithResource("podgroups"),
					Namespaced: new(true),
					InitialSpec: &schedulingv1beta1.PodGroup{
						Spec: schedulingv1beta1.PodGroupSpec{
							SchedulingPolicy: schedulingv1beta1.PodGroupSchedulingPolicy{
								Gang: &schedulingv1beta1.GangSchedulingPolicy{
									MinCount: 5,
								},
							},
						},
					},
					UpdateSpec: func(obj *schedulingv1beta1.PodGroup) *schedulingv1beta1.PodGroup {
						obj.Labels["foo"] = "bar"
						return obj
					},
					UpdateStatus: func(obj *schedulingv1beta1.PodGroup) *schedulingv1beta1.PodGroup {
						obj.Status.Conditions = append(obj.Status.Conditions, metav1.Condition{
							Type:               scheduling.PodGroupInitiallyScheduled,
							Status:             metav1.ConditionFalse,
							Reason:             scheduling.PodGroupReasonUnschedulable,
							Message:            "Test status condition message",
							LastTransitionTime: metav1.Now(),
						})
						return obj
					},
					StrategicMergePatchSpec: `{"metadata": {"labels": {"baz": "qux"}}}`,
				},
			)
		})
	})
})