test

Documentation

Index

• 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 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 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 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.