util

package
v0.7.0 Latest Latest
Warning

This package is not in the latest version of its module.

Go to latest
Published: May 31, 2024 License: Apache-2.0 Imports: 34 Imported by: 0

Documentation

Index

Constants

View Source
const (
	TinyTimeout  = 10 * time.Millisecond
	ShortTimeout = time.Second
	Timeout      = 5 * time.Second
	// LongTimeout is meant for E2E tests when waiting for complex operations
	// such as running pods to completion.
	LongTimeout = 45 * time.Second
	// StartupTimeout is meant to be used for waiting for Kueue to startup, given
	// that cert updates can take up to 3 minutes to propagate to the filesystem.
	// Taken into account that after the certificates are ready, all Kueue's components
	// need started and the time it takes for a change in ready probe response triggers
	// a change in the deployment status.
	StartUpTimeout     = 5 * time.Minute
	ConsistentDuration = time.Second
	Interval           = time.Millisecond * 250
)

Variables

View Source
var (
	IgnoreConditionTimestamps                      = cmpopts.IgnoreFields(metav1.Condition{}, "LastTransitionTime")
	IgnoreConditionTimestampsAndObservedGeneration = cmpopts.IgnoreFields(metav1.Condition{}, "LastTransitionTime", "ObservedGeneration")
	IgnoreConditionMessage                         = cmpopts.IgnoreFields(metav1.Condition{}, "Message")
	IgnoreObjectMetaResourceVersion                = cmpopts.IgnoreFields(metav1.ObjectMeta{}, "ResourceVersion")
)

Functions

func AwaitAndVerifyCreatedWorkload added in v0.5.0

func AwaitAndVerifyCreatedWorkload(ctx context.Context, client client.Client, wlLookupKey types.NamespacedName, createdJob metav1.Object) *kueue.Workload

func AwaitAndVerifyWorkloadQueueName added in v0.5.0

func AwaitAndVerifyWorkloadQueueName(ctx context.Context, client client.Client, createdWorkload *kueue.Workload, wlLookupKey types.NamespacedName, jobQueueName string)

func AwaitWorkloadEvictionByPodsReadyTimeout added in v0.6.3

func AwaitWorkloadEvictionByPodsReadyTimeout(ctx context.Context, k8sClient client.Client, wlKey client.ObjectKey, sleep time.Duration)

func BindPodWithNode added in v0.6.3

func BindPodWithNode(ctx context.Context, k8sClient client.Client, nodeName string, pods ...*corev1.Pod)

func CreateClientUsingCluster added in v0.6.0

func CreateClientUsingCluster(kContext string) client.WithWatch

func CreateVisibilityClient added in v0.6.0

func CreateVisibilityClient(user string) visibilityv1alpha1.VisibilityV1alpha1Interface

func DeleteAdmissionCheck added in v0.5.0

func DeleteAdmissionCheck(ctx context.Context, c client.Client, ac *kueue.AdmissionCheck) error

func DeleteAllJobsInNamespace added in v0.4.0

func DeleteAllJobsInNamespace(ctx context.Context, c client.Client, ns *corev1.Namespace) error

func DeleteAllJobsetsInNamespace added in v0.7.0

func DeleteAllJobsetsInNamespace(ctx context.Context, c client.Client, ns *corev1.Namespace) error

func DeleteAllPodsInNamespace added in v0.6.0

func DeleteAllPodsInNamespace(ctx context.Context, c client.Client, ns *corev1.Namespace) error

func DeleteClusterQueue

func DeleteClusterQueue(ctx context.Context, c client.Client, cq *kueue.ClusterQueue) error

func DeleteLocalQueue

func DeleteLocalQueue(ctx context.Context, c client.Client, q *kueue.LocalQueue) error

func DeleteNamespace

func DeleteNamespace(ctx context.Context, c client.Client, ns *corev1.Namespace) error

DeleteNamespace deletes all objects the tests typically create in the namespace.

func DeleteProvisioningRequestConfig added in v0.5.0

func DeleteProvisioningRequestConfig(ctx context.Context, c client.Client, ac *kueue.ProvisioningRequestConfig) error

func DeleteResourceFlavor

func DeleteResourceFlavor(ctx context.Context, c client.Client, rf *kueue.ResourceFlavor) error

