test

package
v0.3.0 Latest Latest
Warning

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

 Go to latest Published: Jul 10, 2024 License: MIT Imports: 20 Imported by: 0

Documentation

Index

Constants

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

Variables

View Source 
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"},
				},
			},
		},
	}

	MockNodeClaim = v1beta1.NodeClaim{
		ObjectMeta: metav1.ObjectMeta{
			Name:   "testnodeclaim",
			Labels: nodeClaimLabels,
		},
		Spec: v1beta1.NodeClaimSpec{
			Requirements: []v1beta1.NodeSelectorRequirementWithMinValues{
				{
					NodeSelectorRequirement: corev1.NodeSelectorRequirement{
						Key:      corev1.LabelInstanceTypeStable,
						Operator: corev1.NodeSelectorOpIn,
						Values:   []string{"Standard_NC12s_v3"},
					},
					MinValues: lo.ToPtr(1),
				},
			},
		},
	}
)
View Source 
var (
	MockMachineList = &v1alpha5.MachineList{
		Items: []v1alpha5.Machine{
			MockMachine,
		},
	}

	MockNodeClaimList = &v1beta1.NodeClaimList{
		Items: []v1beta1.NodeClaim{
			MockNodeClaim,
		},
	}
)
View Source 
var (
	MockNodeList = &corev1.NodeList{
		Items: nodes,
	}
)
View Source 
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",
				},
			},
		},
	}
)
View Source 
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{},
		},
	}
)
View Source 
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 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)
}

MockClient 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

Get 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

Source Files

View all Source files

Jump to

Keyboard shortcuts

? : This menu
/ : Search site
f or F : Jump to
y or Y : Canonical URL
go.dev uses cookies from Google to deliver and enhance the quality of its services and to analyze traffic. Learn more.