utils

Documentation

Overview

Copyright (c) Microsoft Corporation. Licensed under the MIT license.

Index

• Constants
• Variables
• func Contains(s []string, e string) bool
• func IsAlreadyExistsError() error
• func NewTestScheme() *runtime.Scheme
• func NotFoundError() error
• func RegisterTestModel()
• type MockClient
  • func NewClient() *MockClient
  • func (m *MockClient) Create(ctx context.Context, obj k8sClient.Object, opts ...k8sClient.CreateOption) error
  • func (m *MockClient) CreateMapWithType(t interface{}) map[k8sClient.ObjectKey]k8sClient.Object
  • func (m *MockClient) CreateOrUpdateObjectInMap(obj k8sClient.Object)
  • func (m *MockClient) Delete(ctx context.Context, obj k8sClient.Object, opts ...k8sClient.DeleteOption) error
  • func (m *MockClient) DeleteAllOf(ctx context.Context, obj k8sClient.Object, opts ...k8sClient.DeleteAllOfOption) error
  • func (m *MockClient) Get(ctx context.Context, key types.NamespacedName, obj k8sClient.Object, ...) error
  • func (m *MockClient) GetObjectFromMap(obj k8sClient.Object, key types.NamespacedName)
  • func (m *MockClient) GroupVersionKindFor(obj runtime.Object) (schema.GroupVersionKind, error)
  • func (m *MockClient) IsObjectNamespaced(obj runtime.Object) (bool, error)
  • func (m *MockClient) List(ctx context.Context, list k8sClient.ObjectList, opts ...k8sClient.ListOption) error
  • func (m *MockClient) Patch(ctx context.Context, obj k8sClient.Object, patch k8sClient.Patch, ...) error
  • func (m *MockClient) RESTMapper() meta.RESTMapper
  • func (m *MockClient) Scheme() *runtime.Scheme
  • func (m *MockClient) Status() k8sClient.StatusWriter
  • func (m *MockClient) SubResource(subResource string) k8sClient.SubResourceClient
  • func (m *MockClient) Update(ctx context.Context, obj k8sClient.Object, opts ...k8sClient.UpdateOption) error
• type MockStatusClient
  • func (m *MockStatusClient) Create(ctx context.Context, obj k8sClient.Object, subResource k8sClient.Object, ...) error
  • func (m *MockStatusClient) Patch(ctx context.Context, obj k8sClient.Object, patch k8sClient.Patch, ...) error
  • func (m *MockStatusClient) Update(ctx context.Context, obj k8sClient.Object, ...) error

Constants

const (
	LabelKeyNvidia    = "accelerator"
	LabelValueNvidia  = "nvidia"
	CapacityNvidiaGPU = "nvidia.com/gpu"
)

const (
	// WorkspaceFinalizer is used to make sure that workspace controller handles garbage collection.
	WorkspaceFinalizer = "workspace.finalizer.kaito.sh"
)

Variables

var (
	MockMachine = v1alpha5.Machine{
		ObjectMeta: metav1.ObjectMeta{
			Name:   "testmachine",
			Labels: machineLabels,
		},
		Spec: v1alpha5.MachineSpec{
			MachineTemplateRef: &v1alpha5.MachineTemplateRef{
				Name: "test-machine",
			},
			Requirements: []corev1.NodeSelectorRequirement{
				{
					Key:      corev1.LabelInstanceTypeStable,
					Operator: corev1.NodeSelectorOpIn,
					Values:   []string{"Standard_NC12s_v3"},
				},
				{
					Key:      "karpenter.sh/provisioner-name",
					Operator: corev1.NodeSelectorOpIn,
					Values:   []string{"default"},
				},
			},
		},
	}
)

var (
	MockMachineList = &v1alpha5.MachineList{
		Items: []v1alpha5.Machine{
			MockMachine,
		},
	}
)

var (
	MockNodeList = &corev1.NodeList{
		Items: nodes,
	}
)

var (
	MockWorkspaceDistributedModel = &v1alpha1.Workspace{
		ObjectMeta: metav1.ObjectMeta{
			Name:      "testWorkspace",
			Namespace: "kaito",
		},
		Resource: v1alpha1.ResourceSpec{
			Count:        &gpuNodeCount,
			InstanceType: "Standard_NC12s_v3",
			LabelSelector: &metav1.LabelSelector{
				MatchLabels: map[string]string{
					"apps": "test",
				},
			},
		},
		Inference: v1alpha1.InferenceSpec{
			Preset: &v1alpha1.PresetSpec{
				PresetMeta: v1alpha1.PresetMeta{
					Name: "test-distributed-model",
				},
			},
		},
	}
)

