@@ -25,7 +25,6 @@ import (
25
25
"k8s.io/api/core/v1"
26
26
storagev1 "k8s.io/api/storage/v1"
27
27
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
28
- utilerrors "k8s.io/apimachinery/pkg/util/errors"
29
28
clientset "k8s.io/client-go/kubernetes"
30
29
"k8s.io/kubernetes/test/e2e/framework"
31
30
"k8s.io/kubernetes/test/e2e/storage/testpatterns"
@@ -67,12 +66,10 @@ func (t *volumeModeTestSuite) defineTests(driver TestDriver, pattern testpattern
67
66
config * PerTestConfig
68
67
testCleanup func ()
69
68
70
- cs clientset.Interface
71
- ns * v1.Namespace
72
- sc * storagev1.StorageClass
73
- pvc * v1.PersistentVolumeClaim
74
- pv * v1.PersistentVolume
75
- volume TestVolume
69
+ cs clientset.Interface
70
+ ns * v1.Namespace
71
+ // genericVolumeTestResource contains pv, pvc, sc, etc., owns cleaning that up
72
+ genericVolumeTestResource
76
73
}
77
74
var (
78
75
dInfo = driver .GetDriverInfo ()
@@ -104,6 +101,12 @@ func (t *volumeModeTestSuite) defineTests(driver TestDriver, pattern testpattern
104
101
volumeNodeAffinity * v1.VolumeNodeAffinity
105
102
)
106
103
104
+ l .genericVolumeTestResource = genericVolumeTestResource {
105
+ driver : driver ,
106
+ config : l .config ,
107
+ pattern : pattern ,
108
+ }
109
+
107
110
// Create volume for pre-provisioned volume tests
108
111
l .volume = CreateVolume (driver , l .config , pattern .VolType )
109
112
@@ -144,26 +147,7 @@ func (t *volumeModeTestSuite) defineTests(driver TestDriver, pattern testpattern
144
147
}
145
148
146
149
cleanup := func () {
147
- if l .pv != nil || l .pvc != nil {
148
- By ("Deleting pv and pvc" )
149
- errs := framework .PVPVCCleanup (f .ClientSet , f .Namespace .Name , l .pv , l .pvc )
150
- if len (errs ) > 0 {
151
- framework .Logf ("Failed to delete PV and/or PVC: %v" , utilerrors .NewAggregate (errs ))
152
- }
153
- l .pv = nil
154
- l .pvc = nil
155
- }
156
-
157
- if l .sc != nil {
158
- By ("Deleting sc" )
159
- deleteStorageClass (f .ClientSet , l .sc .Name )
160
- l .sc = nil
161
- }
162
-
163
- if l .volume != nil {
164
- l .volume .DeleteVolume ()
165
- l .volume = nil
166
- }
150
+ l .cleanupResource ()
167
151
168
152
if l .testCleanup != nil {
169
153
l .testCleanup ()
0 commit comments