func DeleteRuntimeClass

func DeleteRuntimeClass(ctx context.Context, c client.Client, runtimeClass *nodev1.RuntimeClass) error

func DeleteWorkload

func DeleteWorkload(ctx context.Context, c client.Client, wl *kueue.Workload) error

func DeleteWorkloadsInNamespace

func DeleteWorkloadsInNamespace(ctx context.Context, c client.Client, ns *corev1.Namespace) error

func ExpectAdmissionAttemptsMetric added in v0.6.1

func ExpectAdmissionAttemptsMetric(pending, admitted int)

func ExpectAdmissionCheckToBeDeleted added in v0.5.0

func ExpectAdmissionCheckToBeDeleted(ctx context.Context, k8sClient client.Client, ac *kueue.AdmissionCheck, deleteAC bool)

func ExpectAdmittedWorkloadsTotalMetric

func ExpectAdmittedWorkloadsTotalMetric(cq *kueue.ClusterQueue, v int)

func ExpectCQResourceBorrowingQuota added in v0.5.0

func ExpectCQResourceBorrowingQuota(cq *kueue.ClusterQueue, flavor, resource string, v float64)

func ExpectCQResourceNominalQuota added in v0.5.0

func ExpectCQResourceNominalQuota(cq *kueue.ClusterQueue, flavor, resource string, v float64)

func ExpectCQResourceReservations added in v0.5.0

func ExpectCQResourceReservations(cq *kueue.ClusterQueue, flavor, resource string, v float64)

func ExpectClusterQueueStatusMetric

func ExpectClusterQueueStatusMetric(cq *kueue.ClusterQueue, status metrics.ClusterQueueStatus)

func ExpectClusterQueueToBeDeleted

func ExpectClusterQueueToBeDeleted(ctx context.Context, k8sClient client.Client, cq *kueue.ClusterQueue, deleteCq bool)

func ExpectClusterQueueWeightedShareMetric added in v0.7.0

func ExpectClusterQueueWeightedShareMetric(cq *kueue.ClusterQueue, v int64)

func ExpectEventsForObjects added in v0.7.0

func ExpectEventsForObjects(eventWatcher watch.Interface, objs sets.Set[types.NamespacedName], filter func(*corev1.Event) bool)

func ExpectEvictedWorkloadsTotalMetric added in v0.7.0

func ExpectEvictedWorkloadsTotalMetric(cqName string, reason string, v int)

func ExpectPendingWorkloadsMetric

func ExpectPendingWorkloadsMetric(cq *kueue.ClusterQueue, active, inadmissible int)

func ExpectPodUnsuspendedWithNodeSelectors added in v0.6.0

func ExpectPodUnsuspendedWithNodeSelectors(ctx context.Context, k8sClient client.Client, key types.NamespacedName, ns map[string]string)

func ExpectPodsFinalized added in v0.6.0

func ExpectPodsFinalized(ctx context.Context, k8sClient client.Client, keys ...types.NamespacedName)

func ExpectProvisioningRequestConfigToBeDeleted added in v0.5.0

func ExpectProvisioningRequestConfigToBeDeleted(ctx context.Context, k8sClient client.Client, prc *kueue.ProvisioningRequestConfig, deleteAC bool)

func ExpectQuotaReservedWorkloadsTotalMetric added in v0.7.0

func ExpectQuotaReservedWorkloadsTotalMetric(cq *kueue.ClusterQueue, v int)

func ExpectReservingActiveWorkloadsMetric added in v0.5.0

func ExpectReservingActiveWorkloadsMetric(cq *kueue.ClusterQueue, v int)

func ExpectResourceFlavorToBeDeleted

func ExpectResourceFlavorToBeDeleted(ctx context.Context, k8sClient client.Client, rf *kueue.ResourceFlavor, deleteRf bool)

func ExpectWorkloadToBeAdmittedAs

func ExpectWorkloadToBeAdmittedAs(ctx context.Context, k8sClient client.Client, wl *kueue.Workload, admission *kueue.Admission)

func ExpectWorkloadToFinish added in v0.6.0

func ExpectWorkloadToFinish(ctx context.Context, k8sClient client.Client, wlKey client.ObjectKey)

func ExpectWorkloadToHaveRequeueState added in v0.6.3