var (
	MockWorkspaceWithInferenceTemplate = &v1alpha1.Workspace{
		ObjectMeta: metav1.ObjectMeta{
			Name:      "testWorkspace",
			Namespace: "kaito",
		},
		Resource: v1alpha1.ResourceSpec{
			Count:        &gpuNodeCount,
			InstanceType: "Standard_NC12s_v3",
			LabelSelector: &metav1.LabelSelector{
				MatchLabels: map[string]string{
					"apps": "test",
				},
			},
		},
		Inference: v1alpha1.InferenceSpec{
			Template: &corev1.PodTemplateSpec{},
		},
	}
)

var (
	MockWorkspaceWithPreset = &v1alpha1.Workspace{
		ObjectMeta: metav1.ObjectMeta{
			Name:      "testWorkspace",
			Namespace: "kaito",
		},
		Resource: v1alpha1.ResourceSpec{
			Count:        &gpuNodeCount,
			InstanceType: "Standard_NC12s_v3",
			LabelSelector: &metav1.LabelSelector{
				MatchLabels: map[string]string{
					"apps": "test",
				},
			},
		},
		Inference: v1alpha1.InferenceSpec{
			Preset: &v1alpha1.PresetSpec{
				PresetMeta: v1alpha1.PresetMeta{
					Name: "test-model",
				},
			},
		},
	}
)

Functions

func Contains

func Contains(s []string, e string) bool

func IsAlreadyExistsError

func IsAlreadyExistsError() error

func NewTestScheme

func NewTestScheme() *runtime.Scheme

func NotFoundError

func NotFoundError() error

func RegisterTestModel

func RegisterTestModel()

Types

type MockClient

type MockClient struct {
	mock.Mock

	ObjectMap  map[reflect.Type]map[k8sClient.ObjectKey]k8sClient.Object
	StatusMock *MockStatusClient
	UpdateCb   func(key types.NamespacedName)
}

Client is a mock for the controller-runtime dynamic client interface.

func NewClient

func NewClient() *MockClient

func (*MockClient) Create

func (m *MockClient) Create(ctx context.Context, obj k8sClient.Object, opts ...k8sClient.CreateOption) error

func (*MockClient) CreateMapWithType

func (m *MockClient) CreateMapWithType(t interface{}) map[k8sClient.ObjectKey]k8sClient.Object

func (*MockClient) CreateOrUpdateObjectInMap

func (m *MockClient) CreateOrUpdateObjectInMap(obj k8sClient.Object)

func (*MockClient) Delete

func (m *MockClient) Delete(ctx context.Context, obj k8sClient.Object, opts ...k8sClient.DeleteOption) error

func (*MockClient) DeleteAllOf

func (m *MockClient) DeleteAllOf(ctx context.Context, obj k8sClient.Object, opts ...k8sClient.DeleteAllOfOption) error

func (*MockClient) Get

func (m *MockClient) Get(ctx context.Context, key types.NamespacedName, obj k8sClient.Object, opts ...k8sClient.GetOption) error

k8s Client interface

func (*MockClient) GetObjectFromMap

func (m *MockClient) GetObjectFromMap(obj k8sClient.Object, key types.NamespacedName)

func (*MockClient) GroupVersionKindFor

func (m *MockClient) GroupVersionKindFor(obj runtime.Object) (schema.GroupVersionKind, error)

GroupVersionKindFor implements client.Client

func (*MockClient) IsObjectNamespaced

func (m *MockClient) IsObjectNamespaced(obj runtime.Object) (bool, error)

IsObjectNamespaced implements client.Client

func (*MockClient) List

func (m *MockClient) List(ctx context.Context, list k8sClient.ObjectList, opts ...k8sClient.ListOption) error

func (*MockClient) Patch

func (m *MockClient) Patch(ctx context.Context, obj k8sClient.Object, patch k8sClient.Patch, opts ...k8sClient.PatchOption) error

func (*MockClient) RESTMapper

func (m *MockClient) RESTMapper() meta.RESTMapper

func (*MockClient) Scheme

func (m *MockClient) Scheme() *runtime.Scheme

func (*MockClient) Status

func (m *MockClient) Status() k8sClient.StatusWriter

func (*MockClient) SubResource

func (m *MockClient) SubResource(subResource string) k8sClient.SubResourceClient

SubResource implements client.Client

func (*MockClient) Update

func (m *MockClient) Update(ctx context.Context, obj k8sClient.Object, opts ...k8sClient.UpdateOption) error

type MockStatusClient

type MockStatusClient struct {
	mock.Mock
}

func (*MockStatusClient) Create

func (m *MockStatusClient) Create(ctx context.Context, obj k8sClient.Object, subResource k8sClient.Object, opts ...k8sClient.SubResourceCreateOption) error

func (*MockStatusClient) Patch

func (m *MockStatusClient) Patch(ctx context.Context, obj k8sClient.Object, patch k8sClient.Patch, opts ...k8sClient.SubResourcePatchOption) error

func (*MockStatusClient) Update

func (m *MockStatusClient) Update(ctx context.Context, obj k8sClient.Object, opts ...k8sClient.SubResourceUpdateOption) error