144 lines
3.7 KiB
Go
144 lines
3.7 KiB
Go
/*
|
|
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/apimachinery/pkg/types"
|
|
"k8s.io/kubernetes/pkg/api/v1"
|
|
)
|
|
|
|
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: v1.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 "testPlugin"
|
|
}
|
|
|
|
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) 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{}
|
|
vpm.InitPlugins(newTestPlugin(), nil)
|
|
|
|
plug, err := vpm.FindPluginByName("testPlugin")
|
|
if err != nil {
|
|
t.Errorf("Can't find the plugin by name")
|
|
}
|
|
if plug.GetPluginName() != "testPlugin" {
|
|
t.Errorf("Wrong name: %s", plug.GetPluginName())
|
|
}
|
|
}
|
|
|
|
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).")
|
|
}
|
|
}
|