Skip to content

Commit ea6ff2f

Browse files
authored
Make use of patch instead of update (#584)
1 parent b70cc80 commit ea6ff2f

File tree

3 files changed

+6
-4
lines changed

3 files changed

+6
-4
lines changed

kubectl-fdb/cmd/exec_test.go

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -96,7 +96,7 @@ func TestBuildCommand(t *testing.T) {
9696
scheme := runtime.NewScheme()
9797
_ = clientgoscheme.AddToScheme(scheme)
9898
_ = fdbtypes.AddToScheme(scheme)
99-
kubeClient := fake.NewFakeClientWithScheme(scheme, &cluster, &podList)
99+
kubeClient := fake.NewClientBuilder().WithScheme(scheme).WithRuntimeObjects(&cluster, &podList).Build()
100100

101101
command, err := buildCommand(kubeClient, tc.ClusterName, tc.Context, namespace, tc.Command)
102102

kubectl-fdb/cmd/remove_instances.go

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -192,6 +192,8 @@ func removeInstances(kubeClient client.Client, clusterName string, instances []s
192192
}
193193
}
194194

195+
patch := client.MergeFrom(cluster.DeepCopy())
196+
195197
for class, amount := range shrinkMap {
196198
cluster.Spec.ProcessCounts.DecreaseCount(class, amount)
197199
}
@@ -209,5 +211,5 @@ func removeInstances(kubeClient client.Client, clusterName string, instances []s
209211
cluster.Spec.InstancesToRemoveWithoutExclusion = append(cluster.Spec.InstancesToRemoveWithoutExclusion, instances...)
210212
}
211213

212-
return kubeClient.Update(ctx.TODO(), &cluster)
214+
return kubeClient.Patch(ctx.TODO(), &cluster, patch)
213215
}

kubectl-fdb/cmd/remove_instances_test.go

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -128,7 +128,7 @@ func TestRemoveInstances(t *testing.T) {
128128
scheme := runtime.NewScheme()
129129
_ = clientgoscheme.AddToScheme(scheme)
130130
_ = fdbtypes.AddToScheme(scheme)
131-
kubeClient := fake.NewFakeClientWithScheme(scheme, &cluster, &podList)
131+
kubeClient := fake.NewClientBuilder().WithScheme(scheme).WithRuntimeObjects(&cluster, &podList).Build()
132132

133133
err := removeInstances(kubeClient, clusterName, tc.Instances, namespace, tc.WithExclusion, tc.WithShrink, true)
134134
if err != nil {
@@ -219,7 +219,7 @@ func TestGetInstanceIDsFromPod(t *testing.T) {
219219
scheme := runtime.NewScheme()
220220
_ = clientgoscheme.AddToScheme(scheme)
221221
_ = fdbtypes.AddToScheme(scheme)
222-
kubeClient := fake.NewFakeClientWithScheme(scheme, &podList)
222+
kubeClient := fake.NewClientBuilder().WithScheme(scheme).WithRuntimeObjects(&podList).Build()
223223

224224
instances, err := getInstanceIDsFromPod(kubeClient, clusterName, tc.Instances, namespace)
225225
if err != nil {

0 commit comments

Comments
 (0)