add fake client for unittest

Author: christopherChenYW

pkg/controllers/builder/configmap_test.go

@@ -14,7 +14,9 @@ import (
 )
 
 func TestBuildHadoopConfigMap(t *testing.T) {
-	mgr := testutil.NewCtrlManager()
+	mgr, envTest := testutil.NewCtrlManager(t)
+	defer envTest.Stop()
+
 	builder := &ConfigMapBuilder{}
 	builder.SetupWithManager(mgr, mgr.GetEventRecorderFor("test"))
 
@@ -66,3 +68,4 @@ func TestBuildHadoopConfigMap(t *testing.T) {
 	assert.Greater(t, len(yarnSite.Properties), 0)
 	assert.Contains(t, yarnSite.Properties, Property{Name: "yarn.nodemanager.resource.cpu-vcores", Value: "1"})
 }
+

pkg/util/testutil/builder.go

@@ -5,8 +5,12 @@ import (
 	"k8s.io/apimachinery/pkg/runtime"
 	utilruntime "k8s.io/apimachinery/pkg/util/runtime"
 	clientgoscheme "k8s.io/client-go/kubernetes/scheme"
-	ctrl "sigs.k8s.io/controller-runtime"
+	"k8s.io/client-go/rest"
+	"sigs.k8s.io/controller-runtime/pkg/client"
+	"sigs.k8s.io/controller-runtime/pkg/client/fake"
+	"sigs.k8s.io/controller-runtime/pkg/envtest"
 	"sigs.k8s.io/controller-runtime/pkg/manager"
+	"testing"
 )
 
 var (
@@ -20,9 +24,25 @@ func init() {
 	//+kubebuilder:scaffold:scheme
 }
 
-func NewCtrlManager() manager.Manager {
-	mgr, _ := ctrl.NewManager(ctrl.GetConfigOrDie(), ctrl.Options{
-		Scheme: scheme,
+func NewCtrlManager(t *testing.T) (manager.Manager, *envtest.Environment) {
+	envTest := &envtest.Environment{}
+	cfg, err := envTest.Start()
+	if err != nil {
+		t.Fatalf("failed to start envtest: %v", err)
+	}
+	defer envTest.Stop()
+
+	// 使用 cfg 创建一个假的 manager.Manager 对象
+	mgr, err := manager.New(cfg, manager.Options{
+		NewClient: func(config *rest.Config, options client.Options) (client.Client, error) {
+			// 返回一个假的客户端
+			return fake.NewClientBuilder().Build(), nil
+		},
 	})
-	return mgr
+
+	if err != nil {
+		t.Fatalf("failed to start envtest: %v", err)
+	}
+	return mgr, envTest
 }
+