Merge pull request #6 from christopherChenYW/master

add setup debug session for github action if fail

Author: chriskery

.github/workflows/unittest.yaml

@@ -33,6 +33,13 @@ jobs:
       - name: Run Go test
         run: |
           make test ENVTEST_K8S_VERSION=${{ matrix.kubernetes-version }}
+          
+      - name: Setup Debug Session
+        uses: mxschmitt/action-tmate@v3
+        timeout-minutes: 15
+        if: failure()
+        with:
+          detached: false
 
       - name: Coveralls report
         uses: shogo82148/actions-goveralls@v1

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
 }
+