test

Documentation

Index

• func AddGpuLabelToNode(node *apiv1.Node)
• func AddGpusToNode(node *apiv1.Node, gpusCount int64)
• func BuildTestNode(name string, millicpu int64, mem int64) *apiv1.Node
• func BuildTestPod(name string, cpu int64, mem int64) *apiv1.Pod
• func GenerateOwnerReferences(name, kind, api string, uid types.UID) []metav1.OwnerReference
• func GetGPULabel() string
• func RefJSON(o runtime.Object) string
• func RequestGpuForPod(pod *apiv1.Pod, gpusCount int64)
• func SetNodeCondition(node *apiv1.Node, conditionType apiv1.NodeConditionType, ...)
• func SetNodeReadyState(node *apiv1.Node, ready bool, lastTransition time.Time)
• type HttpServerMock
  • func NewHttpServerMock() *HttpServerMock

Functions

func AddGpuLabelToNode

func AddGpuLabelToNode(node *apiv1.Node)

AddGpuLabelToNode adds GPULabel to give node. This is used to mock intermediate result that GPU on node is not ready

func AddGpusToNode

func AddGpusToNode(node *apiv1.Node, gpusCount int64)

AddGpusToNode adds GPU capacity to given node. Default accelerator type is used.

func BuildTestNode

func BuildTestNode(name string, millicpu int64, mem int64) *apiv1.Node

BuildTestNode creates a node with specified capacity.

func BuildTestPod

func BuildTestPod(name string, cpu int64, mem int64) *apiv1.Pod

BuildTestPod creates a pod with specified resources.

func GenerateOwnerReferences

func GenerateOwnerReferences(name, kind, api string, uid types.UID) []metav1.OwnerReference

GenerateOwnerReferences builds OwnerReferences with a single reference

func GetGPULabel

func GetGPULabel() string

GetGPULabel return GPULabel on the node. This is only used in unit tests.

func RefJSON

func RefJSON(o runtime.Object) string

RefJSON builds string reference to

func RequestGpuForPod

func RequestGpuForPod(pod *apiv1.Pod, gpusCount int64)

RequestGpuForPod modifies pod's resource requests by adding a number of GPUs to them.

func SetNodeCondition

func SetNodeCondition(node *apiv1.Node, conditionType apiv1.NodeConditionType, status apiv1.ConditionStatus, lastTransition time.Time)

SetNodeCondition sets node condition.

func SetNodeReadyState

func SetNodeReadyState(node *apiv1.Node, ready bool, lastTransition time.Time)

SetNodeReadyState sets node ready state to either ConditionTrue or ConditionFalse.

Types

type HttpServerMock

type HttpServerMock struct {
	mock.Mock
	*httptest.Server
}

HttpServerMock mocks server HTTP.

Example: // Create HttpServerMock. server := NewHttpServerMock() defer server.Close() // Use server.URL to point your code to HttpServerMock. g := newTestGceManager(t, server.URL, ModeGKE) // Declare handled urls and results for them. server.On("handle", "/project1/zones/us-central1-b/listManagedInstances").Return("<managedInstances>").Once() // Call http server in your code. instances, err := g.GetManagedInstances() // Check if expected calls were executed.

mock.AssertExpectationsForObjects(t, server)

func NewHttpServerMock

func NewHttpServerMock() *HttpServerMock

NewHttpServerMock creates new HttpServerMock.