vendor files

vendor/k8s.io/kubernetes/pkg/volume/plugins_test.go

@@ -0,0 +1,167 @@
+/*
+Copyright 2015 The Kubernetes Authors.
+
+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 volume
+
+import (
+	"testing"
+
+	"k8s.io/api/core/v1"
+	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
+	"k8s.io/apimachinery/pkg/types"
+)
+
+const testPluginName = "kubernetes.io/testPlugin"
+
+func TestSpecSourceConverters(t *testing.T) {
+	v := &v1.Volume{
+		Name:         "foo",
+		VolumeSource: v1.VolumeSource{EmptyDir: &v1.EmptyDirVolumeSource{}},
+	}
+
+	converted := NewSpecFromVolume(v)
+	if converted.Volume.EmptyDir == nil {
+		t.Errorf("Unexpected nil EmptyDir: %#v", converted)
+	}
+	if v.Name != converted.Name() {
+		t.Errorf("Expected %v but got %v", v.Name, converted.Name())
+	}
+
+	pv := &v1.PersistentVolume{
+		ObjectMeta: metav1.ObjectMeta{Name: "bar"},
+		Spec: v1.PersistentVolumeSpec{
+			PersistentVolumeSource: v1.PersistentVolumeSource{AWSElasticBlockStore: &v1.AWSElasticBlockStoreVolumeSource{}},
+		},
+	}
+
+	converted = NewSpecFromPersistentVolume(pv, false)
+	if converted.PersistentVolume.Spec.AWSElasticBlockStore == nil {
+		t.Errorf("Unexpected nil AWSElasticBlockStore: %#v", converted)
+	}
+	if pv.Name != converted.Name() {
+		t.Errorf("Expected %v but got %v", pv.Name, converted.Name())
+	}
+}
+
+type testPlugins struct {
+}
+
+func (plugin *testPlugins) Init(host VolumeHost) error {
+	return nil
+}
+
+func (plugin *testPlugins) GetPluginName() string {
+	return testPluginName
+}
+
+func (plugin *testPlugins) GetVolumeName(spec *Spec) (string, error) {
+	return "", nil
+}
+
+func (plugin *testPlugins) CanSupport(spec *Spec) bool {
+	return true
+}
+
+func (plugin *testPlugins) RequiresRemount() bool {
+	return false
+}
+
+func (plugin *testPlugins) SupportsMountOption() bool {
+	return false
+}
+
+func (plugin *testPlugins) SupportsBulkVolumeVerification() bool {
+	return false
+}
+
+func (plugin *testPlugins) NewMounter(spec *Spec, podRef *v1.Pod, opts VolumeOptions) (Mounter, error) {
+	return nil, nil
+}
+
+func (plugin *testPlugins) NewUnmounter(name string, podUID types.UID) (Unmounter, error) {
+	return nil, nil
+}
+
+func (plugin *testPlugins) ConstructVolumeSpec(volumeName, mountPath string) (*Spec, error) {
+	return nil, nil
+}
+
+func newTestPlugin() []VolumePlugin {
+	return []VolumePlugin{&testPlugins{}}
+}
+
+func TestVolumePluginMgrFunc(t *testing.T) {
+	vpm := VolumePluginMgr{}
+	var prober DynamicPluginProber = nil // TODO (#51147) inject mock
+	vpm.InitPlugins(newTestPlugin(), prober, nil)
+
+	plug, err := vpm.FindPluginByName(testPluginName)
+	if err != nil {
+		t.Errorf("Can't find the plugin by name")
+	}
+	if plug.GetPluginName() != testPluginName {
+		t.Errorf("Wrong name: %s", plug.GetPluginName())
+	}
+
+	plug, err = vpm.FindPluginBySpec(nil)
+	if err == nil {
+		t.Errorf("Should return error if volume spec is nil")
+	}
+
+	volumeSpec := &Spec{}
+	plug, err = vpm.FindPluginBySpec(volumeSpec)
+	if err != nil {
+		t.Errorf("Should return test plugin if volume spec is not nil")
+	}
+}
+
+func Test_ValidatePodTemplate(t *testing.T) {
+	pod := &v1.Pod{
+		Spec: v1.PodSpec{
+			Volumes: []v1.Volume{
+				{
+					Name:         "vol",
+					VolumeSource: v1.VolumeSource{},
+				},
+			},
+		},
+	}
+	var want error
+	if got := ValidateRecyclerPodTemplate(pod); got != want {
+		t.Errorf("isPodTemplateValid(%v) returned (%v), want (%v)", pod.String(), got.Error(), want)
+	}
+
+	// Check that the default recycle pod template is valid
+	pod = NewPersistentVolumeRecyclerPodTemplate()
+	want = nil
+	if got := ValidateRecyclerPodTemplate(pod); got != want {
+		t.Errorf("isPodTemplateValid(%v) returned (%v), want (%v)", pod.String(), got.Error(), want)
+	}
+
+	pod = &v1.Pod{
+		Spec: v1.PodSpec{
+			Containers: []v1.Container{
+				{
+					Name: "pv-recycler",
+				},
+			},
+		},
+	}
+	// want = an error
+	if got := ValidateRecyclerPodTemplate(pod); got == nil {
+		t.Errorf("isPodTemplateValid(%v) returned (%v), want (%v)", pod.String(), got, "Error: pod specification does not contain any volume(s).")
+	}
+}