add unittest for leaf_pv_controller

Signed-off-by: duanmengkk <[email protected]>
duanmengkk · Oct 22, 2024 · fe2885c · fe2885c
1 parent 79127ba
commit fe2885c
Showing 1 changed file with 3 additions and 5 deletions.
diff --git a/pkg/clustertree/cluster-manager/controllers/pv/leaf_pv_controller_test.go b/pkg/clustertree/cluster-manager/controllers/pv/leaf_pv_controller_test.go
@@ -1,4 +1,4 @@
-package pv_test
+package pv
 
 import (
 	"context"
@@ -14,19 +14,17 @@ import (
 	ctrl "sigs.k8s.io/controller-runtime"
 	"sigs.k8s.io/controller-runtime/pkg/client/fake"
 	"sigs.k8s.io/controller-runtime/pkg/reconcile"
-
-	"github.com/kosmos.io/kosmos/pkg/clustertree/cluster-manager/controllers/pv"
 )
 
-func setupController() (*pv.LeafPVController, reconcile.Request) {
+func setupController() (*LeafPVController, reconcile.Request) {
 	scheme := runtime.NewScheme()
 	_ = v1.AddToScheme(scheme)
 
 	leafClient := fake.NewClientBuilder().WithScheme(scheme).Build()
 	rootClient := fake.NewClientBuilder().WithScheme(scheme).Build()
 	rootClientSet := fakeclientset.NewSimpleClientset()
 
-	controller := &pv.LeafPVController{
+	controller := &LeafPVController{
 		LeafClient:    leafClient,
 		RootClient:    rootClient,
 		RootClientSet: rootClientSet,