Improve the unit test coverage of pkg/cmd/server package (#6342)

Improve the unit test coverage of pkg/cmd/server package

Signed-off-by: Wenkai Yin(尹文开) <yinw@vmware.com>

pkg/discovery/helper.go

@@ -34,6 +34,8 @@ import (
 	kcmdutil "github.com/vmware-tanzu/velero/third_party/kubernetes/pkg/kubectl/cmd/util"
 )
 
+//go:generate mockery --name Helper
+
 // Helper exposes functions for interacting with the Kubernetes discovery
 // API.
 type Helper interface {

pkg/discovery/mocks/Helper.go

@@ -0,0 +1,156 @@
+// Code generated by mockery v2.28.1. DO NOT EDIT.
+
+package mocks
+
+import (
+	mock "github.com/stretchr/testify/mock"
+	schema "k8s.io/apimachinery/pkg/runtime/schema"
+
+	v1 "k8s.io/apimachinery/pkg/apis/meta/v1"
+
+	version "k8s.io/apimachinery/pkg/version"
+)
+
+// Helper is an autogenerated mock type for the Helper type
+type Helper struct {
+	mock.Mock
+}
+
+// APIGroups provides a mock function with given fields:
+func (_m *Helper) APIGroups() []v1.APIGroup {
+	ret := _m.Called()
+
+	var r0 []v1.APIGroup
+	if rf, ok := ret.Get(0).(func() []v1.APIGroup); ok {
+		r0 = rf()
+	} else {
+		if ret.Get(0) != nil {
+			r0 = ret.Get(0).([]v1.APIGroup)
+		}
+	}
+
+	return r0
+}
+
+// KindFor provides a mock function with given fields: input
+func (_m *Helper) KindFor(input schema.GroupVersionKind) (schema.GroupVersionResource, v1.APIResource, error) {
+	ret := _m.Called(input)
+
+	var r0 schema.GroupVersionResource
+	var r1 v1.APIResource
+	var r2 error
+	if rf, ok := ret.Get(0).(func(schema.GroupVersionKind) (schema.GroupVersionResource, v1.APIResource, error)); ok {
+		return rf(input)
+	}
+	if rf, ok := ret.Get(0).(func(schema.GroupVersionKind) schema.GroupVersionResource); ok {
+		r0 = rf(input)
+	} else {
+		r0 = ret.Get(0).(schema.GroupVersionResource)
+	}
+
+	if rf, ok := ret.Get(1).(func(schema.GroupVersionKind) v1.APIResource); ok {
+		r1 = rf(input)
+	} else {
+		r1 = ret.Get(1).(v1.APIResource)
+	}
+
+	if rf, ok := ret.Get(2).(func(schema.GroupVersionKind) error); ok {
+		r2 = rf(input)
+	} else {
+		r2 = ret.Error(2)
+	}
+
+	return r0, r1, r2
+}
+
+// Refresh provides a mock function with given fields:
+func (_m *Helper) Refresh() error {
+	ret := _m.Called()
+
+	var r0 error
+	if rf, ok := ret.Get(0).(func() error); ok {
+		r0 = rf()
+	} else {
+		r0 = ret.Error(0)
+	}
+
+	return r0
+}
+
+// ResourceFor provides a mock function with given fields: input
+func (_m *Helper) ResourceFor(input schema.GroupVersionResource) (schema.GroupVersionResource, v1.APIResource, error) {
+	ret := _m.Called(input)
+
+	var r0 schema.GroupVersionResource
+	var r1 v1.APIResource
+	var r2 error
+	if rf, ok := ret.Get(0).(func(schema.GroupVersionResource) (schema.GroupVersionResource, v1.APIResource, error)); ok {
+		return rf(input)
+	}
+	if rf, ok := ret.Get(0).(func(schema.GroupVersionResource) schema.GroupVersionResource); ok {
+		r0 = rf(input)
+	} else {
+		r0 = ret.Get(0).(schema.GroupVersionResource)
+	}
+
+	if rf, ok := ret.Get(1).(func(schema.GroupVersionResource) v1.APIResource); ok {
+		r1 = rf(input)
+	} else {
+		r1 = ret.Get(1).(v1.APIResource)
+	}
+
+	if rf, ok := ret.Get(2).(func(schema.GroupVersionResource) error); ok {
+		r2 = rf(input)
+	} else {
+		r2 = ret.Error(2)
+	}
+
+	return r0, r1, r2
+}
+
+// Resources provides a mock function with given fields:
+func (_m *Helper) Resources() []*v1.APIResourceList {
+	ret := _m.Called()
+
+	var r0 []*v1.APIResourceList
+	if rf, ok := ret.Get(0).(func() []*v1.APIResourceList); ok {
+		r0 = rf()
+	} else {
+		if ret.Get(0) != nil {
+			r0 = ret.Get(0).([]*v1.APIResourceList)
+		}
+	}
+
+	return r0
+}
+
+// ServerVersion provides a mock function with given fields:
+func (_m *Helper) ServerVersion() *version.Info {
+	ret := _m.Called()
+
+	var r0 *version.Info
+	if rf, ok := ret.Get(0).(func() *version.Info); ok {
+		r0 = rf()
+	} else {
+		if ret.Get(0) != nil {
+			r0 = ret.Get(0).(*version.Info)
+		}
+	}
+
+	return r0
+}
+
+type mockConstructorTestingTNewHelper interface {
+	mock.TestingT
+	Cleanup(func())
+}
+
+// NewHelper creates a new instance of Helper. It also registers a testing interface on the mock and a cleanup function to assert the mocks expectations.
+func NewHelper(t mockConstructorTestingTNewHelper) *Helper {
+	mock := &Helper{}
+	mock.Mock.Test(t)
+
+	t.Cleanup(func() { mock.AssertExpectations(t) })
+
+	return mock
+}