fix: fix e2e using new builder

Signed-off-by: mlycore <[email protected]>
apache · Jul 7, 2023 · 795b74a · 795b74a
1 parent f6e208e
commit 795b74a
Showing 1 changed file with 7 additions and 7 deletions.
diff --git a/shardingsphere-operator/test/e2e/e2e_suite_test.go b/shardingsphere-operator/test/e2e/e2e_suite_test.go
@@ -25,10 +25,11 @@ import (
 
 	"github.com/apache/shardingsphere-on-cloud/shardingsphere-operator/api/v1alpha1"
 	"github.com/apache/shardingsphere-on-cloud/shardingsphere-operator/pkg/controllers"
+	"github.com/apache/shardingsphere-on-cloud/shardingsphere-operator/pkg/kubernetes"
 	mockChaos "github.com/apache/shardingsphere-on-cloud/shardingsphere-operator/pkg/kubernetes/chaosmesh/mocks"
 	"github.com/apache/shardingsphere-on-cloud/shardingsphere-operator/pkg/kubernetes/configmap"
-	"github.com/apache/shardingsphere-on-cloud/shardingsphere-operator/pkg/kubernetes/deployment"
 	"github.com/apache/shardingsphere-on-cloud/shardingsphere-operator/pkg/kubernetes/service"
+	"github.com/apache/shardingsphere-on-cloud/shardingsphere-operator/pkg/reconcile/computenode"
 
 	"github.com/golang/mock/gomock"
 	. "github.com/onsi/ginkgo/v2"
@@ -104,12 +105,11 @@ var _ = BeforeSuite(func() {
 	Expect(err).ToNot(HaveOccurred())
 
 	err = (&controllers.ComputeNodeReconciler{
-		Client:     k8sManager.GetClient(),
-		Scheme:     k8sManager.GetScheme(),
-		Log:        logf.Log.WithName("controllers").WithName("ComputeNode"),
-		Deployment: deployment.NewDeploymentClient(k8sManager.GetClient()),
-		Service:    service.NewServiceClient(k8sManager.GetClient()),
-		ConfigMap:  configmap.NewConfigMapClient(k8sManager.GetClient()),
+		Client:    k8sManager.GetClient(),
+		Scheme:    k8sManager.GetScheme(),
+		Log:       logf.Log.WithName("controllers").WithName("ComputeNode"),
+		Builder:   computenode.NewBuilder(),
+		Resources: kubernetes.NewResources(k8sManager.GetClient()),
 	}).SetupWithManager(k8sManager)
 	Expect(err).ToNot(HaveOccurred())