rename daemonset (#5390)

Signed-off-by: Lyndon-Li <lyonghui@vmware.com> Signed-off-by: Lyndon-Li <lyonghui@vmware.com>
2026-09-03 06:37:11 +00:00 · 2022-10-11 14:18:21 +08:00
parent 19e158a2a6
commit 83ea1cc58b
18 changed files with 124 additions and 84 deletions
@@ -0,0 +1,110 @@
+/*
+Copyright 2019 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 nodeagent
+
+import (
+	"context"
+	"os"
+	"path/filepath"
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+	corev1 "k8s.io/api/core/v1"
+	v1 "k8s.io/api/core/v1"
+	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
+	"k8s.io/client-go/kubernetes/fake"
+
+	"github.com/vmware-tanzu/velero/pkg/builder"
+	testutil "github.com/vmware-tanzu/velero/pkg/test"
+)
+
+func Test_validatePodVolumesHostPath(t *testing.T) {
+	tests := []struct {
+		name    string
+		pods    []*corev1.Pod
+		dirs    []string
+		wantErr bool
+	}{
+		{
+			name: "no error when pod volumes are present",
+			pods: []*corev1.Pod{
+				builder.ForPod("foo", "bar").ObjectMeta(builder.WithUID("foo")).Result(),
+				builder.ForPod("zoo", "raz").ObjectMeta(builder.WithUID("zoo")).Result(),
+			},
+			dirs:    []string{"foo", "zoo"},
+			wantErr: false,
+		},
+		{
+			name: "no error when pod volumes are present and there are mirror pods",
+			pods: []*corev1.Pod{
+				builder.ForPod("foo", "bar").ObjectMeta(builder.WithUID("foo")).Result(),
+				builder.ForPod("zoo", "raz").ObjectMeta(builder.WithUID("zoo"), builder.WithAnnotations(v1.MirrorPodAnnotationKey, "baz")).Result(),
+			},
+			dirs:    []string{"foo", "baz"},
+			wantErr: false,
+		},
+		{
+			name: "error when all pod volumes missing",
+			pods: []*corev1.Pod{
+				builder.ForPod("foo", "bar").ObjectMeta(builder.WithUID("foo")).Result(),
+				builder.ForPod("zoo", "raz").ObjectMeta(builder.WithUID("zoo")).Result(),
+			},
+			dirs:    []string{"unexpected-dir"},
+			wantErr: true,
+		},
+		{
+			name: "error when some pod volumes missing",
+			pods: []*corev1.Pod{
+				builder.ForPod("foo", "bar").ObjectMeta(builder.WithUID("foo")).Result(),
+				builder.ForPod("zoo", "raz").ObjectMeta(builder.WithUID("zoo")).Result(),
+			},
+			dirs:    []string{"foo"},
+			wantErr: true,
+		},
+	}
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			fs := testutil.NewFakeFileSystem()
+
+			for _, dir := range tt.dirs {
+				err := fs.MkdirAll(filepath.Join("/host_pods/", dir), os.ModePerm)
+				if err != nil {
+					t.Error(err)
+				}
+			}
+
+			kubeClient := fake.NewSimpleClientset()
+			for _, pod := range tt.pods {
+				_, err := kubeClient.CoreV1().Pods(pod.GetNamespace()).Create(context.TODO(), pod, metav1.CreateOptions{})
+				if err != nil {
+					t.Error(err)
+				}
+			}
+
+			s := &resticServer{
+				logger:     testutil.NewLogger(),
+				fileSystem: fs,
+			}
+
+			err := s.validatePodVolumesHostPath(kubeClient)
+			if tt.wantErr {
+				assert.Error(t, err)
+			} else {
+				assert.NoError(t, err)
+			}
+		})
+	}
+}