func ExpectWorkloadToHaveRequeueState(ctx context.Context, k8sClient client.Client, wlKey client.ObjectKey, expected *kueue.RequeueState, hasRequeueAt bool)

func ExpectWorkloadsToBeAdmitted

func ExpectWorkloadsToBeAdmitted(ctx context.Context, k8sClient client.Client, wls ...*kueue.Workload)

func ExpectWorkloadsToBeAdmittedCount added in v0.6.2

func ExpectWorkloadsToBeAdmittedCount(ctx context.Context, k8sClient client.Client, count int, wls ...*kueue.Workload)

func ExpectWorkloadsToBeFrozen

func ExpectWorkloadsToBeFrozen(ctx context.Context, k8sClient client.Client, cq string, wls ...*kueue.Workload)

func ExpectWorkloadsToBePending

func ExpectWorkloadsToBePending(ctx context.Context, k8sClient client.Client, wls ...*kueue.Workload)

func ExpectWorkloadsToBePreempted added in v0.5.0

func ExpectWorkloadsToBePreempted(ctx context.Context, k8sClient client.Client, wls ...*kueue.Workload)

func ExpectWorkloadsToBeWaiting

func ExpectWorkloadsToBeWaiting(ctx context.Context, k8sClient client.Client, wls ...*kueue.Workload)

func ExpectWorkloadsToHaveQuotaReservation added in v0.5.0

func ExpectWorkloadsToHaveQuotaReservation(ctx context.Context, k8sClient client.Client, cqName string, wls ...*kueue.Workload)

func FilterAdmittedWorkloads added in v0.4.0

func FilterAdmittedWorkloads(ctx context.Context, k8sClient client.Client, wls ...*kueue.Workload) []*kueue.Workload

func FilterEvictedWorkloads added in v0.6.2

func FilterEvictedWorkloads(ctx context.Context, k8sClient client.Client, wls ...*kueue.Workload) []*kueue.Workload

func FinishEvictionForWorkloads added in v0.4.0

func FinishEvictionForWorkloads(ctx context.Context, k8sClient client.Client, wls ...*kueue.Workload)

func FinishWorkloads

func FinishWorkloads(ctx context.Context, k8sClient client.Client, workloads ...*kueue.Workload)

func SetAdmissionCheckActive added in v0.5.0

func SetAdmissionCheckActive(ctx context.Context, k8sClient client.Client, admissionCheck *kueue.AdmissionCheck, status metav1.ConditionStatus)

func SetPodsPhase added in v0.6.0

func SetPodsPhase(ctx context.Context, k8sClient client.Client, phase corev1.PodPhase, pods ...*corev1.Pod)

func SetQuotaReservation added in v0.5.0

func SetQuotaReservation(ctx context.Context, k8sClient client.Client, wl *kueue.Workload, admission *kueue.Admission) error

func SetRequeuedConditionWithPodsReadyTimeout added in v0.7.0

func SetRequeuedConditionWithPodsReadyTimeout(ctx context.Context, k8sClient client.Client, wlKey client.ObjectKey)

func SetWorkloadsAdmissionCheck added in v0.5.0

func SetWorkloadsAdmissionCheck(ctx context.Context, k8sClient client.Client, wl *kueue.Workload, check string, state kueue.CheckState, expectExisting bool)

func SyncAdmittedConditionForWorkloads added in v0.5.0

func SyncAdmittedConditionForWorkloads(ctx context.Context, k8sClient client.Client, wls ...*kueue.Workload)

SyncAdmittedConditionForWorkloads sets the Admission condition of the provided workloads based on the state of quota reservation and admission checks. It should be use in tests that are not running the workload controller.

func UnholdQueue added in v0.6.0

func UnholdQueue(ctx context.Context, k8sClient client.Client, cq *kueue.ClusterQueue)

func VerifyWorkloadPriority added in v0.5.0

func VerifyWorkloadPriority(createdWorkload *kueue.Workload, priorityClassName string, priorityValue int32)

func WaitForKueueAvailability added in v0.6.0

func WaitForKueueAvailability(ctx context.Context, k8sClient client.Client)

Types

This section is empty.

Jump to

Keyboard shortcuts

? : This menu
/ : Search site
f or F : Jump to
y or Y : Canonical URL