Use beta default class annotation for default storageclass tests.

This commit is contained in:
Michelle Au 2017-03-15 11:44:14 -07:00
parent fb243a4b57
commit 3fd61fbfe1

View File

@ -361,12 +361,12 @@ func updateDefaultStorageClass(c clientset.Interface, defaultStr string) {
Expect(err).NotTo(HaveOccurred())
if defaultStr == "" {
delete(sc.Annotations, storageutil.IsDefaultStorageClassAnnotation)
delete(sc.Annotations, storageutil.BetaIsDefaultStorageClassAnnotation)
} else {
if sc.Annotations == nil {
sc.Annotations = make(map[string]string)
}
sc.Annotations[storageutil.IsDefaultStorageClassAnnotation] = defaultStr
sc.Annotations[storageutil.BetaIsDefaultStorageClassAnnotation] = defaultStr
}
sc, err = c.StorageV1().StorageClasses().Update(sc)