testutil

Documentation

Index

• Constants
• Variables
• func CheckCondition(tfJob *tfv1alpha2.TFJob, condition tfv1alpha2.TFJobConditionType, ...) bool
• func ConvertTFJobToUnstructured(tfJob *tfv1alpha2.TFJob) (*unstructured.Unstructured, error)
• func GenLabels(jobName string) map[string]string
• func GenOwnerReference(tfjob *tfv1alpha2.TFJob) *metav1.OwnerReference
• func GetKey(tfJob *tfv1alpha2.TFJob, t *testing.T) string
• func NewBasePod(name string, tfJob *tfv1alpha2.TFJob, t *testing.T) *v1.Pod
• func NewBaseService(name string, tfJob *tfv1alpha2.TFJob, t *testing.T) *v1.Service
• func NewPod(tfJob *tfv1alpha2.TFJob, typ string, index int, t *testing.T) *v1.Pod
• func NewPodList(count int32, status v1.PodPhase, tfJob *tfv1alpha2.TFJob, typ string, ...) []*v1.Pod
• func NewService(tfJob *tfv1alpha2.TFJob, typ string, index int, t *testing.T) *v1.Service
• func NewServiceList(count int32, tfJob *tfv1alpha2.TFJob, typ string, t *testing.T) []*v1.Service
• func NewTFJob(worker, ps int) *tfv1alpha2.TFJob
• func NewTFJobWithChief(worker, ps int) *tfv1alpha2.TFJob
• func NewTFJobWithCleanPolicy(chief, worker, ps int, policy tfv1alpha2.CleanPodPolicy) *tfv1alpha2.TFJob
• func NewTFJobWithCleanupJobDelay(chief, worker, ps int, ttl *int32) *tfv1alpha2.TFJob
• func NewTFJobWithEvaluator(worker, ps, evaluator int) *tfv1alpha2.TFJob
• func NewTFReplicaSpecTemplate() v1.PodTemplateSpec
• func SetPodsStatuses(podIndexer cache.Indexer, tfJob *tfv1alpha2.TFJob, typ string, ...)
• func SetServices(serviceIndexer cache.Indexer, tfJob *tfv1alpha2.TFJob, typ string, ...)
• func SetTFJobCompletionTime(tfJob *tfv1alpha2.TFJob)

Constants

const (
	TestImageName = "test-image-for-kubeflow-tf-operator:latest"
	TestTFJobName = "test-tfjob"
	LabelWorker   = "worker"
	LabelPS       = "ps"

	SleepInterval = 500 * time.Millisecond
	ThreadCount   = 1
)

const (
	LabelGroupName = "group_name"
	LabelTFJobName = "tf_job_name"
)

Variables

var (
	// KeyFunc is the short name to DeletionHandlingMetaNamespaceKeyFunc.
	// IndexerInformer uses a delta queue, therefore for deletes we have to use this
	// key function but it should be just fine for non delete events.
	KeyFunc   = cache.DeletionHandlingMetaNamespaceKeyFunc
	GroupName = tfv1alpha2.GroupName
)

var (
	AlwaysReady = func() bool { return true }
)

Functions

func CheckCondition

func CheckCondition(tfJob *tfv1alpha2.TFJob, condition tfv1alpha2.TFJobConditionType, reason string) bool

func ConvertTFJobToUnstructured

func ConvertTFJobToUnstructured(tfJob *tfv1alpha2.TFJob) (*unstructured.Unstructured, error)

ConvertTFJobToUnstructured uses JSON to convert TFJob to Unstructured.

func GenLabels

func GenLabels(jobName string) map[string]string

func GenOwnerReference

func GenOwnerReference(tfjob *tfv1alpha2.TFJob) *metav1.OwnerReference

func GetKey

func GetKey(tfJob *tfv1alpha2.TFJob, t *testing.T) string

func NewBasePod

func NewBasePod(name string, tfJob *tfv1alpha2.TFJob, t *testing.T) *v1.Pod

func NewBaseService

func NewBaseService(name string, tfJob *tfv1alpha2.TFJob, t *testing.T) *v1.Service

func NewPod

func NewPod(tfJob *tfv1alpha2.TFJob, typ string, index int, t *testing.T) *v1.Pod

func NewPodList

func NewPodList(count int32, status v1.PodPhase, tfJob *tfv1alpha2.TFJob, typ string, start int32, t *testing.T) []*v1.Pod

create count pods with the given phase for the given tfJob

func NewService

func NewService(tfJob *tfv1alpha2.TFJob, typ string, index int, t *testing.T) *v1.Service

func NewServiceList

func NewServiceList(count int32, tfJob *tfv1alpha2.TFJob, typ string, t *testing.T) []*v1.Service

NewServiceList creates count pods with the given phase for the given tfJob

func NewTFJob

func NewTFJob(worker, ps int) *tfv1alpha2.TFJob

func NewTFJobWithChief

func NewTFJobWithChief(worker, ps int) *tfv1alpha2.TFJob

func NewTFJobWithCleanPolicy

func NewTFJobWithCleanPolicy(chief, worker, ps int, policy tfv1alpha2.CleanPodPolicy) *tfv1alpha2.TFJob

func NewTFJobWithCleanupJobDelay

func NewTFJobWithCleanupJobDelay(chief, worker, ps int, ttl *int32) *tfv1alpha2.TFJob

func NewTFJobWithEvaluator

func NewTFJobWithEvaluator(worker, ps, evaluator int) *tfv1alpha2.TFJob

func NewTFReplicaSpecTemplate

func NewTFReplicaSpecTemplate() v1.PodTemplateSpec

func SetPodsStatuses

func SetPodsStatuses(podIndexer cache.Indexer, tfJob *tfv1alpha2.TFJob, typ string, pendingPods, activePods, succeededPods, failedPods int32, t *testing.T)

func SetServices

func SetServices(serviceIndexer cache.Indexer, tfJob *tfv1alpha2.TFJob, typ string, activeWorkerServices int32, t *testing.T)

func SetTFJobCompletionTime

func SetTFJobCompletionTime(tfJob *tfv1alpha2.TFJob)