testutil

package
v0.8.3 Latest Latest
Warning

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

 Go to latest Published: Apr 10, 2024 License: AGPL-3.0 Imports: 33 Imported by: 0

Documentation

Index

Constants

This section is empty.

Variables

View Source 
var (
	DefaultStorageClassName = "default-sc-for-testing"
)

Functions

func CreateMockStorageClass 

func CreateMockStorageClass(testCtx *testutil.TestContext, storageClassName string) *storagev1.StorageClass

func CreateVolumeSnapshotClass 

func CreateVolumeSnapshotClass(testCtx *testutil.TestContext, storageProvisioner string) *snapshotv1.VolumeSnapshotClass

func DeletePodLabelKey 

func DeletePodLabelKey(ctx context.Context, testCtx testutil.TestContext, podName, labelKey string)

DeletePodLabelKey deletes the specified label of the pod.

func GetDefaultStorageClass 

func GetDefaultStorageClass(testCtx *testutil.TestContext) *storagev1.StorageClass

func InitRSMStatus 

func InitRSMStatus(testCtx testutil.TestContext, rsm *workloads.ReplicatedStateMachine, controllerRevision string)

func InitStatefulSetStatus 

func InitStatefulSetStatus(testCtx testutil.TestContext, statefulset *appsv1.StatefulSet, controllerRevision string)

func IsMockVolumeSnapshotEnabled 

func IsMockVolumeSnapshotEnabled(testCtx *testutil.TestContext, storageClassName string) bool

func ListAndCheckDeployment 

func ListAndCheckDeployment(testCtx *testutil.TestContext, key types.NamespacedName) *appsv1.DeploymentList

func ListAndCheckPodCountWithComponent 

func ListAndCheckPodCountWithComponent(testCtx *testutil.TestContext, key types.NamespacedName, componentName string, cnt int) *corev1.PodList

func ListAndCheckRSM 

func ListAndCheckRSM(testCtx *testutil.TestContext, key types.NamespacedName) *workloads.ReplicatedStateMachineList

func ListAndCheckRSMItemsCount 

func ListAndCheckRSMItemsCount(testCtx *testutil.TestContext, key types.NamespacedName, cnt int) *workloads.ReplicatedStateMachineList

func ListAndCheckRSMWithComponent 

func ListAndCheckRSMWithComponent(testCtx *testutil.TestContext, key types.NamespacedName, componentName string) *workloads.ReplicatedStateMachineList

func ListAndCheckStatefulSet 

func ListAndCheckStatefulSet(testCtx *testutil.TestContext, key types.NamespacedName) *appsv1.StatefulSetList

func ListAndCheckStatefulSetItemsCount 

func ListAndCheckStatefulSetItemsCount(testCtx *testutil.TestContext, key types.NamespacedName, cnt int) *appsv1.StatefulSetList

func ListAndCheckStatefulSetWithComponent 

func ListAndCheckStatefulSetWithComponent(testCtx *testutil.TestContext, key types.NamespacedName, componentName string) *appsv1.StatefulSetList

func MockDeploymentReady 

func MockDeploymentReady(deploy *appsv1.Deployment, rsAvailableReason, rsName string)

MockDeploymentReady mocks deployment is ready

func MockDisableVolumeSnapshot 

func MockDisableVolumeSnapshot(testCtx *testutil.TestContext, storageClassName string)

func MockEnableVolumeSnapshot 

func MockEnableVolumeSnapshot(testCtx *testutil.TestContext, storageClassName string)

func MockPodAvailable 

func MockPodAvailable(pod *corev1.Pod, lastTransitionTime metav1.Time)

MockPodAvailable mocks pod is available

func MockPodIsTerminating 

func MockPodIsTerminating(ctx context.Context, testCtx testutil.TestContext, pod *corev1.Pod)

MockPodIsTerminating mocks pod is terminating.

func MockRSMReady 

func MockRSMReady(rsm *workloads.ReplicatedStateMachine, pods ...*corev1.Pod)

MockRSMReady mocks the RSM workload to ready state.

func MockStatefulSetReady 

