mirror of https://github.com/k3s-io/k3s
102 lines
3.0 KiB
Go
102 lines
3.0 KiB
Go
/*
|
|
Copyright 2018 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 testsuites
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
storagev1 "k8s.io/api/storage/v1"
|
|
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
|
|
"k8s.io/kubernetes/test/e2e/framework"
|
|
"k8s.io/kubernetes/test/e2e/storage/testpatterns"
|
|
)
|
|
|
|
// GetDriverNameWithFeatureTags returns driver name with feature tags
|
|
// For example)
|
|
// - [Driver: nfs]
|
|
// - [Driver: rbd][Feature:Volumes]
|
|
func GetDriverNameWithFeatureTags(driver TestDriver) string {
|
|
dInfo := driver.GetDriverInfo()
|
|
|
|
return fmt.Sprintf("[Driver: %s]%s", dInfo.Name, dInfo.FeatureTag)
|
|
}
|
|
|
|
// CreateVolume creates volume for test unless dynamicPV test
|
|
func CreateVolume(driver TestDriver, volType testpatterns.TestVolType) interface{} {
|
|
switch volType {
|
|
case testpatterns.InlineVolume:
|
|
fallthrough
|
|
case testpatterns.PreprovisionedPV:
|
|
if pDriver, ok := driver.(PreprovisionedVolumeTestDriver); ok {
|
|
return pDriver.CreateVolume(volType)
|
|
}
|
|
case testpatterns.DynamicPV:
|
|
// No need to create volume
|
|
default:
|
|
framework.Failf("Invalid volType specified: %v", volType)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// DeleteVolume deletes volume for test unless dynamicPV test
|
|
func DeleteVolume(driver TestDriver, volType testpatterns.TestVolType, testResource interface{}) {
|
|
switch volType {
|
|
case testpatterns.InlineVolume:
|
|
fallthrough
|
|
case testpatterns.PreprovisionedPV:
|
|
if pDriver, ok := driver.(PreprovisionedVolumeTestDriver); ok {
|
|
pDriver.DeleteVolume(volType, testResource)
|
|
}
|
|
case testpatterns.DynamicPV:
|
|
// No need to delete volume
|
|
default:
|
|
framework.Failf("Invalid volType specified: %v", volType)
|
|
}
|
|
}
|
|
|
|
// GetStorageClass constructs a new StorageClass instance
|
|
// with a unique name that is based on namespace + suffix.
|
|
func GetStorageClass(
|
|
provisioner string,
|
|
parameters map[string]string,
|
|
bindingMode *storagev1.VolumeBindingMode,
|
|
ns string,
|
|
suffix string,
|
|
) *storagev1.StorageClass {
|
|
if bindingMode == nil {
|
|
defaultBindingMode := storagev1.VolumeBindingImmediate
|
|
bindingMode = &defaultBindingMode
|
|
}
|
|
return &storagev1.StorageClass{
|
|
TypeMeta: metav1.TypeMeta{
|
|
Kind: "StorageClass",
|
|
},
|
|
ObjectMeta: metav1.ObjectMeta{
|
|
// Name must be unique, so let's base it on namespace name
|
|
Name: ns + "-" + suffix,
|
|
},
|
|
Provisioner: provisioner,
|
|
Parameters: parameters,
|
|
VolumeBindingMode: bindingMode,
|
|
}
|
|
}
|
|
|
|
// GetUniqueDriverName returns unique driver name that can be used parallelly in tests
|
|
func GetUniqueDriverName(driver TestDriver) string {
|
|
return fmt.Sprintf("%s-%s", driver.GetDriverInfo().Name, driver.GetDriverInfo().Config.Framework.UniqueName)
|
|
}
|