@@ -26,6 +26,8 @@ import (
26
26
"sigs.k8s.io/controller-runtime/pkg/controller"
27
27
28
28
"sigs.k8s.io/cluster-api/api/v1beta1/index"
29
+ "sigs.k8s.io/cluster-api/controllers/clustercache"
30
+ "sigs.k8s.io/cluster-api/controllers/remote"
29
31
"sigs.k8s.io/cluster-api/internal/test/envtest"
30
32
)
31
33
@@ -42,12 +44,24 @@ func TestMain(m *testing.M) {
42
44
}
43
45
44
46
setupReconcilers := func (ctx context.Context , mgr ctrl.Manager ) {
45
- machinePoolReconciler := MachinePoolReconciler {
46
- Client : mgr .GetClient (),
47
- recorder : mgr .GetEventRecorderFor ("machinepool-controller" ),
48
- }
49
- err := machinePoolReconciler .SetupWithManager (ctx , mgr , controller.Options {MaxConcurrentReconciles : 1 })
47
+ clusterCache , err := clustercache .SetupWithManager (ctx , mgr , clustercache.Options {
48
+ SecretClient : mgr .GetClient (),
49
+ Cache : clustercache.CacheOptions {
50
+ Indexes : []clustercache.CacheOptionsIndex {clustercache .NodeProviderIDIndex },
51
+ },
52
+ Client : clustercache.ClientOptions {
53
+ UserAgent : remote .DefaultClusterAPIUserAgent ("test-controller-manager" ),
54
+ },
55
+ }, controller.Options {MaxConcurrentReconciles : 10 })
50
56
if err != nil {
57
+ panic (fmt .Sprintf ("Failed to create new cluster cache tracker: %v" , err ))
58
+ }
59
+
60
+ if err := (& MachinePoolReconciler {
61
+ Client : mgr .GetClient (),
62
+ ClusterCache : clusterCache ,
63
+ recorder : mgr .GetEventRecorderFor ("machinepool-controller" ),
64
+ }).SetupWithManager (ctx , mgr , controller.Options {MaxConcurrentReconciles : 1 }); err != nil {
51
65
panic (fmt .Sprintf ("Failed to set up machine pool reconciler: %v" , err ))
52
66
}
53
67
}
0 commit comments