func MockStatefulSetReady(sts *appsv1.StatefulSet)

MockStatefulSetReady mocks the StatefulSet workload is ready.

func NewFakeRSM 

func NewFakeRSM(name string, replicas int) *workloads.ReplicatedStateMachine

NewFakeRSM creates a fake RSM workload object for testing.

func NewFakeRSMPod 

func NewFakeRSMPod(rsm *workloads.ReplicatedStateMachine, ordinal int) *corev1.Pod

NewFakeRSMPod creates a fake pod of the RSM workload for testing.

func NewFakeStatefulSet 

func NewFakeStatefulSet(name string, replicas int) *appsv1.StatefulSet

NewFakeStatefulSet creates a fake StatefulSet workload object for testing.

func NewFakeStatefulSetPod 

func NewFakeStatefulSetPod(set *appsv1.StatefulSet, ordinal int) *corev1.Pod

NewFakeStatefulSetPod creates a fake pod of the StatefulSet workload for testing.

func PatchRSMStatus 

func PatchRSMStatus(testCtx *testutil.TestContext, stsName string, status workloads.ReplicatedStateMachineStatus)

func PatchStatefulSetStatus 

func PatchStatefulSetStatus(testCtx *testutil.TestContext, stsName string, status appsv1.StatefulSetStatus)

func RemovePodFinalizer 

func RemovePodFinalizer(ctx context.Context, testCtx testutil.TestContext, pod *corev1.Pod)

RemovePodFinalizer removes the pod finalizer to delete the pod finally.

func SetGetReturnedObject 

func SetGetReturnedObject(out client.Object, expectedObj client.Object)

func SetListReturnedObjects 

func SetListReturnedObjects(list client.ObjectList, objects []runtime.Object) error

func SetupK8sMock 

func SetupK8sMock() (*gomock.Controller, *mock_client.MockClient)

func UpdatePodStatusScheduleFailed 

func UpdatePodStatusScheduleFailed(ctx context.Context, testCtx testutil.TestContext, podName, namespace string)

UpdatePodStatusScheduleFailed updates the pod status to mock the schedule failure.

Types

type CallMockGetReturnedOptions 

type CallMockGetReturnedOptions = func(callHelper *callHelper, call *gomock.Call, _ HandleGetReturnedObject) error

type CallMockListReturnedOptions 

type CallMockListReturnedOptions = func(callHelper *callHelper, call *gomock.Call, _ HandleListReturnedObject) error

type CallMockOptions 

type CallMockOptions = func(call *gomock.Call)

func WithAnyTimes 

func WithAnyTimes() CallMockOptions

func WithFailed 

func WithFailed(err error, times ...CallMockOptions) CallMockOptions

func WithMaxTimes 

func WithMaxTimes(n int) CallMockOptions

func WithMinTimes 

func WithMinTimes(n int) CallMockOptions

func WithSucceed 

func WithSucceed(times ...CallMockOptions) CallMockOptions

func WithTimes 

func WithTimes(n int) CallMockOptions

type CallMockPatchReturnedOptions 

type CallMockPatchReturnedOptions = func(callHelper *callHelper, call *gomock.Call, _ HandlePatchReturnedObject) error

type CallMockReturnedOptions 

type CallMockReturnedOptions = func(callHelper *callHelper, call *gomock.Call)

func WithCreateReturned 

func WithCreateReturned(action HandleCreateReturnedObject, times ...CallMockOptions) CallMockReturnedOptions

func WithGetReturned 

func WithGetReturned(action HandleGetReturnedObject, times ...CallMockOptions) CallMockReturnedOptions

func WithListReturned 

func WithListReturned(action HandleListReturnedObject, times ...CallMockOptions) CallMockReturnedOptions

func WithPatchReturned 

func WithPatchReturned(action HandlePatchReturnedObject, times ...CallMockOptions) CallMockReturnedOptions

type CallbackFn 

type CallbackFn = func(sequence int, r []runtime.Object)

type CallerFunction 

type CallerFunction = func() *gomock.Call

type DoReturnedFunction 

type DoReturnedFunction = any

type Getter 

