mirror of
https://github.com/vmware-tanzu/velero.git
synced 2026-09-13 11:34:54 +00:00
Shubham Pampattiwar
f457a95802
These single-object delete helpers were introduced earlier but are no longer called in production code: DeleteSecretsWithLabel and DeleteConfigMapsWithLabel now delete inline with UID preconditions. Remove the dead functions and their tests. Signed-off-by: Shubham Pampattiwar <spampatt@redhat.com>
363 lines
11 KiB
Go
363 lines
11 KiB
Go
/*
|
|
Copyright the Velero contributors.
|
|
|
|
Licensed under the Apache License, Version 2.0 (the "License");
|
|
you may not use this file except in compliance with the License.
|
|
You may obtain a copy of the License at
|
|
|
|
http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
Unless required by applicable law or agreed to in writing, software
|
|
distributed under the License is distributed on an "AS IS" BASIS,
|
|
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
See the License for the specific language governing permissions and
|
|
limitations under the License.
|
|
*/
|
|
|
|
package kube
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/sirupsen/logrus"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
corev1api "k8s.io/api/core/v1"
|
|
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
|
|
k8sruntime "k8s.io/apimachinery/pkg/runtime"
|
|
"k8s.io/client-go/kubernetes/fake"
|
|
)
|
|
|
|
const testCopyLabel = "velero.io/backup-pvc-secret"
|
|
|
|
func TestCopySecret(t *testing.T) {
|
|
log := logrus.New()
|
|
|
|
tests := []struct {
|
|
name string
|
|
secretName string
|
|
sourceNS string
|
|
targetNS string
|
|
ownerName string
|
|
objects []k8sruntime.Object
|
|
expectErr bool
|
|
errContains string
|
|
}{
|
|
{
|
|
name: "successfully copies secret to target namespace",
|
|
secretName: "ceph-csi-kms-token",
|
|
sourceNS: "app-ns",
|
|
targetNS: "velero",
|
|
ownerName: "du-123",
|
|
objects: []k8sruntime.Object{
|
|
&corev1api.Secret{
|
|
ObjectMeta: metav1.ObjectMeta{Name: "ceph-csi-kms-token", Namespace: "app-ns"},
|
|
Data: map[string][]byte{"token": []byte("vault-token-a")},
|
|
Type: corev1api.SecretTypeOpaque,
|
|
},
|
|
},
|
|
},
|
|
{
|
|
name: "returns error when source secret does not exist",
|
|
secretName: "missing-secret",
|
|
sourceNS: "app-ns",
|
|
targetNS: "velero",
|
|
ownerName: "du-123",
|
|
objects: []k8sruntime.Object{},
|
|
expectErr: true,
|
|
errContains: "error getting secret",
|
|
},
|
|
{
|
|
name: "no-op when target already has secret with same data and same owner",
|
|
secretName: "ceph-csi-kms-token",
|
|
sourceNS: "app-ns",
|
|
targetNS: "velero",
|
|
ownerName: "du-123",
|
|
objects: []k8sruntime.Object{
|
|
&corev1api.Secret{
|
|
ObjectMeta: metav1.ObjectMeta{Name: "ceph-csi-kms-token", Namespace: "app-ns"},
|
|
Data: map[string][]byte{"token": []byte("same-token")},
|
|
Type: corev1api.SecretTypeOpaque,
|
|
},
|
|
&corev1api.Secret{
|
|
ObjectMeta: metav1.ObjectMeta{
|
|
Name: "ceph-csi-kms-token", Namespace: "velero",
|
|
Labels: map[string]string{testCopyLabel: "du-123"},
|
|
},
|
|
Data: map[string][]byte{"token": []byte("same-token")},
|
|
Type: corev1api.SecretTypeOpaque,
|
|
},
|
|
},
|
|
},
|
|
{
|
|
name: "returns collision when same data but different owner",
|
|
secretName: "ceph-csi-kms-token",
|
|
sourceNS: "app-ns",
|
|
targetNS: "velero",
|
|
ownerName: "du-456",
|
|
objects: []k8sruntime.Object{
|
|
&corev1api.Secret{
|
|
ObjectMeta: metav1.ObjectMeta{Name: "ceph-csi-kms-token", Namespace: "app-ns"},
|
|
Data: map[string][]byte{"token": []byte("same-token")},
|
|
Type: corev1api.SecretTypeOpaque,
|
|
},
|
|
&corev1api.Secret{
|
|
ObjectMeta: metav1.ObjectMeta{
|
|
Name: "ceph-csi-kms-token", Namespace: "velero",
|
|
Labels: map[string]string{testCopyLabel: "du-123"},
|
|
},
|
|
Data: map[string][]byte{"token": []byte("same-token")},
|
|
Type: corev1api.SecretTypeOpaque,
|
|
},
|
|
},
|
|
expectErr: true,
|
|
errContains: "collision",
|
|
},
|
|
{
|
|
name: "returns collision error when target has secret with different data",
|
|
secretName: "ceph-csi-kms-token",
|
|
sourceNS: "app-ns",
|
|
targetNS: "velero",
|
|
ownerName: "du-123",
|
|
objects: []k8sruntime.Object{
|
|
&corev1api.Secret{
|
|
ObjectMeta: metav1.ObjectMeta{Name: "ceph-csi-kms-token", Namespace: "app-ns"},
|
|
Data: map[string][]byte{"token": []byte("token-a")},
|
|
Type: corev1api.SecretTypeOpaque,
|
|
},
|
|
&corev1api.Secret{
|
|
ObjectMeta: metav1.ObjectMeta{Name: "ceph-csi-kms-token", Namespace: "velero"},
|
|
Data: map[string][]byte{"token": []byte("token-b")},
|
|
Type: corev1api.SecretTypeOpaque,
|
|
},
|
|
},
|
|
expectErr: true,
|
|
errContains: "secret collision",
|
|
},
|
|
}
|
|
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
fakeClient := fake.NewSimpleClientset(tt.objects...)
|
|
|
|
err := CopySecret(context.Background(), fakeClient.CoreV1(),
|
|
tt.secretName, tt.sourceNS, tt.targetNS,
|
|
map[string]string{testCopyLabel: tt.ownerName}, log)
|
|
|
|
if tt.expectErr {
|
|
require.Error(t, err)
|
|
if tt.errContains != "" {
|
|
assert.Contains(t, err.Error(), tt.errContains)
|
|
}
|
|
return
|
|
}
|
|
|
|
require.NoError(t, err)
|
|
|
|
copied, getErr := fakeClient.CoreV1().Secrets(tt.targetNS).Get(
|
|
context.Background(), tt.secretName, metav1.GetOptions{})
|
|
require.NoError(t, getErr)
|
|
assert.NotNil(t, copied)
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestDeleteSecretsWithLabel(t *testing.T) {
|
|
log := logrus.New()
|
|
|
|
fakeClient := fake.NewSimpleClientset(
|
|
&corev1api.Secret{
|
|
ObjectMeta: metav1.ObjectMeta{
|
|
Name: "secret-1", Namespace: "velero",
|
|
Labels: map[string]string{testCopyLabel: "du-123"},
|
|
},
|
|
},
|
|
&corev1api.Secret{
|
|
ObjectMeta: metav1.ObjectMeta{
|
|
Name: "secret-2", Namespace: "velero",
|
|
Labels: map[string]string{testCopyLabel: "du-456"},
|
|
},
|
|
},
|
|
)
|
|
|
|
DeleteSecretsWithLabel(context.Background(), fakeClient.CoreV1(), "velero",
|
|
testCopyLabel, "du-123", log)
|
|
|
|
_, err := fakeClient.CoreV1().Secrets("velero").Get(
|
|
context.Background(), "secret-1", metav1.GetOptions{})
|
|
require.Error(t, err, "secret-1 should be deleted")
|
|
|
|
_, err = fakeClient.CoreV1().Secrets("velero").Get(
|
|
context.Background(), "secret-2", metav1.GetOptions{})
|
|
assert.NoError(t, err, "secret-2 should still exist")
|
|
}
|
|
|
|
func TestCopyConfigMap(t *testing.T) {
|
|
log := logrus.New()
|
|
|
|
tests := []struct {
|
|
name string
|
|
cmName string
|
|
sourceNS string
|
|
targetNS string
|
|
ownerName string
|
|
objects []k8sruntime.Object
|
|
expectErr bool
|
|
errContains string
|
|
}{
|
|
{
|
|
name: "successfully copies configmap to target namespace",
|
|
cmName: "ceph-csi-kms-config",
|
|
sourceNS: "app-ns",
|
|
targetNS: "velero",
|
|
ownerName: "du-123",
|
|
objects: []k8sruntime.Object{
|
|
&corev1api.ConfigMap{
|
|
ObjectMeta: metav1.ObjectMeta{Name: "ceph-csi-kms-config", Namespace: "app-ns"},
|
|
Data: map[string]string{"vaultAddress": "https://vault.example.com"},
|
|
},
|
|
},
|
|
},
|
|
{
|
|
name: "returns error when source configmap does not exist",
|
|
cmName: "missing-cm",
|
|
sourceNS: "app-ns",
|
|
targetNS: "velero",
|
|
ownerName: "du-123",
|
|
objects: []k8sruntime.Object{},
|
|
expectErr: true,
|
|
errContains: "error getting configmap",
|
|
},
|
|
{
|
|
name: "no-op when target already has configmap with same data and same owner",
|
|
cmName: "ceph-csi-kms-config",
|
|
sourceNS: "app-ns",
|
|
targetNS: "velero",
|
|
ownerName: "du-123",
|
|
objects: []k8sruntime.Object{
|
|
&corev1api.ConfigMap{
|
|
ObjectMeta: metav1.ObjectMeta{Name: "ceph-csi-kms-config", Namespace: "app-ns"},
|
|
Data: map[string]string{"vaultAddress": "https://vault.example.com"},
|
|
},
|
|
&corev1api.ConfigMap{
|
|
ObjectMeta: metav1.ObjectMeta{
|
|
Name: "ceph-csi-kms-config", Namespace: "velero",
|
|
Labels: map[string]string{testCopyLabel: "du-123"},
|
|
},
|
|
Data: map[string]string{"vaultAddress": "https://vault.example.com"},
|
|
},
|
|
},
|
|
},
|
|
{
|
|
name: "returns collision when same data but different owner",
|
|
cmName: "ceph-csi-kms-config",
|
|
sourceNS: "app-ns",
|
|
targetNS: "velero",
|
|
ownerName: "du-456",
|
|
objects: []k8sruntime.Object{
|
|
&corev1api.ConfigMap{
|
|
ObjectMeta: metav1.ObjectMeta{Name: "ceph-csi-kms-config", Namespace: "app-ns"},
|
|
Data: map[string]string{"vaultAddress": "https://vault.example.com"},
|
|
},
|
|
&corev1api.ConfigMap{
|
|
ObjectMeta: metav1.ObjectMeta{
|
|
Name: "ceph-csi-kms-config", Namespace: "velero",
|
|
Labels: map[string]string{testCopyLabel: "du-123"},
|
|
},
|
|
Data: map[string]string{"vaultAddress": "https://vault.example.com"},
|
|
},
|
|
},
|
|
expectErr: true,
|
|
errContains: "collision",
|
|
},
|
|
{
|
|
name: "copies configmap with BinaryData",
|
|
cmName: "ceph-csi-kms-config",
|
|
sourceNS: "app-ns",
|
|
targetNS: "velero",
|
|
ownerName: "du-123",
|
|
objects: []k8sruntime.Object{
|
|
&corev1api.ConfigMap{
|
|
ObjectMeta: metav1.ObjectMeta{Name: "ceph-csi-kms-config", Namespace: "app-ns"},
|
|
BinaryData: map[string][]byte{"ca.crt": []byte("binary-ca-bundle")},
|
|
},
|
|
},
|
|
},
|
|
{
|
|
name: "returns collision error when target has configmap with different data",
|
|
cmName: "ceph-csi-kms-config",
|
|
sourceNS: "app-ns",
|
|
targetNS: "velero",
|
|
ownerName: "du-123",
|
|
objects: []k8sruntime.Object{
|
|
&corev1api.ConfigMap{
|
|
ObjectMeta: metav1.ObjectMeta{Name: "ceph-csi-kms-config", Namespace: "app-ns"},
|
|
Data: map[string]string{"vaultAddress": "https://vault-a.example.com"},
|
|
},
|
|
&corev1api.ConfigMap{
|
|
ObjectMeta: metav1.ObjectMeta{Name: "ceph-csi-kms-config", Namespace: "velero"},
|
|
Data: map[string]string{"vaultAddress": "https://vault-b.example.com"},
|
|
},
|
|
},
|
|
expectErr: true,
|
|
errContains: "secret collision",
|
|
},
|
|
}
|
|
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
fakeClient := fake.NewSimpleClientset(tt.objects...)
|
|
|
|
err := CopyConfigMap(context.Background(), fakeClient.CoreV1(),
|
|
tt.cmName, tt.sourceNS, tt.targetNS,
|
|
map[string]string{testCopyLabel: tt.ownerName}, log)
|
|
|
|
if tt.expectErr {
|
|
require.Error(t, err)
|
|
if tt.errContains != "" {
|
|
assert.Contains(t, err.Error(), tt.errContains)
|
|
}
|
|
return
|
|
}
|
|
|
|
require.NoError(t, err)
|
|
|
|
copied, getErr := fakeClient.CoreV1().ConfigMaps(tt.targetNS).Get(
|
|
context.Background(), tt.cmName, metav1.GetOptions{})
|
|
require.NoError(t, getErr)
|
|
assert.NotNil(t, copied)
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestDeleteConfigMapsWithLabel(t *testing.T) {
|
|
log := logrus.New()
|
|
|
|
fakeClient := fake.NewSimpleClientset(
|
|
&corev1api.ConfigMap{
|
|
ObjectMeta: metav1.ObjectMeta{
|
|
Name: "cm-1", Namespace: "velero",
|
|
Labels: map[string]string{testCopyLabel: "du-123"},
|
|
},
|
|
},
|
|
&corev1api.ConfigMap{
|
|
ObjectMeta: metav1.ObjectMeta{
|
|
Name: "cm-2", Namespace: "velero",
|
|
Labels: map[string]string{testCopyLabel: "du-456"},
|
|
},
|
|
},
|
|
)
|
|
|
|
DeleteConfigMapsWithLabel(context.Background(), fakeClient.CoreV1(), "velero",
|
|
testCopyLabel, "du-123", log)
|
|
|
|
_, err := fakeClient.CoreV1().ConfigMaps("velero").Get(
|
|
context.Background(), "cm-1", metav1.GetOptions{})
|
|
require.Error(t, err, "cm-1 should be deleted")
|
|
|
|
_, err = fakeClient.CoreV1().ConfigMaps("velero").Get(
|
|
context.Background(), "cm-2", metav1.GetOptions{})
|
|
assert.NoError(t, err, "cm-2 should still exist")
|
|
}
|