workermgr

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

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

type WorkerMgrAPI interface {
	// Work executes the work on given node via worker pod
	Work(ctx context.Context, devConfig *amdv1alpha1.DeviceConfig, node *v1.Node) error
	// Cleanup cleanup the work on given node
	Cleanup(ctx context.Context, devConfig *amdv1alpha1.DeviceConfig, node *v1.Node) error
	// GetWorkerPod fetches the worker pod info from cluster
	GetWorkerPod(ctx context.Context, devConfig *amdv1alpha1.DeviceConfig, node *v1.Node) (*v1.Pod, error)
	// Add a node label to mark that the work is completed
	AddWorkReadyLabel(ctx context.Context, logger logr.Logger, nsn types.NamespacedName, nodeName string)
	// GetWorkReadyLabel get the label key to mark that the work is completed
	GetWorkReadyLabel(nsn types.NamespacedName) string
	// Remove the node label that indicates the work is completed
	RemoveWorkReadyLabel(ctx context.Context, logger logr.Logger, nsn types.NamespacedName, nodeName string)
}