type Getter = func(key client.ObjectKey, obj client.Object) (bool, error)

type HandleCreateReturnedObject 

type HandleCreateReturnedObject = func(obj client.Object) error

func WithCreatedFailedResult 

func WithCreatedFailedResult() HandleCreateReturnedObject

func WithCreatedSucceedResult 

func WithCreatedSucceedResult() HandleCreateReturnedObject

type HandleGetReturnedObject 

type HandleGetReturnedObject = func(key client.ObjectKey, obj client.Object) error

func WithConstructGetResult 

func WithConstructGetResult(mockObj client.Object) HandleGetReturnedObject

func WithConstructSequenceResult 

func WithConstructSequenceResult(mockObjs map[client.ObjectKey][]MockGetReturned) HandleGetReturnedObject

func WithConstructSimpleGetResult 

func WithConstructSimpleGetResult(mockObjs []client.Object, get ...Getter) HandleGetReturnedObject

type HandleListReturnedObject 

type HandleListReturnedObject = func(list client.ObjectList) error

func WithConstructListReturnedResult 

func WithConstructListReturnedResult(r []runtime.Object) HandleListReturnedObject

func WithConstructListSequenceResult 

func WithConstructListSequenceResult(mockObjsList [][]runtime.Object, fns ...CallbackFn) HandleListReturnedObject

type HandlePatchReturnedObject 

type HandlePatchReturnedObject = func(obj client.Object, patch client.Patch) error

type K8sClientMockHelper 

type K8sClientMockHelper struct {
	// contains filtered or unexported fields
}

func NewK8sMockClient 

func NewK8sMockClient() *K8sClientMockHelper

func (*K8sClientMockHelper) Client 

func (helper *K8sClientMockHelper) Client() client.Client

func (*K8sClientMockHelper) Controller 

func (helper *K8sClientMockHelper) Controller() *gomock.Controller

func (*K8sClientMockHelper) Finish 

func (helper *K8sClientMockHelper) Finish()

func (*K8sClientMockHelper) MockCreateMethod 

func (helper *K8sClientMockHelper) MockCreateMethod(options ...any)

func (*K8sClientMockHelper) MockDeleteMethod 

func (helper *K8sClientMockHelper) MockDeleteMethod(options ...any)

func (*K8sClientMockHelper) MockGetMethod 

func (helper *K8sClientMockHelper) MockGetMethod(options ...any)

func (*K8sClientMockHelper) MockListMethod 

func (helper *K8sClientMockHelper) MockListMethod(options ...any)

func (*K8sClientMockHelper) MockPatchMethod 

func (helper *K8sClientMockHelper) MockPatchMethod(options ...any)

func (*K8sClientMockHelper) MockStatusMethod 

func (helper *K8sClientMockHelper) MockStatusMethod() *mock_client.MockStatusWriter

func (*K8sClientMockHelper) MockUpdateMethod 

func (helper *K8sClientMockHelper) MockUpdateMethod(options ...any)

func (*K8sClientMockHelper) StatusWriter 

func (helper *K8sClientMockHelper) StatusWriter() *mock_client.MockStatusWriter

type MockGetReturned 

type MockGetReturned struct {
	Object client.Object
	Err    error
}

type Mysql 

type Mysql struct {
	// contains filtered or unexported fields
}

Mysql represents MySQL output bindings.

func (*Mysql) Close 

func (m *Mysql) Close() error

Close closes the DB.

func (*Mysql) GetRole 

func (m *Mysql) GetRole(ctx context.Context) (string, error)

func (*Mysql) Init 

func (m *Mysql) Init(metadata map[string]string) error

Init initializes the MySQL binding.

type Tunnel 

type Tunnel struct {
	// contains filtered or unexported fields
}

func OpenTunnel 

func OpenTunnel(svc *corev1.Service) (*Tunnel, error)

func (*Tunnel) Close 

func (t *Tunnel) Close() error

func (*Tunnel) GetMySQLConn 

func (t *Tunnel) GetMySQLConn() (*Mysql, error)

Source Files

View all Source files

Directories

Path Synopsis

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.