TestMu Conference is Back! Future of Testing + AI
How to use SetNamespace method of config Package
Best Testkube code snippet using config.SetNamespace
reconciler_test.go
Source:reconciler_test.go 
...105 want: controllerruntime.Result{},106 prepare: func(ctx context.Context, fetcher *mocks.MockResourceFetcher, resourceUpdater *mocks.MockResourceUpdater, name string, namespace string) {107 cluster := &anywherev1.Cluster{}108 cluster.SetName(name)109 cluster.SetNamespace(namespace)110 fetcher.EXPECT().FetchCluster(gomock.Any(), gomock.Any()).Return(cluster, nil)111 spec := test.NewFullClusterSpec(t, "testdata/eksa-cluster.yaml")112 cluster.Spec = spec.Cluster.Spec113 fetcher.EXPECT().FetchAppliedSpec(ctx, gomock.Any()).Return(spec, nil)114 fetcher.EXPECT().FetchObject(gomock.Any(), gomock.Any(), gomock.Any()).Do(func(ctx context.Context, objectKey types.NamespacedName, obj client.Object) {115 clusterSpec := &anywherev1.VSphereDatacenterConfig{}116 if err := yaml.Unmarshal([]byte(vsphereDatacenterConfigSpecPath), clusterSpec); err != nil {117 t.Errorf("unmarshal failed: %v", err)118 }119 cluster := obj.(*anywherev1.VSphereDatacenterConfig)120 cluster.SetName(objectKey.Name)121 cluster.SetNamespace(objectKey.Namespace)122 cluster.Spec = clusterSpec.Spec123 assert.Equal(t, objectKey.Name, "test-cluster", "expected Name to be test-cluster")124 }).Return(nil)125 fetcher.EXPECT().FetchObject(gomock.Any(), gomock.Any(), gomock.Any()).Do(func(ctx context.Context, objectKey types.NamespacedName, obj client.Object) {126 clusterSpec := &anywherev1.VSphereMachineConfig{}127 if err := yaml.Unmarshal([]byte(vsphereMachineConfigSpecPath), clusterSpec); err != nil {128 t.Errorf("unmarshal failed: %v", err)129 }130 cluster := obj.(*anywherev1.VSphereMachineConfig)131 cluster.SetName(objectKey.Name)132 cluster.SetNamespace(objectKey.Namespace)133 cluster.Spec = clusterSpec.Spec134 assert.Equal(t, objectKey.Name, "test-cluster", "expected Name to be test-cluster")135 }).Return(nil)136 fetcher.EXPECT().FetchObject(gomock.Any(), gomock.Any(), gomock.Any()).Do(func(ctx context.Context, objectKey types.NamespacedName, obj client.Object) {137 clusterSpec := &anywherev1.VSphereMachineConfig{}138 if err := yaml.Unmarshal([]byte(vsphereMachineConfigSpecPath), clusterSpec); err != nil {139 t.Errorf("unmarshal failed: %v", err)140 }141 cluster := obj.(*anywherev1.VSphereMachineConfig)142 cluster.SetName(objectKey.Name)143 cluster.SetNamespace(objectKey.Namespace)144 cluster.Spec = clusterSpec.Spec145 assert.Equal(t, objectKey.Name, "test-cluster", "expected Name to be test-cluster")146 }).Return(nil)147 fetcher.EXPECT().FetchObject(gomock.Any(), gomock.Any(), gomock.Any()).Do(func(ctx context.Context, objectKey types.NamespacedName, obj client.Object) {148 clusterSpec := &anywherev1.VSphereMachineConfig{}149 if err := yaml.Unmarshal([]byte(vsphereMachineConfigSpecPath), clusterSpec); err != nil {150 t.Errorf("unmarshal failed: %v", err)151 }152 cluster := obj.(*anywherev1.VSphereMachineConfig)153 cluster.SetName(objectKey.Name)154 cluster.SetNamespace(objectKey.Namespace)155 cluster.Spec = clusterSpec.Spec156 assert.Equal(t, objectKey.Name, "test-cluster", "expected Name to be test-cluster")157 }).Return(nil)158 kubeAdmControlPlane := &controlplanev1.KubeadmControlPlane{}159 if err := yaml.Unmarshal([]byte(vsphereKubeadmcontrolplaneFile), kubeAdmControlPlane); err != nil {160 t.Errorf("unmarshal failed: %v", err)161 }162 etcdadmCluster := &etcdv1.EtcdadmCluster{}163 if err := yaml.Unmarshal([]byte(vsphereEtcdadmclusterFile), etcdadmCluster); err != nil {164 t.Errorf("unmarshal failed: %v", err)165 }166 machineDeployment := &clusterv1.MachineDeployment{}167 if err := yaml.Unmarshal([]byte(vsphereMachineDeploymentFile), machineDeployment); err != nil {168 t.Errorf("unmarshalling machinedeployment failed: %v", err)169 }170 workerNodeMachineConfig := &anywherev1.VSphereMachineConfig{171 Spec: anywherev1.VSphereMachineConfigSpec{172 Users: []anywherev1.UserConfiguration{173 {174 Name: "capv",175 SshAuthorizedKeys: []string{"ssh-rsa ssh_key_value"},176 },177 },178 },179 }180 fetcher.EXPECT().MachineDeployment(ctx, gomock.Any(), gomock.Any()).Return(machineDeployment, nil)181 fetcher.EXPECT().Etcd(ctx, gomock.Any()).Return(etcdadmCluster, nil)182 fetcher.EXPECT().ExistingVSphereDatacenterConfig(ctx, gomock.Any(), gomock.Any()).Return(&anywherev1.VSphereDatacenterConfig{}, nil)183 fetcher.EXPECT().ExistingVSphereControlPlaneMachineConfig(ctx, gomock.Any()).Return(&anywherev1.VSphereMachineConfig{}, nil)184 fetcher.EXPECT().ExistingVSphereEtcdMachineConfig(ctx, gomock.Any()).Return(&anywherev1.VSphereMachineConfig{}, nil)185 fetcher.EXPECT().ExistingVSphereWorkerMachineConfig(ctx, gomock.Any(), gomock.Any()).Return(workerNodeMachineConfig, nil)186 fetcher.EXPECT().ExistingWorkerNodeGroupConfig(ctx, gomock.Any(), gomock.Any()).Return(&anywherev1.WorkerNodeGroupConfiguration{}, nil)187 fetcher.EXPECT().VSphereCredentials(ctx).Return(&corev1.Secret{188 Data: map[string][]byte{"username": []byte("username"), "password": []byte("password")},189 }, nil)190 fetcher.EXPECT().Fetch(ctx, gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).AnyTimes().Return(nil, errors.NewNotFound(schema.GroupResource{Group: "testgroup", Resource: "testresource"}, ""))191 resourceUpdater.EXPECT().ApplyPatch(ctx, gomock.Any(), false).Return(nil)192 resourceUpdater.EXPECT().ForceApplyTemplate(ctx, gomock.Any(), gomock.Any()).Do(func(ctx context.Context, template *unstructured.Unstructured, dryRun bool) {193 assert.Equal(t, false, dryRun, "Expected dryRun didn't match")194 switch template.GetKind() {195 case "VSphereMachineTemplate":196 if strings.Contains(template.GetName(), "worker-node") {197 expectedMachineTemplate := &unstructured.Unstructured{}198 if err := yaml.Unmarshal([]byte(vsphereMachineTemplateFile), expectedMachineTemplate); err != nil {199 t.Errorf("unmarshal failed: %v", err)200 }201 assert.Equal(t, expectedMachineTemplate, template, "values", expectedMachineTemplate, template)202 }203 case "MachineDeployment":204 expectedMCDeployment := &unstructured.Unstructured{}205 if err := yaml.Unmarshal([]byte(expectedVSphereMachineDeploymentFile), expectedMCDeployment); err != nil {206 t.Errorf("unmarshal failed: %v", err)207 }208 assert.Equal(t, expectedMCDeployment, template, "values", expectedMCDeployment, template)209 }210 }).AnyTimes().Return(nil)211 },212 },213 {214 name: "worker node reconcile (Vsphere provider) - worker nodes has NO machine-template changes",215 args: args{216 namespace: "namespaceA",217 name: "nameA",218 objectKey: types.NamespacedName{219 Name: "nameA",220 Namespace: "namespaceA",221 },222 },223 want: controllerruntime.Result{},224 prepare: func(ctx context.Context, fetcher *mocks.MockResourceFetcher, resourceUpdater *mocks.MockResourceUpdater, name string, namespace string) {225 cluster := &anywherev1.Cluster{}226 cluster.SetName(name)227 cluster.SetNamespace(namespace)228 fetcher.EXPECT().FetchCluster(gomock.Any(), gomock.Any()).Return(cluster, nil)229 spec := test.NewFullClusterSpec(t, "testdata/eksa-cluster_no_changes.yaml")230 cluster.Spec = spec.Cluster.Spec231 fetcher.EXPECT().FetchAppliedSpec(ctx, gomock.Any()).Return(spec, nil)232 datacenterSpec := &anywherev1.VSphereDatacenterConfig{}233 if err := yaml.Unmarshal([]byte(vsphereDatacenterConfigSpecPath), datacenterSpec); err != nil {234 t.Errorf("unmarshal failed: %v", err)235 }236 fetcher.EXPECT().FetchObject(gomock.Any(), gomock.Any(), gomock.Any()).Do(func(ctx context.Context, objectKey types.NamespacedName, obj client.Object) {237 cluster := obj.(*anywherev1.VSphereDatacenterConfig)238 cluster.SetName(objectKey.Name)239 cluster.SetNamespace(objectKey.Name)240 cluster.Spec = datacenterSpec.Spec241 assert.Equal(t, objectKey.Name, "test-cluster", "expected Name to be test-cluster")242 }).Return(nil)243 existingVSDatacenter := &anywherev1.VSphereDatacenterConfig{}244 existingVSDatacenter.Spec = datacenterSpec.Spec245 fetcher.EXPECT().ExistingVSphereDatacenterConfig(ctx, gomock.Any(), gomock.Any()).Return(existingVSDatacenter, nil)246 machineSpec := &anywherev1.VSphereMachineConfig{}247 if err := yaml.Unmarshal([]byte(vsphereMachineConfigSpecPath), machineSpec); err != nil {248 t.Errorf("unmarshal failed: %v", err)249 }250 fetcher.EXPECT().FetchObject(gomock.Any(), gomock.Any(), gomock.Any()).Do(func(ctx context.Context, objectKey types.NamespacedName, obj client.Object) {251 cluster := obj.(*anywherev1.VSphereMachineConfig)252 cluster.SetName(objectKey.Name)253 cluster.SetNamespace(objectKey.Namespace)254 cluster.Spec = machineSpec.Spec255 assert.Equal(t, objectKey.Name, "test-cluster", "expected Name to be test-cluster")256 }).Return(nil)257 fetcher.EXPECT().FetchObject(gomock.Any(), gomock.Any(), gomock.Any()).Do(func(ctx context.Context, objectKey types.NamespacedName, obj client.Object) {258 cluster := obj.(*anywherev1.VSphereMachineConfig)259 cluster.SetName(objectKey.Name)260 cluster.SetNamespace(objectKey.Namespace)261 cluster.Spec = machineSpec.Spec262 assert.Equal(t, objectKey.Name, "test-cluster", "expected Name to be test-cluster")263 }).Return(nil)264 existingWorkerNodeGroupConfiguration := &anywherev1.WorkerNodeGroupConfiguration{265 Name: "md-0",266 Count: 3,267 MachineGroupRef: nil,268 }269 fetcher.EXPECT().ExistingWorkerNodeGroupConfig(ctx, gomock.Any(), gomock.Any()).Return(existingWorkerNodeGroupConfiguration, nil)270 existingVSMachine := &anywherev1.VSphereMachineConfig{}271 existingVSMachine.Spec = machineSpec.Spec272 workerNodeMachineConfig := &anywherev1.VSphereMachineConfig{273 Spec: anywherev1.VSphereMachineConfigSpec{274 Users: []anywherev1.UserConfiguration{275 {276 Name: "capv",277 SshAuthorizedKeys: []string{"ssh-rsa ssh_key_value"},278 },279 },280 },281 }282 fetcher.EXPECT().ExistingVSphereControlPlaneMachineConfig(ctx, gomock.Any()).Return(&anywherev1.VSphereMachineConfig{}, nil)283 fetcher.EXPECT().ExistingVSphereWorkerMachineConfig(ctx, gomock.Any(), gomock.Any()).Return(workerNodeMachineConfig, nil)284 kubeAdmControlPlane := &controlplanev1.KubeadmControlPlane{}285 if err := yaml.Unmarshal([]byte(vsphereKubeadmcontrolplaneFile), kubeAdmControlPlane); err != nil {286 t.Errorf("unmarshal failed: %v", err)287 }288 machineDeployment := &clusterv1.MachineDeployment{}289 if err := yaml.Unmarshal([]byte(vsphereMachineDeploymentFile), machineDeployment); err != nil {290 t.Errorf("unmarshal failed: %v", err)291 }292 fetcher.EXPECT().MachineDeployment(ctx, gomock.Any(), gomock.Any()).Return(machineDeployment, nil)293 fetcher.EXPECT().VSphereCredentials(ctx).Return(&corev1.Secret{294 Data: map[string][]byte{"username": []byte("username"), "password": []byte("password")},295 }, nil)296 fetcher.EXPECT().Fetch(ctx, gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).AnyTimes().Return(nil, errors.NewNotFound(schema.GroupResource{Group: "testgroup", Resource: "testresource"}, ""))297 resourceUpdater.EXPECT().ForceApplyTemplate(ctx, gomock.Any(), gomock.Any()).Do(func(ctx context.Context, template *unstructured.Unstructured, dryRun bool) {298 assert.Equal(t, false, dryRun, "Expected dryRun didn't match")299 println(template.GetName(), " : ", template.GetKind())300 switch template.GetKind() {301 case "MachineDeployment":302 expectedMCDeployment := &unstructured.Unstructured{}303 if err := yaml.Unmarshal([]byte(expectedVSphereMachineDeploymentOnlyReplica), expectedMCDeployment); err != nil {304 t.Errorf("unmarshal failed: %v", err)305 }306 assert.Equal(t, expectedMCDeployment, template, "values", expectedMCDeployment, template)307 }308 }).AnyTimes().Return(nil)309 },310 },311 {312 name: "worker node reconcile (Vsphere provider) - worker node taints have changed",313 args: args{314 namespace: "namespaceA",315 name: "nameA",316 objectKey: types.NamespacedName{317 Name: "nameA",318 Namespace: "namespaceA",319 },320 },321 want: controllerruntime.Result{},322 prepare: func(ctx context.Context, fetcher *mocks.MockResourceFetcher, resourceUpdater *mocks.MockResourceUpdater, name string, namespace string) {323 cluster := &anywherev1.Cluster{}324 cluster.SetName(name)325 cluster.SetNamespace(namespace)326 fetcher.EXPECT().FetchCluster(gomock.Any(), gomock.Any()).Return(cluster, nil)327 spec := test.NewFullClusterSpec(t, "testdata/eksa-cluster.yaml")328 cluster.Spec = spec.Cluster.Spec329 fetcher.EXPECT().FetchAppliedSpec(ctx, gomock.Any()).Return(spec, nil)330 fetcher.EXPECT().FetchObject(gomock.Any(), gomock.Any(), gomock.Any()).Do(func(ctx context.Context, objectKey types.NamespacedName, obj client.Object) {331 clusterSpec := &anywherev1.VSphereDatacenterConfig{}332 if err := yaml.Unmarshal([]byte(vsphereDatacenterConfigSpecPath), clusterSpec); err != nil {333 t.Errorf("unmarshal failed: %v", err)334 }335 cluster := obj.(*anywherev1.VSphereDatacenterConfig)336 cluster.SetName(objectKey.Name)337 cluster.SetNamespace(objectKey.Namespace)338 cluster.Spec = clusterSpec.Spec339 assert.Equal(t, objectKey.Name, "test-cluster", "expected Name to be test-cluster")340 }).Return(nil)341 fetcher.EXPECT().FetchObject(gomock.Any(), gomock.Any(), gomock.Any()).Do(func(ctx context.Context, objectKey types.NamespacedName, obj client.Object) {342 clusterSpec := &anywherev1.VSphereMachineConfig{}343 if err := yaml.Unmarshal([]byte(vsphereMachineConfigSpecPath), clusterSpec); err != nil {344 t.Errorf("unmarshal failed: %v", err)345 }346 cluster := obj.(*anywherev1.VSphereMachineConfig)347 cluster.SetName(objectKey.Name)348 cluster.SetNamespace(objectKey.Namespace)349 cluster.Spec = clusterSpec.Spec350 assert.Equal(t, objectKey.Name, "test-cluster", "expected Name to be test-cluster")351 }).Return(nil)352 fetcher.EXPECT().FetchObject(gomock.Any(), gomock.Any(), gomock.Any()).Do(func(ctx context.Context, objectKey types.NamespacedName, obj client.Object) {353 clusterSpec := &anywherev1.VSphereMachineConfig{}354 if err := yaml.Unmarshal([]byte(vsphereMachineConfigSpecPath), clusterSpec); err != nil {355 t.Errorf("unmarshal failed: %v", err)356 }357 cluster := obj.(*anywherev1.VSphereMachineConfig)358 cluster.SetName(objectKey.Name)359 cluster.SetNamespace(objectKey.Namespace)360 cluster.Spec = clusterSpec.Spec361 assert.Equal(t, objectKey.Name, "test-cluster", "expected Name to be test-cluster")362 }).Return(nil)363 fetcher.EXPECT().FetchObject(gomock.Any(), gomock.Any(), gomock.Any()).Do(func(ctx context.Context, objectKey types.NamespacedName, obj client.Object) {364 clusterSpec := &anywherev1.VSphereMachineConfig{}365 if err := yaml.Unmarshal([]byte(vsphereMachineConfigSpecPath), clusterSpec); err != nil {366 t.Errorf("unmarshal failed: %v", err)367 }368 cluster := obj.(*anywherev1.VSphereMachineConfig)369 cluster.SetName(objectKey.Name)370 cluster.SetNamespace(objectKey.Namespace)371 cluster.Spec = clusterSpec.Spec372 assert.Equal(t, objectKey.Name, "test-cluster", "expected Name to be test-cluster")373 }).Return(nil)374 kubeAdmControlPlane := &controlplanev1.KubeadmControlPlane{}375 if err := yaml.Unmarshal([]byte(vsphereKubeadmcontrolplaneFile), kubeAdmControlPlane); err != nil {376 t.Errorf("unmarshal failed: %v", err)377 }378 etcdadmCluster := &etcdv1.EtcdadmCluster{}379 if err := yaml.Unmarshal([]byte(vsphereEtcdadmclusterFile), etcdadmCluster); err != nil {380 t.Errorf("unmarshal failed: %v", err)381 }382 existingWorkerNodeGroupConfiguration := &anywherev1.WorkerNodeGroupConfiguration{383 Name: "md-0",384 Count: 3,385 MachineGroupRef: nil,386 Taints: []corev1.Taint{387 {388 Key: "key1",389 Value: "val1",390 Effect: "PreferNoSchedule",391 },392 },393 }394 fetcher.EXPECT().Etcd(ctx, gomock.Any()).Return(etcdadmCluster, nil)395 fetcher.EXPECT().ExistingVSphereDatacenterConfig(ctx, gomock.Any(), gomock.Any()).Return(&anywherev1.VSphereDatacenterConfig{}, nil)396 fetcher.EXPECT().ExistingVSphereControlPlaneMachineConfig(ctx, gomock.Any()).Return(&anywherev1.VSphereMachineConfig{}, nil)397 fetcher.EXPECT().ExistingVSphereEtcdMachineConfig(ctx, gomock.Any()).Return(&anywherev1.VSphereMachineConfig{}, nil)398 fetcher.EXPECT().ExistingVSphereWorkerMachineConfig(ctx, gomock.Any(), gomock.Any()).Return(&anywherev1.VSphereMachineConfig{}, nil)399 fetcher.EXPECT().ExistingWorkerNodeGroupConfig(ctx, gomock.Any(), gomock.Any()).Return(existingWorkerNodeGroupConfiguration, nil)400 fetcher.EXPECT().VSphereCredentials(ctx).Return(&corev1.Secret{401 Data: map[string][]byte{"username": []byte("username"), "password": []byte("password")},402 }, nil)403 fetcher.EXPECT().Fetch(ctx, gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).AnyTimes().Return(nil, errors.NewNotFound(schema.GroupResource{Group: "testgroup", Resource: "testresource"}, ""))404 resourceUpdater.EXPECT().ApplyPatch(ctx, gomock.Any(), false).Return(nil)405 resourceUpdater.EXPECT().ForceApplyTemplate(ctx, gomock.Any(), gomock.Any()).Do(func(ctx context.Context, template *unstructured.Unstructured, dryRun bool) {406 assert.Equal(t, false, dryRun, "Expected dryRun didn't match")407 switch template.GetKind() {408 case "KubeadmConfigTemplate":409 existingKubeadmConfigTemplate := &unstructured.Unstructured{}410 if err := yaml.Unmarshal([]byte(vsphereKubeadmconfigTemplateSpecPath), existingKubeadmConfigTemplate); err != nil {411 t.Errorf("unmarshal failed: %v", err)412 }413 assert.Equal(t, existingKubeadmConfigTemplate, template, "values", existingKubeadmConfigTemplate, template)414 case "MachineDeployment":415 expectedMCDeployment := &unstructured.Unstructured{}416 if err := yaml.Unmarshal([]byte(expectedVSphereMachineDeploymentTemplateChanged), expectedMCDeployment); err != nil {417 t.Errorf("unmarshal failed: %v", err)418 }419 assert.Equal(t, expectedMCDeployment, template, "values", expectedMCDeployment, template)420 }421 }).AnyTimes().Return(nil)422 },423 },424 {425 name: "worker node reconcile (Vsphere provider) - worker node labels have changed",426 args: args{427 namespace: "namespaceA",428 name: "nameA",429 objectKey: types.NamespacedName{430 Name: "nameA",431 Namespace: "namespaceA",432 },433 },434 want: controllerruntime.Result{},435 prepare: func(ctx context.Context, fetcher *mocks.MockResourceFetcher, resourceUpdater *mocks.MockResourceUpdater, name string, namespace string) {436 cluster := &anywherev1.Cluster{}437 cluster.SetName(name)438 cluster.SetNamespace(namespace)439 fetcher.EXPECT().FetchCluster(gomock.Any(), gomock.Any()).Return(cluster, nil)440 spec := test.NewFullClusterSpec(t, "testdata/eksa-cluster.yaml")441 cluster.Spec = spec.Cluster.Spec442 fetcher.EXPECT().FetchAppliedSpec(ctx, gomock.Any()).Return(spec, nil)443 fetcher.EXPECT().FetchObject(gomock.Any(), gomock.Any(), gomock.Any()).Do(func(ctx context.Context, objectKey types.NamespacedName, obj client.Object) {444 clusterSpec := &anywherev1.VSphereDatacenterConfig{}445 if err := yaml.Unmarshal([]byte(vsphereDatacenterConfigSpecPath), clusterSpec); err != nil {446 t.Errorf("unmarshal failed: %v", err)447 }448 cluster := obj.(*anywherev1.VSphereDatacenterConfig)449 cluster.SetName(objectKey.Name)450 cluster.SetNamespace(objectKey.Namespace)451 cluster.Spec = clusterSpec.Spec452 assert.Equal(t, objectKey.Name, "test-cluster", "expected Name to be test-cluster")453 }).Return(nil)454 fetcher.EXPECT().FetchObject(gomock.Any(), gomock.Any(), gomock.Any()).Do(func(ctx context.Context, objectKey types.NamespacedName, obj client.Object) {455 clusterSpec := &anywherev1.VSphereMachineConfig{}456 if err := yaml.Unmarshal([]byte(vsphereMachineConfigSpecPath), clusterSpec); err != nil {457 t.Errorf("unmarshal failed: %v", err)458 }459 cluster := obj.(*anywherev1.VSphereMachineConfig)460 cluster.SetName(objectKey.Name)461 cluster.SetNamespace(objectKey.Namespace)462 cluster.Spec = clusterSpec.Spec463 assert.Equal(t, objectKey.Name, "test-cluster", "expected Name to be test-cluster")464 }).Return(nil)465 fetcher.EXPECT().FetchObject(gomock.Any(), gomock.Any(), gomock.Any()).Do(func(ctx context.Context, objectKey types.NamespacedName, obj client.Object) {466 clusterSpec := &anywherev1.VSphereMachineConfig{}467 if err := yaml.Unmarshal([]byte(vsphereMachineConfigSpecPath), clusterSpec); err != nil {468 t.Errorf("unmarshal failed: %v", err)469 }470 cluster := obj.(*anywherev1.VSphereMachineConfig)471 cluster.SetName(objectKey.Name)472 cluster.SetNamespace(objectKey.Namespace)473 cluster.Spec = clusterSpec.Spec474 assert.Equal(t, objectKey.Name, "test-cluster", "expected Name to be test-cluster")475 }).Return(nil)476 fetcher.EXPECT().FetchObject(gomock.Any(), gomock.Any(), gomock.Any()).Do(func(ctx context.Context, objectKey types.NamespacedName, obj client.Object) {477 clusterSpec := &anywherev1.VSphereMachineConfig{}478 if err := yaml.Unmarshal([]byte(vsphereMachineConfigSpecPath), clusterSpec); err != nil {479 t.Errorf("unmarshal failed: %v", err)480 }481 cluster := obj.(*anywherev1.VSphereMachineConfig)482 cluster.SetName(objectKey.Name)483 cluster.SetNamespace(objectKey.Namespace)484 cluster.Spec = clusterSpec.Spec485 assert.Equal(t, objectKey.Name, "test-cluster", "expected Name to be test-cluster")486 }).Return(nil)487 kubeAdmControlPlane := &controlplanev1.KubeadmControlPlane{}488 if err := yaml.Unmarshal([]byte(vsphereKubeadmcontrolplaneFile), kubeAdmControlPlane); err != nil {489 t.Errorf("unmarshal failed: %v", err)490 }491 etcdadmCluster := &etcdv1.EtcdadmCluster{}492 if err := yaml.Unmarshal([]byte(vsphereEtcdadmclusterFile), etcdadmCluster); err != nil {493 t.Errorf("unmarshal failed: %v", err)494 }495 existingWorkerNodeGroupConfiguration := &anywherev1.WorkerNodeGroupConfiguration{496 Name: "md-0",497 Count: 3,498 MachineGroupRef: nil,499 Labels: map[string]string{500 "Key1": "Val1",501 "Key2": "Val2",502 },503 }504 fetcher.EXPECT().Etcd(ctx, gomock.Any()).Return(etcdadmCluster, nil)505 fetcher.EXPECT().ExistingVSphereDatacenterConfig(ctx, gomock.Any(), gomock.Any()).Return(&anywherev1.VSphereDatacenterConfig{}, nil)506 fetcher.EXPECT().ExistingVSphereControlPlaneMachineConfig(ctx, gomock.Any()).Return(&anywherev1.VSphereMachineConfig{}, nil)507 fetcher.EXPECT().ExistingVSphereEtcdMachineConfig(ctx, gomock.Any()).Return(&anywherev1.VSphereMachineConfig{}, nil)508 fetcher.EXPECT().ExistingVSphereWorkerMachineConfig(ctx, gomock.Any(), gomock.Any()).Return(&anywherev1.VSphereMachineConfig{}, nil)509 fetcher.EXPECT().ExistingWorkerNodeGroupConfig(ctx, gomock.Any(), gomock.Any()).Return(existingWorkerNodeGroupConfiguration, nil)510 fetcher.EXPECT().VSphereCredentials(ctx).Return(&corev1.Secret{511 Data: map[string][]byte{"username": []byte("username"), "password": []byte("password")},512 }, nil)513 fetcher.EXPECT().Fetch(ctx, gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).AnyTimes().Return(nil, errors.NewNotFound(schema.GroupResource{Group: "testgroup", Resource: "testresource"}, ""))514 resourceUpdater.EXPECT().ApplyPatch(ctx, gomock.Any(), false).Return(nil)515 resourceUpdater.EXPECT().ForceApplyTemplate(ctx, gomock.Any(), gomock.Any()).Do(func(ctx context.Context, template *unstructured.Unstructured, dryRun bool) {516 assert.Equal(t, false, dryRun, "Expected dryRun didn't match")517 switch template.GetKind() {518 case "KubeadmConfigTemplate":519 existingKubeadmConfigTemplate := &unstructured.Unstructured{}520 if err := yaml.Unmarshal([]byte(vsphereKubeadmconfigTemplateSpecPath), existingKubeadmConfigTemplate); err != nil {521 t.Errorf("unmarshal failed: %v", err)522 }523 assert.Equal(t, existingKubeadmConfigTemplate, template, "values", existingKubeadmConfigTemplate, template)524 case "MachineDeployment":525 expectedMCDeployment := &unstructured.Unstructured{}526 if err := yaml.Unmarshal([]byte(expectedVSphereMachineDeploymentTemplateChanged), expectedMCDeployment); err != nil {527 t.Errorf("unmarshal failed: %v", err)528 }529 assert.Equal(t, expectedMCDeployment, template, "values", expectedMCDeployment, template)530 }531 }).AnyTimes().Return(nil)532 },533 },534 }535 for _, tt := range tests {536 t.Run(tt.name, func(t *testing.T) {537 ctx := context.Background()538 mockCtrl := gomock.NewController(t)539 fetcher := mocks.NewMockResourceFetcher(mockCtrl)540 resourceUpdater := mocks.NewMockResourceUpdater(mockCtrl)541 tt.prepare(ctx, fetcher, resourceUpdater, tt.args.name, tt.args.namespace)542 cor := resource.NewClusterReconciler(fetcher, resourceUpdater, test.FakeNow, logr.Discard())543 if err := cor.Reconcile(ctx, tt.args.objectKey, false); (err != nil) != tt.wantErr {544 t.Errorf("Reconcile() error = %v, wantErr %v", err, tt.wantErr)545 }546 })547 }548}549func TestClusterReconcilerReconcileCloudStack(t *testing.T) {550 type args struct {551 objectKey types.NamespacedName552 name string553 namespace string554 }555 tests := []struct {556 name string557 args args558 want controllerruntime.Result559 wantErr bool560 prepare func(context.Context, *mocks.MockResourceFetcher, *mocks.MockResourceUpdater, string, string)561 }{562 {563 name: "worker node reconcile (Cloudstack provider) - worker nodes has changes",564 args: args{565 namespace: "namespaceA",566 name: "nameA",567 objectKey: types.NamespacedName{568 Name: "nameA",569 Namespace: "namespaceA",570 },571 },572 want: controllerruntime.Result{},573 prepare: func(ctx context.Context, fetcher *mocks.MockResourceFetcher, resourceUpdater *mocks.MockResourceUpdater, name string, namespace string) {574 cluster := &anywherev1.Cluster{}575 cluster.SetName(name)576 cluster.SetNamespace(namespace)577 fetcher.EXPECT().FetchCluster(gomock.Any(), gomock.Any()).Return(cluster, nil)578 spec := test.NewFullClusterSpec(t, "testdata/eksa-cluster-cloudstack.yaml")579 cluster.Spec = spec.Cluster.Spec580 fetcher.EXPECT().FetchAppliedSpec(ctx, gomock.Any()).Return(spec, nil)581 fetcher.EXPECT().FetchObject(gomock.Any(), gomock.Any(), gomock.Any()).Do(func(ctx context.Context, objectKey types.NamespacedName, obj client.Object) {582 clusterSpec := &anywherev1.CloudStackDatacenterConfig{}583 if err := yaml.Unmarshal([]byte(cloudstackDatacenterConfigSpecPath), clusterSpec); err != nil {584 t.Errorf("unmarshal failed: %v", err)585 }586 cluster := obj.(*anywherev1.CloudStackDatacenterConfig)587 cluster.SetName(objectKey.Name)588 cluster.SetNamespace(objectKey.Namespace)589 cluster.Spec = clusterSpec.Spec590 assert.Equal(t, objectKey.Name, "test-cluster", "expected Name to be test-cluster")591 }).Return(nil)592 fetcher.EXPECT().FetchObject(gomock.Any(), gomock.Any(), gomock.Any()).Do(func(ctx context.Context, objectKey types.NamespacedName, obj client.Object) {593 clusterSpec := &anywherev1.CloudStackMachineConfig{}594 if err := yaml.Unmarshal([]byte(cloudstackMachineConfigSpecPath), clusterSpec); err != nil {595 t.Errorf("unmarshal failed: %v", err)596 }597 cluster := obj.(*anywherev1.CloudStackMachineConfig)598 cluster.SetName(objectKey.Name)599 cluster.SetNamespace(objectKey.Namespace)600 cluster.Spec = clusterSpec.Spec601 assert.Equal(t, objectKey.Name, "test-cluster", "expected Name to be test-cluster")602 }).Return(nil)603 fetcher.EXPECT().FetchObject(gomock.Any(), gomock.Any(), gomock.Any()).Do(func(ctx context.Context, objectKey types.NamespacedName, obj client.Object) {604 clusterSpec := &anywherev1.CloudStackMachineConfig{}605 if err := yaml.Unmarshal([]byte(cloudstackMachineConfigSpecPath), clusterSpec); err != nil {606 t.Errorf("unmarshal failed: %v", err)607 }608 cluster := obj.(*anywherev1.CloudStackMachineConfig)609 cluster.SetName(objectKey.Name)610 cluster.SetNamespace(objectKey.Namespace)611 cluster.Spec = clusterSpec.Spec612 assert.Equal(t, objectKey.Name, "test-cluster", "expected Name to be test-cluster")613 }).Return(nil)614 kubeAdmControlPlane := &controlplanev1.KubeadmControlPlane{}615 if err := yaml.Unmarshal([]byte(cloudstackKubeadmcontrolplaneFile), kubeAdmControlPlane); err != nil {616 t.Errorf("unmarshal failed: %v", err)617 }618 machineDeployment := &clusterv1.MachineDeployment{}619 if err := yaml.Unmarshal([]byte(cloudstackMachineDeploymentFile), machineDeployment); err != nil {620 t.Errorf("unmarshalling machinedeployment failed: %v", err)621 }622 fetcher.EXPECT().MachineDeployment(ctx, gomock.Any(), gomock.Any()).Return(machineDeployment, nil)623 fetcher.EXPECT().ExistingCloudStackDatacenterConfig(ctx, gomock.Any(), gomock.Any()).Return(&anywherev1.CloudStackDatacenterConfig{}, nil)624 fetcher.EXPECT().ExistingCloudStackControlPlaneMachineConfig(ctx, gomock.Any()).Return(&anywherev1.CloudStackMachineConfig{}, nil)625 fetcher.EXPECT().ExistingCloudStackWorkerMachineConfig(ctx, gomock.Any(), gomock.Any()).Return(&anywherev1.CloudStackMachineConfig{}, nil)626 fetcher.EXPECT().ExistingWorkerNodeGroupConfig(ctx, gomock.Any(), gomock.Any()).Return(&anywherev1.WorkerNodeGroupConfiguration{}, nil)627 fetcher.EXPECT().Fetch(ctx, gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).AnyTimes().Return(nil, errors.NewNotFound(schema.GroupResource{Group: "testgroup", Resource: "testresource"}, ""))628 resourceUpdater.EXPECT().ForceApplyTemplate(ctx, gomock.Any(), gomock.Any()).Do(func(ctx context.Context, template *unstructured.Unstructured, dryRun bool) {629 assert.Equal(t, false, dryRun, "Expected dryRun didn't match")630 switch template.GetKind() {631 case "CloudStackMachineTemplate":632 if strings.Contains(template.GetName(), "worker-node") {633 expectedMachineTemplate := &unstructured.Unstructured{}634 if err := yaml.Unmarshal([]byte(cloudstackMachineTemplateFile), expectedMachineTemplate); err != nil {635 t.Errorf("unmarshal failed: %v", err)636 }637 assert.Equal(t, expectedMachineTemplate, template, "values", expectedMachineTemplate, template)638 }639 case "MachineDeployment":640 expectedMCDeployment := &unstructured.Unstructured{}641 if err := yaml.Unmarshal([]byte(expectedCloudStackMachineDeploymentFile), expectedMCDeployment); err != nil {642 t.Errorf("unmarshal failed: %v", err)643 }644 assert.Equal(t, expectedMCDeployment, template, "values", expectedMCDeployment, template)645 }646 }).AnyTimes().Return(nil)647 },648 },649 {650 name: "worker node reconcile (Cloudstack provider) - worker nodes has NO machine-template changes",651 args: args{652 namespace: "namespaceA",653 name: "nameA",654 objectKey: types.NamespacedName{655 Name: "nameA",656 Namespace: "namespaceA",657 },658 },659 want: controllerruntime.Result{},660 prepare: func(ctx context.Context, fetcher *mocks.MockResourceFetcher, resourceUpdater *mocks.MockResourceUpdater, name string, namespace string) {661 cluster := &anywherev1.Cluster{}662 cluster.SetName(name)663 cluster.SetNamespace(namespace)664 fetcher.EXPECT().FetchCluster(gomock.Any(), gomock.Any()).Return(cluster, nil)665 spec := test.NewFullClusterSpec(t, "testdata/eksa-cluster-cloudstack_no_changes.yaml")666 cluster.Spec = spec.Cluster.Spec667 fetcher.EXPECT().FetchAppliedSpec(ctx, gomock.Any()).Return(spec, nil)668 datacenterSpec := &anywherev1.CloudStackDatacenterConfig{}669 if err := yaml.Unmarshal([]byte(cloudstackDatacenterConfigSpecPath), datacenterSpec); err != nil {670 t.Errorf("unmarshal failed: %v", err)671 }672 fetcher.EXPECT().FetchObject(gomock.Any(), gomock.Any(), gomock.Any()).Do(func(ctx context.Context, objectKey types.NamespacedName, obj client.Object) {673 cluster := obj.(*anywherev1.CloudStackDatacenterConfig)674 cluster.SetName(objectKey.Name)675 cluster.SetNamespace(objectKey.Name)676 cluster.Spec = datacenterSpec.Spec677 assert.Equal(t, objectKey.Name, "test-cluster", "expected Name to be test-cluster")678 }).Return(nil)679 machineSpec := &anywherev1.CloudStackMachineConfig{}680 if err := yaml.Unmarshal([]byte(cloudstackMachineConfigSpecPath), machineSpec); err != nil {681 t.Errorf("unmarshal failed: %v", err)682 }683 fetcher.EXPECT().FetchObject(gomock.Any(), gomock.Any(), gomock.Any()).Do(func(ctx context.Context, objectKey types.NamespacedName, obj client.Object) {684 cluster := obj.(*anywherev1.CloudStackMachineConfig)685 cluster.SetName(objectKey.Name)686 cluster.SetNamespace(objectKey.Namespace)687 cluster.Spec = machineSpec.Spec688 assert.Equal(t, objectKey.Name, "test-cluster", "expected Name to be test-cluster")689 }).Return(nil)690 fetcher.EXPECT().FetchObject(gomock.Any(), gomock.Any(), gomock.Any()).Do(func(ctx context.Context, objectKey types.NamespacedName, obj client.Object) {691 cluster := obj.(*anywherev1.CloudStackMachineConfig)692 cluster.SetName(objectKey.Name)693 cluster.SetNamespace(objectKey.Namespace)694 cluster.Spec = machineSpec.Spec695 assert.Equal(t, objectKey.Name, "test-cluster", "expected Name to be test-cluster")696 }).Return(nil)697 existingWorkerNodeGroupConfiguration := &anywherev1.WorkerNodeGroupConfiguration{698 Name: "md-0",699 Count: 3,700 MachineGroupRef: nil,701 }702 fetcher.EXPECT().ExistingWorkerNodeGroupConfig(ctx, gomock.Any(), gomock.Any()).Return(existingWorkerNodeGroupConfiguration, nil)703 existingCSMachine := &anywherev1.CloudStackMachineConfig{}704 existingCSMachine.Spec = machineSpec.Spec705 fetcher.EXPECT().ExistingCloudStackDatacenterConfig(ctx, gomock.Any(), gomock.Any()).Return(&anywherev1.CloudStackDatacenterConfig{}, nil)706 fetcher.EXPECT().ExistingCloudStackControlPlaneMachineConfig(ctx, gomock.Any()).Return(&anywherev1.CloudStackMachineConfig{}, nil)707 fetcher.EXPECT().ExistingCloudStackWorkerMachineConfig(ctx, gomock.Any(), gomock.Any()).Return(&anywherev1.CloudStackMachineConfig{}, nil)708 kubeAdmControlPlane := &controlplanev1.KubeadmControlPlane{}709 if err := yaml.Unmarshal([]byte(cloudstackKubeadmcontrolplaneFile), kubeAdmControlPlane); err != nil {710 t.Errorf("unmarshal failed: %v", err)711 }712 machineDeployment := &clusterv1.MachineDeployment{}713 if err := yaml.Unmarshal([]byte(cloudstackMachineDeploymentFile), machineDeployment); err != nil {714 t.Errorf("unmarshal failed: %v", err)715 }716 fetcher.EXPECT().MachineDeployment(ctx, gomock.Any(), gomock.Any()).Return(machineDeployment, nil)717 fetcher.EXPECT().Fetch(ctx, gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).AnyTimes().Return(nil, errors.NewNotFound(schema.GroupResource{Group: "testgroup", Resource: "testresource"}, ""))718 resourceUpdater.EXPECT().ForceApplyTemplate(ctx, gomock.Any(), gomock.Any()).Do(func(ctx context.Context, template *unstructured.Unstructured, dryRun bool) {719 assert.Equal(t, false, dryRun, "Expected dryRun didn't match")720 println(template.GetName(), " : ", template.GetKind())721 switch template.GetKind() {722 case "MachineDeployment":723 expectedMCDeployment := &unstructured.Unstructured{}724 if err := yaml.Unmarshal([]byte(expectedCloudStackMachineDeploymentOnlyReplica), expectedMCDeployment); err != nil {725 t.Errorf("unmarshal failed: %v", err)726 }727 assert.Equal(t, expectedMCDeployment, template, "values", expectedMCDeployment, template)728 }729 }).AnyTimes().Return(nil)730 },731 },732 {733 name: "worker node reconcile (Cloudstack provider) - worker node taints have changed",734 args: args{735 namespace: "namespaceA",736 name: "nameA",737 objectKey: types.NamespacedName{738 Name: "nameA",739 Namespace: "namespaceA",740 },741 },742 want: controllerruntime.Result{},743 prepare: func(ctx context.Context, fetcher *mocks.MockResourceFetcher, resourceUpdater *mocks.MockResourceUpdater, name string, namespace string) {744 cluster := &anywherev1.Cluster{}745 cluster.SetName(name)746 cluster.SetNamespace(namespace)747 fetcher.EXPECT().FetchCluster(gomock.Any(), gomock.Any()).Return(cluster, nil)748 spec := test.NewFullClusterSpec(t, "testdata/eksa-cluster-cloudstack.yaml")749 cluster.Spec = spec.Cluster.Spec750 fetcher.EXPECT().FetchAppliedSpec(ctx, gomock.Any()).Return(spec, nil)751 fetcher.EXPECT().FetchObject(gomock.Any(), gomock.Any(), gomock.Any()).Do(func(ctx context.Context, objectKey types.NamespacedName, obj client.Object) {752 clusterSpec := &anywherev1.CloudStackDatacenterConfig{}753 if err := yaml.Unmarshal([]byte(cloudstackDatacenterConfigSpecPath), clusterSpec); err != nil {754 t.Errorf("unmarshal failed: %v", err)755 }756 cluster := obj.(*anywherev1.CloudStackDatacenterConfig)757 cluster.SetName(objectKey.Name)758 cluster.SetNamespace(objectKey.Namespace)759 cluster.Spec = clusterSpec.Spec760 assert.Equal(t, objectKey.Name, "test-cluster", "expected Name to be test-cluster")761 }).Return(nil)762 fetcher.EXPECT().FetchObject(gomock.Any(), gomock.Any(), gomock.Any()).Do(func(ctx context.Context, objectKey types.NamespacedName, obj client.Object) {763 clusterSpec := &anywherev1.CloudStackMachineConfig{}764 if err := yaml.Unmarshal([]byte(cloudstackMachineConfigSpecPath), clusterSpec); err != nil {765 t.Errorf("unmarshal failed: %v", err)766 }767 cluster := obj.(*anywherev1.CloudStackMachineConfig)768 cluster.SetName(objectKey.Name)769 cluster.SetNamespace(objectKey.Namespace)770 cluster.Spec = clusterSpec.Spec771 assert.Equal(t, objectKey.Name, "test-cluster", "expected Name to be test-cluster")772 }).Return(nil)773 fetcher.EXPECT().FetchObject(gomock.Any(), gomock.Any(), gomock.Any()).Do(func(ctx context.Context, objectKey types.NamespacedName, obj client.Object) {774 clusterSpec := &anywherev1.CloudStackMachineConfig{}775 if err := yaml.Unmarshal([]byte(cloudstackMachineConfigSpecPath), clusterSpec); err != nil {776 t.Errorf("unmarshal failed: %v", err)777 }778 cluster := obj.(*anywherev1.CloudStackMachineConfig)779 cluster.SetName(objectKey.Name)780 cluster.SetNamespace(objectKey.Namespace)781 cluster.Spec = clusterSpec.Spec782 assert.Equal(t, objectKey.Name, "test-cluster", "expected Name to be test-cluster")783 }).Return(nil)784 kubeAdmControlPlane := &controlplanev1.KubeadmControlPlane{}785 if err := yaml.Unmarshal([]byte(cloudstackKubeadmcontrolplaneFile), kubeAdmControlPlane); err != nil {786 t.Errorf("unmarshal failed: %v", err)787 }788 existingWorkerNodeGroupConfiguration := &anywherev1.WorkerNodeGroupConfiguration{789 Name: "md-0",790 Count: 3,791 MachineGroupRef: nil,792 Taints: []corev1.Taint{793 {794 Key: "key1",795 Value: "val1",796 Effect: "PreferNoSchedule",797 },798 },799 }800 oldCloudstackProviderFeatureValue := os.Getenv(features.CloudStackProviderEnvVar)801 os.Unsetenv(features.CloudStackProviderEnvVar)802 defer os.Setenv(features.CloudStackProviderEnvVar, oldCloudstackProviderFeatureValue)803 fetcher.EXPECT().ExistingCloudStackDatacenterConfig(ctx, gomock.Any(), gomock.Any()).Return(&anywherev1.CloudStackDatacenterConfig{}, nil)804 fetcher.EXPECT().ExistingCloudStackControlPlaneMachineConfig(ctx, gomock.Any()).Return(&anywherev1.CloudStackMachineConfig{}, nil)805 fetcher.EXPECT().ExistingCloudStackWorkerMachineConfig(ctx, gomock.Any(), gomock.Any()).Return(&anywherev1.CloudStackMachineConfig{}, nil)806 fetcher.EXPECT().ExistingWorkerNodeGroupConfig(ctx, gomock.Any(), gomock.Any()).Return(existingWorkerNodeGroupConfiguration, nil)807 fetcher.EXPECT().Fetch(ctx, gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).AnyTimes().Return(nil, errors.NewNotFound(schema.GroupResource{Group: "testgroup", Resource: "testresource"}, ""))808 resourceUpdater.EXPECT().ForceApplyTemplate(ctx, gomock.Any(), gomock.Any()).Do(func(ctx context.Context, template *unstructured.Unstructured, dryRun bool) {809 assert.Equal(t, false, dryRun, "Expected dryRun didn't match")810 switch template.GetKind() {811 case "KubeadmConfigTemplate":812 existingKubeadmConfigTemplate := &unstructured.Unstructured{}813 if err := yaml.Unmarshal([]byte(cloudstackKubeadmconfigTemplateSpecPath), existingKubeadmConfigTemplate); err != nil {814 t.Errorf("unmarshal failed: %v", err)815 }816 assert.Equal(t, existingKubeadmConfigTemplate, template, "values", existingKubeadmConfigTemplate, template)817 case "MachineDeployment":818 expectedMCDeployment := &unstructured.Unstructured{}819 if err := yaml.Unmarshal([]byte(expectedCloudStackMachineDeploymentTemplateChanged), expectedMCDeployment); err != nil {820 t.Errorf("unmarshal failed: %v", err)821 }822 assert.Equal(t, expectedMCDeployment, template, "values", expectedMCDeployment, template)823 }824 }).AnyTimes().Return(nil)825 },826 },827 {828 name: "worker node reconcile (Cloudstack provider) - worker node labels have changed",829 args: args{830 namespace: "namespaceA",831 name: "nameA",832 objectKey: types.NamespacedName{833 Name: "nameA",834 Namespace: "namespaceA",835 },836 },837 want: controllerruntime.Result{},838 prepare: func(ctx context.Context, fetcher *mocks.MockResourceFetcher, resourceUpdater *mocks.MockResourceUpdater, name string, namespace string) {839 cluster := &anywherev1.Cluster{}840 cluster.SetName(name)841 cluster.SetNamespace(namespace)842 fetcher.EXPECT().FetchCluster(gomock.Any(), gomock.Any()).Return(cluster, nil)843 spec := test.NewFullClusterSpec(t, "testdata/eksa-cluster-cloudstack.yaml")844 cluster.Spec = spec.Cluster.Spec845 fetcher.EXPECT().FetchAppliedSpec(ctx, gomock.Any()).Return(spec, nil)846 fetcher.EXPECT().FetchObject(gomock.Any(), gomock.Any(), gomock.Any()).Do(func(ctx context.Context, objectKey types.NamespacedName, obj client.Object) {847 clusterSpec := &anywherev1.CloudStackDatacenterConfig{}848 if err := yaml.Unmarshal([]byte(cloudstackDatacenterConfigSpecPath), clusterSpec); err != nil {849 t.Errorf("unmarshal failed: %v", err)850 }851 cluster := obj.(*anywherev1.CloudStackDatacenterConfig)852 cluster.SetName(objectKey.Name)853 cluster.SetNamespace(objectKey.Namespace)854 cluster.Spec = clusterSpec.Spec855 assert.Equal(t, objectKey.Name, "test-cluster", "expected Name to be test-cluster")856 }).Return(nil)857 fetcher.EXPECT().FetchObject(gomock.Any(), gomock.Any(), gomock.Any()).Do(func(ctx context.Context, objectKey types.NamespacedName, obj client.Object) {858 clusterSpec := &anywherev1.CloudStackMachineConfig{}859 if err := yaml.Unmarshal([]byte(cloudstackMachineConfigSpecPath), clusterSpec); err != nil {860 t.Errorf("unmarshal failed: %v", err)861 }862 cluster := obj.(*anywherev1.CloudStackMachineConfig)863 cluster.SetName(objectKey.Name)864 cluster.SetNamespace(objectKey.Namespace)865 cluster.Spec = clusterSpec.Spec866 assert.Equal(t, objectKey.Name, "test-cluster", "expected Name to be test-cluster")867 }).Return(nil)868 fetcher.EXPECT().FetchObject(gomock.Any(), gomock.Any(), gomock.Any()).Do(func(ctx context.Context, objectKey types.NamespacedName, obj client.Object) {869 clusterSpec := &anywherev1.CloudStackMachineConfig{}870 if err := yaml.Unmarshal([]byte(cloudstackMachineConfigSpecPath), clusterSpec); err != nil {871 t.Errorf("unmarshal failed: %v", err)872 }873 cluster := obj.(*anywherev1.CloudStackMachineConfig)874 cluster.SetName(objectKey.Name)875 cluster.SetNamespace(objectKey.Namespace)876 cluster.Spec = clusterSpec.Spec877 assert.Equal(t, objectKey.Name, "test-cluster", "expected Name to be test-cluster")878 }).Return(nil)879 kubeAdmControlPlane := &controlplanev1.KubeadmControlPlane{}880 if err := yaml.Unmarshal([]byte(cloudstackKubeadmcontrolplaneFile), kubeAdmControlPlane); err != nil {881 t.Errorf("unmarshal failed: %v", err)882 }883 existingWorkerNodeGroupConfiguration := &anywherev1.WorkerNodeGroupConfiguration{884 Name: "md-0",885 Count: 3,886 MachineGroupRef: nil,887 Labels: map[string]string{888 "Key1": "Val1",889 "Key2": "Val2",890 },891 }892 fetcher.EXPECT().ExistingCloudStackDatacenterConfig(ctx, gomock.Any(), gomock.Any()).Return(&anywherev1.CloudStackDatacenterConfig{}, nil)893 fetcher.EXPECT().ExistingCloudStackControlPlaneMachineConfig(ctx, gomock.Any()).Return(&anywherev1.CloudStackMachineConfig{}, nil)894 fetcher.EXPECT().ExistingCloudStackWorkerMachineConfig(ctx, gomock.Any(), gomock.Any()).Return(&anywherev1.CloudStackMachineConfig{}, nil)895 fetcher.EXPECT().ExistingWorkerNodeGroupConfig(ctx, gomock.Any(), gomock.Any()).Return(existingWorkerNodeGroupConfiguration, nil)896 fetcher.EXPECT().Fetch(ctx, gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).AnyTimes().Return(nil, errors.NewNotFound(schema.GroupResource{Group: "testgroup", Resource: "testresource"}, ""))897 resourceUpdater.EXPECT().ForceApplyTemplate(ctx, gomock.Any(), gomock.Any()).Do(func(ctx context.Context, template *unstructured.Unstructured, dryRun bool) {898 assert.Equal(t, false, dryRun, "Expected dryRun didn't match")899 switch template.GetKind() {900 case "KubeadmConfigTemplate":901 existingKubeadmConfigTemplate := &unstructured.Unstructured{}902 if err := yaml.Unmarshal([]byte(cloudstackKubeadmconfigTemplateSpecPath), existingKubeadmConfigTemplate); err != nil {903 t.Errorf("unmarshal failed: %v", err)904 }905 assert.Equal(t, existingKubeadmConfigTemplate, template, "values", existingKubeadmConfigTemplate, template)906 case "MachineDeployment":907 expectedMCDeployment := &unstructured.Unstructured{}908 if err := yaml.Unmarshal([]byte(expectedCloudStackMachineDeploymentTemplateChanged), expectedMCDeployment); err != nil {909 t.Errorf("unmarshal failed: %v", err)910 }911 assert.Equal(t, expectedMCDeployment, template, "values", expectedMCDeployment, template)912 }913 }).AnyTimes().Return(nil)914 },915 },916 {917 name: "cp & etcd reconcile (Cloudstack provider) - cp & etcd templates have changed, but don't reconcile",918 args: args{919 namespace: "namespaceA",920 name: "test-cluster",921 objectKey: types.NamespacedName{922 Name: "test-cluster",923 Namespace: "namespaceA",924 },925 },926 want: controllerruntime.Result{},927 prepare: func(ctx context.Context, fetcher *mocks.MockResourceFetcher, resourceUpdater *mocks.MockResourceUpdater, name string, namespace string) {928 cluster := &anywherev1.Cluster{}929 cluster.SetName(name)930 cluster.SetNamespace(namespace)931 fetcher.EXPECT().FetchCluster(gomock.Any(), gomock.Any()).Return(cluster, nil)932 spec := test.NewFullClusterSpec(t, "testdata/eksa-cluster-cloudstack-etcd.yaml")933 cluster.Spec = spec.Cluster.Spec934 fetcher.EXPECT().FetchAppliedSpec(ctx, gomock.Any()).Return(spec, nil)935 fetcher.EXPECT().FetchObject(gomock.Any(), gomock.Any(), gomock.Any()).Do(func(ctx context.Context, objectKey types.NamespacedName, obj client.Object) {936 clusterSpec := &anywherev1.CloudStackDatacenterConfig{}937 if err := yaml.Unmarshal([]byte(cloudstackDatacenterConfigSpecPath), clusterSpec); err != nil {938 t.Errorf("unmarshal failed: %v", err)939 }940 cluster := obj.(*anywherev1.CloudStackDatacenterConfig)941 cluster.SetName(objectKey.Name)942 cluster.SetNamespace(objectKey.Namespace)943 cluster.Spec = clusterSpec.Spec944 assert.Equal(t, objectKey.Name, "test-cluster", "expected Name to be test-cluster")945 }).Return(nil)946 fetcher.EXPECT().FetchObject(gomock.Any(), gomock.Any(), gomock.Any()).Do(func(ctx context.Context, objectKey types.NamespacedName, obj client.Object) {947 clusterSpec := &anywherev1.CloudStackMachineConfig{}948 if err := yaml.Unmarshal([]byte(cloudstackMachineConfigSpecPath), clusterSpec); err != nil {949 t.Errorf("unmarshal failed: %v", err)950 }951 cluster := obj.(*anywherev1.CloudStackMachineConfig)952 cluster.SetName(objectKey.Name)953 cluster.SetNamespace(objectKey.Namespace)954 cluster.Spec = clusterSpec.Spec955 assert.Equal(t, objectKey.Name, "test-cluster", "expected Name to be test-cluster")956 }).Return(nil).Times(3)957 kubeAdmControlPlane := &controlplanev1.KubeadmControlPlane{}958 if err := yaml.Unmarshal([]byte(cloudstackKubeadmcontrolplaneFile), kubeAdmControlPlane); err != nil {959 t.Errorf("unmarshal failed: %v", err)960 }961 etcdadmCluster := &etcdv1.EtcdadmCluster{}962 if err := yaml.Unmarshal([]byte(cloudstackEtcdadmclusterFile), etcdadmCluster); err != nil {963 t.Errorf("unmarshal failed: %v", err)964 }965 existingWorkerNodeGroupConfiguration := &anywherev1.WorkerNodeGroupConfiguration{966 Name: "md-0",967 Count: 3,968 MachineGroupRef: nil,969 }970 existingCSDatacenterConfig := &anywherev1.CloudStackDatacenterConfig{971 Spec: anywherev1.CloudStackDatacenterConfigSpec{972 Domain: "root",973 Account: "admin",974 Zones: []anywherev1.CloudStackZone{975 {976 Name: "zone1",977 Network: anywherev1.CloudStackResourceIdentifier{978 Name: "net1",979 },980 },981 {982 Name: "zone2",983 Network: anywherev1.CloudStackResourceIdentifier{984 Name: "net2",985 },986 },987 },988 },989 }990 oldCloudstackProviderFeatureValue := os.Getenv(features.CloudStackProviderEnvVar)991 os.Unsetenv(features.CloudStackProviderEnvVar)992 defer os.Setenv(features.CloudStackProviderEnvVar, oldCloudstackProviderFeatureValue)993 fetcher.EXPECT().Etcd(ctx, gomock.Any()).Return(etcdadmCluster, nil)994 fetcher.EXPECT().ExistingCloudStackDatacenterConfig(ctx, gomock.Any(), gomock.Any()).Return(existingCSDatacenterConfig, nil)995 fetcher.EXPECT().ExistingCloudStackControlPlaneMachineConfig(ctx, gomock.Any()).Return(&anywherev1.CloudStackMachineConfig{}, nil)996 fetcher.EXPECT().ExistingCloudStackEtcdMachineConfig(ctx, gomock.Any()).Return(&anywherev1.CloudStackMachineConfig{}, nil)997 fetcher.EXPECT().ExistingCloudStackWorkerMachineConfig(ctx, gomock.Any(), gomock.Any()).Return(&anywherev1.CloudStackMachineConfig{}, nil)998 fetcher.EXPECT().ExistingWorkerNodeGroupConfig(ctx, gomock.Any(), gomock.Any()).Return(existingWorkerNodeGroupConfiguration, nil)999 machineDeployment := &clusterv1.MachineDeployment{}1000 if err := yaml.Unmarshal([]byte(cloudstackMachineDeploymentFile), machineDeployment); err != nil {1001 t.Errorf("unmarshal failed: %v", err)1002 }1003 fetcher.EXPECT().MachineDeployment(ctx, gomock.Any(), gomock.Any()).Return(machineDeployment, nil)1004 fetcher.EXPECT().Fetch(ctx, gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).AnyTimes().Return(nil, errors.NewNotFound(schema.GroupResource{Group: "testgroup", Resource: "testresource"}, ""))1005 resourceUpdater.EXPECT().ApplyPatch(ctx, gomock.Any(), false).Return(nil)1006 resourceUpdater.EXPECT().ForceApplyTemplate(ctx, gomock.Any(), gomock.Any()).Do(func(ctx context.Context, template *unstructured.Unstructured, dryRun bool) {1007 assert.Equal(t, false, dryRun, "Expected dryRun didn't match")1008 switch template.GetKind() {1009 case "CloudStackMachineTemplate":1010 if strings.Contains(template.GetName(), "worker-node") {1011 expectedMachineTemplate := &unstructured.Unstructured{}1012 if err := yaml.Unmarshal([]byte(cloudstackMachineTemplateFile), expectedMachineTemplate); err != nil {1013 t.Errorf("unmarshal failed: %v", err)1014 }1015 assert.Equal(t, expectedMachineTemplate, template, "values", expectedMachineTemplate, template)1016 }1017 if strings.Contains(template.GetName(), "etcd-template") {1018 expectedMachineTemplate := &unstructured.Unstructured{}1019 if err := yaml.Unmarshal([]byte(cloudstackEtcdMachineTemplateFile), expectedMachineTemplate); err != nil {1020 t.Errorf("unmarshal failed: %v", err)1021 }1022 assert.Equal(t, expectedMachineTemplate, template, "values", expectedMachineTemplate, template)1023 }1024 if strings.Contains(template.GetName(), "control-plane-template") {1025 expectedMachineTemplate := &unstructured.Unstructured{}1026 if err := yaml.Unmarshal([]byte(cloudstackCPMachineTemplateFile), expectedMachineTemplate); err != nil {1027 t.Errorf("unmarshal failed: %v", err)1028 }1029 assert.Equal(t, expectedMachineTemplate, template, "values", expectedMachineTemplate, template)1030 }1031 case "MachineDeployment":1032 expectedMCDeployment := &unstructured.Unstructured{}1033 if err := yaml.Unmarshal([]byte(expectedCloudStackMachineDeploymentFile), expectedMCDeployment); err != nil {1034 t.Errorf("unmarshal failed: %v", err)1035 }1036 assert.Equal(t, expectedMCDeployment, template, "values", expectedMCDeployment, template)1037 }1038 }).AnyTimes().Return(nil)1039 },1040 },1041 }1042 oldCloudstackProviderFeatureValue := os.Getenv(features.CloudStackProviderEnvVar)1043 os.Setenv(features.CloudStackProviderEnvVar, "true")1044 defer os.Setenv(features.CloudStackProviderEnvVar, oldCloudstackProviderFeatureValue)1045 for _, tt := range tests {1046 t.Run(tt.name, func(t *testing.T) {1047 ctx := context.Background()1048 mockCtrl := gomock.NewController(t)1049 fetcher := mocks.NewMockResourceFetcher(mockCtrl)1050 resourceUpdater := mocks.NewMockResourceUpdater(mockCtrl)1051 tt.prepare(ctx, fetcher, resourceUpdater, tt.args.name, tt.args.namespace)1052 cor := resource.NewClusterReconciler(fetcher, resourceUpdater, test.FakeNow, logr.Discard())1053 if err := cor.Reconcile(ctx, tt.args.objectKey, false); (err != nil) != tt.wantErr {1054 t.Errorf("Reconcile() error = %v, wantErr %v", err, tt.wantErr)1055 }1056 })1057 }1058}1059func TestClusterReconcilerReconcileNutanix(t *testing.T) {1060 err := os.Setenv(features.NutanixProviderEnvVar, "true")1061 require.NoError(t, err)1062 assert.True(t, features.NutanixProvider().IsActive())1063 type args struct {1064 objectKey types.NamespacedName1065 name string1066 namespace string1067 }1068 tests := []struct {1069 name string1070 args args1071 want controllerruntime.Result1072 wantErr bool1073 prepare func(context.Context, *mocks.MockResourceFetcher, string, string)1074 }{1075 {1076 name: "node reconcile - no error",1077 args: args{1078 namespace: "namespaceA",1079 name: "nameA",1080 objectKey: types.NamespacedName{1081 Name: "nameA",1082 Namespace: "namespaceA",1083 },1084 },1085 want: controllerruntime.Result{},1086 prepare: func(ctx context.Context, fetcher *mocks.MockResourceFetcher, name string, namespace string) {1087 cluster := &anywherev1.Cluster{}1088 cluster.SetName(name)1089 cluster.SetNamespace(namespace)1090 fetcher.EXPECT().FetchCluster(gomock.Any(), gomock.Any()).Return(cluster, nil)1091 spec := test.NewFullClusterSpec(t, "testdata/nutanix/eksa-cluster.yaml")1092 cluster.Spec = spec.Cluster.Spec1093 fetcher.EXPECT().FetchAppliedSpec(ctx, gomock.Any()).Return(spec, nil)1094 fetcher.EXPECT().FetchObject(gomock.Any(), gomock.Any(), gomock.Any()).Do(func(ctx context.Context, objectKey types.NamespacedName, obj client.Object) {1095 clusterSpec := &anywherev1.NutanixDatacenterConfig{}1096 if err := yaml.Unmarshal([]byte(nutanixDatacenterConfigSpec), clusterSpec); err != nil {1097 t.Errorf("unmarshal failed: %v", err)1098 }1099 cluster := obj.(*anywherev1.NutanixDatacenterConfig)1100 cluster.SetName(objectKey.Name)1101 cluster.SetNamespace(objectKey.Namespace)1102 cluster.Spec = clusterSpec.Spec1103 assert.Equal(t, objectKey.Name, "eksa-unit-test", "expected Name to be test-cluster")1104 }).Return(nil)1105 fetcher.EXPECT().FetchObject(gomock.Any(), gomock.Any(), gomock.Any()).Do(func(ctx context.Context, objectKey types.NamespacedName, obj client.Object) {1106 clusterSpec := &anywherev1.NutanixMachineConfig{}1107 if err := yaml.Unmarshal([]byte(nutanixCPMachineConfigSpec), clusterSpec); err != nil {1108 t.Errorf("unmarshal failed: %v", err)1109 }1110 cluster := obj.(*anywherev1.NutanixMachineConfig)1111 cluster.SetName(objectKey.Name)1112 cluster.SetNamespace(objectKey.Namespace)1113 cluster.Spec = clusterSpec.Spec1114 assert.Equal(t, objectKey.Name, "eksa-unit-test", "expected Name to be test-cluster")1115 }).Return(nil)1116 fetcher.EXPECT().FetchObject(gomock.Any(), gomock.Any(), gomock.Any()).Do(func(ctx context.Context, objectKey types.NamespacedName, obj client.Object) {1117 clusterSpec := &anywherev1.NutanixMachineConfig{}1118 if err := yaml.Unmarshal([]byte(nutanixWorkerMachineConfigSpec), clusterSpec); err != nil {1119 t.Errorf("unmarshal failed: %v", err)1120 }1121 cluster := obj.(*anywherev1.NutanixMachineConfig)1122 cluster.SetName(objectKey.Name)1123 cluster.SetNamespace(objectKey.Namespace)1124 cluster.Spec = clusterSpec.Spec1125 assert.Equal(t, objectKey.Name, "eksa-unit-test", "expected Name to be test-cluster")1126 }).Return(nil)1127 fetcher.EXPECT().FetchObject(gomock.Any(), gomock.Any(), gomock.Any()).Do(func(ctx context.Context, objectKey types.NamespacedName, obj client.Object) {1128 clusterSpec := &anywherev1.NutanixMachineConfig{}1129 if err := yaml.Unmarshal([]byte(nutanixEtcdMachineConfigSpec), clusterSpec); err != nil {1130 t.Errorf("unmarshal failed: %v", err)1131 }1132 cluster := obj.(*anywherev1.NutanixMachineConfig)1133 cluster.SetName(objectKey.Name)1134 cluster.SetNamespace(objectKey.Namespace)1135 cluster.Spec = clusterSpec.Spec1136 assert.Equal(t, objectKey.Name, "eksa-unit-test", "expected Name to be test-cluster")1137 }).Return(nil)1138 },1139 },1140 {1141 name: "node reconcile - failure to fetch cluster",1142 args: args{1143 namespace: "namespaceA",1144 name: "nameA",1145 objectKey: types.NamespacedName{1146 Name: "nameA",1147 Namespace: "namespaceA",1148 },1149 },1150 wantErr: true,1151 prepare: func(ctx context.Context, fetcher *mocks.MockResourceFetcher, name string, namespace string) {1152 fetcher.EXPECT().FetchCluster(gomock.Any(), gomock.Any()).Return(nil, fmt.Errorf("error"))1153 },1154 },1155 {1156 name: "node reconcile - failure to fetch applied spec",1157 args: args{1158 namespace: "namespaceA",1159 name: "nameA",1160 objectKey: types.NamespacedName{1161 Name: "nameA",1162 Namespace: "namespaceA",1163 },1164 },1165 wantErr: true,1166 prepare: func(ctx context.Context, fetcher *mocks.MockResourceFetcher, name string, namespace string) {1167 cluster := &anywherev1.Cluster{}1168 cluster.SetName(name)1169 cluster.SetNamespace(namespace)1170 fetcher.EXPECT().FetchCluster(gomock.Any(), gomock.Any()).Return(cluster, nil)1171 spec := test.NewFullClusterSpec(t, "testdata/nutanix/eksa-cluster.yaml")1172 cluster.Spec = spec.Cluster.Spec1173 fetcher.EXPECT().FetchAppliedSpec(ctx, gomock.Any()).Return(nil, fmt.Errorf("error"))1174 },1175 },1176 {1177 name: "node reconcile - failure to fetch datacenter config",1178 args: args{1179 namespace: "namespaceA",1180 name: "nameA",1181 objectKey: types.NamespacedName{1182 Name: "nameA",1183 Namespace: "namespaceA",1184 },1185 },1186 wantErr: true,1187 prepare: func(ctx context.Context, fetcher *mocks.MockResourceFetcher, name string, namespace string) {1188 cluster := &anywherev1.Cluster{}1189 cluster.SetName(name)1190 cluster.SetNamespace(namespace)1191 fetcher.EXPECT().FetchCluster(gomock.Any(), gomock.Any()).Return(cluster, nil)1192 spec := test.NewFullClusterSpec(t, "testdata/nutanix/eksa-cluster.yaml")1193 cluster.Spec = spec.Cluster.Spec1194 fetcher.EXPECT().FetchAppliedSpec(ctx, gomock.Any()).Return(spec, nil)1195 fetcher.EXPECT().FetchObject(gomock.Any(), gomock.Any(), gomock.Any()).Do(func(ctx context.Context, objectKey types.NamespacedName, obj client.Object) {}).Return(fmt.Errorf("error"))1196 },1197 },1198 {1199 name: "node reconcile - failure to fetch control plane node machine config",1200 args: args{1201 namespace: "namespaceA",1202 name: "nameA",1203 objectKey: types.NamespacedName{1204 Name: "nameA",1205 Namespace: "namespaceA",1206 },1207 },1208 wantErr: true,1209 prepare: func(ctx context.Context, fetcher *mocks.MockResourceFetcher, name string, namespace string) {1210 cluster := &anywherev1.Cluster{}1211 cluster.SetName(name)1212 cluster.SetNamespace(namespace)1213 fetcher.EXPECT().FetchCluster(gomock.Any(), gomock.Any()).Return(cluster, nil)1214 spec := test.NewFullClusterSpec(t, "testdata/nutanix/eksa-cluster.yaml")1215 cluster.Spec = spec.Cluster.Spec1216 fetcher.EXPECT().FetchAppliedSpec(ctx, gomock.Any()).Return(spec, nil)1217 fetcher.EXPECT().FetchObject(gomock.Any(), gomock.Any(), gomock.Any()).Do(func(ctx context.Context, objectKey types.NamespacedName, obj client.Object) {1218 clusterSpec := &anywherev1.NutanixDatacenterConfig{}1219 if err := yaml.Unmarshal([]byte(nutanixDatacenterConfigSpec), clusterSpec); err != nil {1220 t.Errorf("unmarshal failed: %v", err)1221 }1222 cluster := obj.(*anywherev1.NutanixDatacenterConfig)1223 cluster.SetName(objectKey.Name)1224 cluster.SetNamespace(objectKey.Namespace)1225 cluster.Spec = clusterSpec.Spec1226 assert.Equal(t, objectKey.Name, "eksa-unit-test", "expected Name to be test-cluster")1227 }).Return(nil)1228 fetcher.EXPECT().FetchObject(gomock.Any(), gomock.Any(), gomock.Any()).Do(func(ctx context.Context, objectKey types.NamespacedName, obj client.Object) {}).Return(fmt.Errorf("error"))1229 },1230 },1231 {1232 name: "node reconcile - failure to fetch worker node machine config",1233 args: args{1234 namespace: "namespaceA",1235 name: "nameA",1236 objectKey: types.NamespacedName{1237 Name: "nameA",1238 Namespace: "namespaceA",1239 },1240 },1241 wantErr: true,1242 prepare: func(ctx context.Context, fetcher *mocks.MockResourceFetcher, name string, namespace string) {1243 cluster := &anywherev1.Cluster{}1244 cluster.SetName(name)1245 cluster.SetNamespace(namespace)1246 fetcher.EXPECT().FetchCluster(gomock.Any(), gomock.Any()).Return(cluster, nil)1247 spec := test.NewFullClusterSpec(t, "testdata/nutanix/eksa-cluster.yaml")1248 cluster.Spec = spec.Cluster.Spec1249 fetcher.EXPECT().FetchAppliedSpec(ctx, gomock.Any()).Return(spec, nil)1250 fetcher.EXPECT().FetchObject(gomock.Any(), gomock.Any(), gomock.Any()).Do(func(ctx context.Context, objectKey types.NamespacedName, obj client.Object) {1251 clusterSpec := &anywherev1.NutanixDatacenterConfig{}1252 if err := yaml.Unmarshal([]byte(nutanixDatacenterConfigSpec), clusterSpec); err != nil {1253 t.Errorf("unmarshal failed: %v", err)1254 }1255 cluster := obj.(*anywherev1.NutanixDatacenterConfig)1256 cluster.SetName(objectKey.Name)1257 cluster.SetNamespace(objectKey.Namespace)1258 cluster.Spec = clusterSpec.Spec1259 assert.Equal(t, objectKey.Name, "eksa-unit-test", "expected Name to be test-cluster")1260 }).Return(nil)1261 fetcher.EXPECT().FetchObject(gomock.Any(), gomock.Any(), gomock.Any()).Do(func(ctx context.Context, objectKey types.NamespacedName, obj client.Object) {1262 clusterSpec := &anywherev1.NutanixMachineConfig{}1263 if err := yaml.Unmarshal([]byte(nutanixCPMachineConfigSpec), clusterSpec); err != nil {1264 t.Errorf("unmarshal failed: %v", err)1265 }1266 cluster := obj.(*anywherev1.NutanixMachineConfig)1267 cluster.SetName(objectKey.Name)1268 cluster.SetNamespace(objectKey.Namespace)1269 cluster.Spec = clusterSpec.Spec1270 assert.Equal(t, objectKey.Name, "eksa-unit-test", "expected Name to be test-cluster")1271 }).Return(nil)1272 fetcher.EXPECT().FetchObject(gomock.Any(), gomock.Any(), gomock.Any()).Do(func(ctx context.Context, objectKey types.NamespacedName, obj client.Object) {}).Return(fmt.Errorf("error"))1273 },1274 },1275 {1276 name: "node reconcile - failure to fetch etcd node machine config",1277 args: args{1278 namespace: "namespaceA",1279 name: "nameA",1280 objectKey: types.NamespacedName{1281 Name: "nameA",1282 Namespace: "namespaceA",1283 },1284 },1285 wantErr: true,1286 prepare: func(ctx context.Context, fetcher *mocks.MockResourceFetcher, name string, namespace string) {1287 cluster := &anywherev1.Cluster{}1288 cluster.SetName(name)1289 cluster.SetNamespace(namespace)1290 fetcher.EXPECT().FetchCluster(gomock.Any(), gomock.Any()).Return(cluster, nil)1291 spec := test.NewFullClusterSpec(t, "testdata/nutanix/eksa-cluster.yaml")1292 cluster.Spec = spec.Cluster.Spec1293 fetcher.EXPECT().FetchAppliedSpec(ctx, gomock.Any()).Return(spec, nil)1294 fetcher.EXPECT().FetchObject(gomock.Any(), gomock.Any(), gomock.Any()).Do(func(ctx context.Context, objectKey types.NamespacedName, obj client.Object) {1295 clusterSpec := &anywherev1.NutanixDatacenterConfig{}1296 if err := yaml.Unmarshal([]byte(nutanixDatacenterConfigSpec), clusterSpec); err != nil {1297 t.Errorf("unmarshal failed: %v", err)1298 }1299 cluster := obj.(*anywherev1.NutanixDatacenterConfig)1300 cluster.SetName(objectKey.Name)1301 cluster.SetNamespace(objectKey.Namespace)1302 cluster.Spec = clusterSpec.Spec1303 assert.Equal(t, objectKey.Name, "eksa-unit-test", "expected Name to be test-cluster")1304 }).Return(nil)1305 fetcher.EXPECT().FetchObject(gomock.Any(), gomock.Any(), gomock.Any()).Do(func(ctx context.Context, objectKey types.NamespacedName, obj client.Object) {1306 clusterSpec := &anywherev1.NutanixMachineConfig{}1307 if err := yaml.Unmarshal([]byte(nutanixCPMachineConfigSpec), clusterSpec); err != nil {1308 t.Errorf("unmarshal failed: %v", err)1309 }1310 cluster := obj.(*anywherev1.NutanixMachineConfig)1311 cluster.SetName(objectKey.Name)1312 cluster.SetNamespace(objectKey.Namespace)1313 cluster.Spec = clusterSpec.Spec1314 assert.Equal(t, objectKey.Name, "eksa-unit-test", "expected Name to be test-cluster")1315 }).Return(nil)1316 fetcher.EXPECT().FetchObject(gomock.Any(), gomock.Any(), gomock.Any()).Do(func(ctx context.Context, objectKey types.NamespacedName, obj client.Object) {1317 clusterSpec := &anywherev1.NutanixMachineConfig{}1318 if err := yaml.Unmarshal([]byte(nutanixWorkerMachineConfigSpec), clusterSpec); err != nil {1319 t.Errorf("unmarshal failed: %v", err)1320 }1321 cluster := obj.(*anywherev1.NutanixMachineConfig)1322 cluster.SetName(objectKey.Name)1323 cluster.SetNamespace(objectKey.Namespace)1324 cluster.Spec = clusterSpec.Spec1325 assert.Equal(t, objectKey.Name, "eksa-unit-test", "expected Name to be test-cluster")1326 }).Return(nil)1327 fetcher.EXPECT().FetchObject(gomock.Any(), gomock.Any(), gomock.Any()).Do(func(ctx context.Context, objectKey types.NamespacedName, obj client.Object) {}).Return(fmt.Errorf("error"))1328 },1329 },1330 }1331 for _, tt := range tests {1332 t.Run(tt.name, func(t *testing.T) {1333 ctx := context.Background()1334 mockCtrl := gomock.NewController(t)1335 fetcher := mocks.NewMockResourceFetcher(mockCtrl)1336 resourceUpdater := mocks.NewMockResourceUpdater(mockCtrl)1337 tt.prepare(ctx, fetcher, tt.args.name, tt.args.namespace)...
spec_namespace.go
Source:spec_namespace.go
1package mgr2import (3 "context"4 "fmt"5 "os"6 "path/filepath"7 "strconv"8 "strings"9 specs "github.com/opencontainers/runtime-spec/specs-go"10)11// isEmpty indicates whether namespace mode is empty.12func isEmpty(mode string) bool {13 return mode == ""14}15// isNone indicates whether container's namespace mode is set to "none".16func isNone(mode string) bool {17 return mode == "none"18}19// isHost indicates whether the container shares the host's corresponding namespace.20func isHost(mode string) bool {21 return mode == "host"22}23// isShareable indicates whether the containers namespace can be shared with another container.24func isShareable(mode string) bool {25 return mode == "shareable"26}27// isContainer indicates whether the container uses another container's corresponding namespace.28func isContainer(mode string) bool {29 parts := strings.SplitN(mode, ":", 2)30 return len(parts) > 1 && parts[0] == "container"31}32// isPrivate indicates whether the container uses its own namespace.33func isPrivate(ns specs.LinuxNamespaceType, mode string) bool {34 switch ns {35 case specs.IPCNamespace:36 return mode == "private"37 case specs.NetworkNamespace, specs.PIDNamespace:38 return !(isHost(mode) || isContainer(mode))39 case specs.UserNamespace, specs.UTSNamespace:40 return !(isHost(mode))41 }42 return false43}44// connectedContainer is the id or name of the container whose namespace this container share with.45func connectedContainer(mode string) string {46 parts := strings.SplitN(mode, ":", 2)47 if len(parts) == 2 {48 return parts[1]49 }50 return ""51}52func getIpcContainer(ctx context.Context, mgr ContainerMgr, id string) (*ContainerMeta, error) {53 // Check whether the container exists.54 c, err := mgr.Get(ctx, id)55 if err != nil {56 return nil, fmt.Errorf("can't join IPC namespace of container %q: %v", id, err)57 }58 // TODO: check whether the container is running and not restarting.59 // TODO: check whether the container's ipc namespace is shareable.60 return c, nil61}62func getPidContainer(ctx context.Context, mgr ContainerMgr, id string) (*ContainerMeta, error) {63 // Check the container exists.64 c, err := mgr.Get(ctx, id)65 if err != nil {66 return nil, fmt.Errorf("can't join PID namespace of %q: %v", id, err)67 }68 // TODO: check whether the container is running and not restarting.69 return c, nil70}71// TODO72func setupUserNamespace(ctx context.Context, meta *ContainerMeta, spec *SpecWrapper) error {73 return nil74}75func setupNetworkNamespace(ctx context.Context, meta *ContainerMeta, spec *SpecWrapper) error {76 s := spec.s77 ns := specs.LinuxNamespace{Type: specs.NetworkNamespace}78 setNamespace(s, ns)79 for _, ns := range s.Linux.Namespaces {80 if ns.Type == "network" && ns.Path == "" && !meta.Config.NetworkDisabled {81 target, err := os.Readlink(filepath.Join("/proc", strconv.Itoa(os.Getpid()), "exe"))82 if err != nil {83 return err84 }85 s.Hooks = &specs.Hooks{86 Prestart: []specs.Hook{{87 Path: target, // FIXME: cross-platform88 Args: []string{"libnetwork-setkey", meta.ID, spec.netMgr.Controller().ID()},89 }},90 }91 }92 }93 return nil94}95func setupIpcNamespace(ctx context.Context, meta *ContainerMeta, spec *SpecWrapper) error {96 ipcMode := meta.HostConfig.IpcMode97 switch {98 case isContainer(ipcMode):99 ns := specs.LinuxNamespace{Type: specs.IPCNamespace}100 c, err := getIpcContainer(ctx, spec.ctrMgr, connectedContainer(ipcMode))101 if err != nil {102 return fmt.Errorf("setup container ipc namespace mode failed: %v", err)103 }104 ns.Path = fmt.Sprintf("/proc/%d/ns/ipc", c.State.Pid)105 setNamespace(spec.s, ns)106 case isHost(ipcMode):107 removeNamespace(spec.s, specs.IPCNamespace)108 default:109 ns := specs.LinuxNamespace{Type: specs.IPCNamespace}110 setNamespace(spec.s, ns)111 }112 return nil113}114func setupPidNamespace(ctx context.Context, meta *ContainerMeta, spec *SpecWrapper) error {115 pidMode := meta.HostConfig.PidMode116 switch {117 case isContainer(pidMode):118 ns := specs.LinuxNamespace{Type: specs.PIDNamespace}119 c, err := getPidContainer(ctx, spec.ctrMgr, connectedContainer(pidMode))120 if err != nil {121 return fmt.Errorf("setup container pid namespace mode failed: %v", err)122 }123 ns.Path = fmt.Sprintf("/proc/%d/ns/pid", c.State.Pid)124 setNamespace(spec.s, ns)125 case isHost(pidMode):126 removeNamespace(spec.s, specs.PIDNamespace)127 default:128 ns := specs.LinuxNamespace{Type: specs.PIDNamespace}129 setNamespace(spec.s, ns)130 }131 return nil132}133func setupUtsNamespace(ctx context.Context, meta *ContainerMeta, spec *SpecWrapper) error {134 utsMode := meta.HostConfig.UTSMode135 switch {136 case isHost(utsMode):137 removeNamespace(spec.s, specs.UTSNamespace)138 default:139 ns := specs.LinuxNamespace{Type: specs.UTSNamespace}140 setNamespace(spec.s, ns)141 // set hostname142 if hostname := meta.Config.Hostname.String(); hostname != "" {143 spec.s.Hostname = hostname144 }145 }146 return nil147}148func setNamespace(s *specs.Spec, ns specs.LinuxNamespace) {149 for i, n := range s.Linux.Namespaces {150 if n.Type == ns.Type {151 s.Linux.Namespaces[i] = ns152 return153 }154 }155 s.Linux.Namespaces = append(s.Linux.Namespaces, ns)156}157func removeNamespace(s *specs.Spec, nsType specs.LinuxNamespaceType) {158 for i, n := range s.Linux.Namespaces {159 if n.Type == nsType {160 s.Linux.Namespaces = append(s.Linux.Namespaces[:i], s.Linux.Namespaces[i+1:]...)161 return162 }163 }164}...
options.go
Source:options.go
...39 if err != nil {40 return err41 }42 if nsFlag != "" {43 o.SetNamespace(nsFlag)44 } else {45 o.SetNamespace(nsConfig)46 }47 return nil48}49// SetNamespace configures the namespace50func (o *Options) SetNamespace(ns string) {51 o.UserNamespace = ns52}53// SetClient configures the client54func (o *Options) SetClient(client client.Client) {55 o.Client = client56}57// SetClientset configures the clientset58func (o *Options) SetClientset(clientset *kubernetes.Clientset) {59 o.Clientset = clientset60}61// GetClientConfig returns the client config62func (o *Options) GetClientConfig() clientcmd.ClientConfig {63 return o.ConfigFlags.ToRawKubeConfigLoader()64}...
SetNamespace
Using AI Code Generation
1import (2func main() {3 config, err := clientcmd.BuildConfigFromFlags("", "/home/username/.kube/config")4 if err != nil {5 log.Fatal(err)6 }7 config.SetNamespace("default")8 clientset, err := kubernetes.NewForConfig(config)9 if err != nil {10 log.Fatal(err)11 }12 pods, err := clientset.CoreV1().Pods("").List(metav1.ListOptions{})13 if err != nil {14 log.Fatal(err)15 }16 for _, pod := range pods.Items {17 fmt.Printf("%s %s18 }19}
SetNamespace
Using AI Code Generation
1import (2func main() {3 config, err := clientcmd.BuildConfigFromFlags("", "/home/username/.kube/config")4 if err != nil {5 panic(err.Error())6 }7 clientset, err := kubernetes.NewForConfig(config)8 if err != nil {9 panic(err.Error())10 }11 fmt.Println("namespace: ", namespace)12 config.SetNamespace("default")13 fmt.Println("namespace: ", namespace)14 namespace = clientset.CoreV1().RESTClient().Get().AbsPath("namespaces").Do().Namespace()15 fmt.Println("namespace: ", namespace)16 clientset.CoreV1().RESTClient().Get().AbsPath("namespaces").Do().SetNamespace("default")17 namespace = clientset.CoreV1().RESTClient().Get().AbsPath("namespaces").Do().Namespace()18 fmt.Println("namespace: ", namespace)19}
SetNamespace
Using AI Code Generation
1import (2func main() {3 viper.SetConfigName("config")4 viper.AddConfigPath(".")5 err := viper.ReadInConfig()6 if err != nil {7 fmt.Println("Error while reading config file", err)8 }9 fmt.Println("config file name:", viper.ConfigFileUsed())10 viper.SetNamespace("database")11 fmt.Println("Namespace: ", viper.Get("name"))12}
SetNamespace
Using AI Code Generation
1func main() {2 config, err := clientcmd.BuildConfigFromFlags("", "/home/username/.kube/config")3 if err != nil {4 panic(err.Error())5 }6 clientset, err := kubernetes.NewForConfig(config)7 if err != nil {8 panic(err.Error())9 }10 config.SetNamespace(namespace)11 pods, err := clientset.CoreV1().Pods("").List(metav1.ListOptions{})12 if err != nil {13 panic(err.Error())14 }15 fmt.Printf("There are %d pods in the cluster16", len(pods.Items))17 for _, pod := range pods.Items {18 fmt.Printf("Pod Name: %s19 }20}
SetNamespace
Using AI Code Generation
1func main() {2 config := clientcmd.NewNonInteractiveDeferredLoadingClientConfig(3 clientcmd.NewDefaultClientConfigLoadingRules(),4 &clientcmd.ConfigOverrides{})5 config.SetNamespace("default")6 client, err := kubernetes.NewForConfig(config.ClientConfig())7 if err != nil {8 panic(err.Error())9 }10 pods, err := client.CoreV1().Pods("").List(metav1.ListOptions{})11 if err != nil {12 panic(err.Error())13 }14 fmt.Printf("There are %d pods in the cluster15", len(pods.Items))16}
SetNamespace
Using AI Code Generation
1import (2func main() {3 config := viper.New()4 config.SetNamespace("dev")5 config.SetConfigFile("config.yml")6 err := config.ReadInConfig()7 if err != nil {8 fmt.Println("Error in reading the config file", err)9 }10 fmt.Println(config.GetString("url"))11}
Automation Testing Tutorials
Learn to execute automation testing from scratch with LambdaTest Learning Hub. Right from setting up the prerequisites to run your first automation test, to following best practices and diving deeper into advanced test scenarios. LambdaTest Learning Hubs compile a list of step-by-step guides to help you be proficient with different test automation frameworks i.e. Selenium, Cypress, TestNG etc.
LambdaTest Learning Hubs:
- JUnit Tutorial
- TestNG Tutorial
- Webdriver Tutorial
- WebDriverIO Tutorial
- Protractor Tutorial
- Selenium 4 Tutorial
- Jenkins Tutorial
- NUnit Tutorial
- Jest Tutorial
- Playwright Tutorial
- Cypress Tutorial
- PyTest Tutorial
YouTube
You could also refer to video tutorials over LambdaTest YouTube channel to get step by step demonstration from industry experts.
Run Testkube automation tests on LambdaTest cloud grid
Perform automation testing on 3000+ real desktop and mobile devices online.
Most used method in
- NewConfigureAPIURICmd
- NewConfigureNamespaceCmd
- Init
- Load
- Save
- TestSaveTelemetryEnabled
- EnableAnalytics
- DisableAnalytics
- SetNamespace
- SetInitialized
- EnableOAuth
- DisableOauth
- getDir
- getPath
- NewConfigMapConfig
- GetUniqueClusterId
- GetTelemetryEnabled
- Get
- Upsert
- NewMongoRespository
- getRepository
- TestStorage
Try LambdaTest Now !!
Get 100 minutes of automation test minutes FREE!!
Was this article helpful?
