@@ -36,7 +36,7 @@ import (
36
36
"k8s.io/client-go/discovery"
37
37
"k8s.io/client-go/dynamic"
38
38
clientset "k8s.io/client-go/kubernetes"
39
- "k8s.io/kubernetes/pkg/features "
39
+ "k8s.io/component-base/featuregate "
40
40
"k8s.io/kubernetes/test/e2e/framework"
41
41
e2enode "k8s.io/kubernetes/test/e2e/framework/node"
42
42
e2essh "k8s.io/kubernetes/test/e2e/framework/ssh"
@@ -45,6 +45,9 @@ import (
45
45
// TestContext should be used by all tests to access common context data.
46
46
var TestContext framework.TestContextType
47
47
48
+ // New local storage types to support local storage capacity isolation
49
+ var localStorageCapacityIsolation featuregate.Feature = "LocalStorageCapacityIsolation"
50
+
48
51
func skipInternalf (caller int , format string , args ... interface {}) {
49
52
msg := fmt .Sprintf (format , args ... )
50
53
framework .Logf (msg )
@@ -131,8 +134,8 @@ func SkipUnlessAtLeast(value int, minValue int, message string) {
131
134
132
135
// SkipUnlessLocalEphemeralStorageEnabled skips if the LocalStorageCapacityIsolation is not enabled.
133
136
func SkipUnlessLocalEphemeralStorageEnabled () {
134
- if ! utilfeature .DefaultFeatureGate .Enabled (features . LocalStorageCapacityIsolation ) {
135
- skipInternalf (1 , "Only supported when %v feature is enabled" , features . LocalStorageCapacityIsolation )
137
+ if ! utilfeature .DefaultFeatureGate .Enabled (localStorageCapacityIsolation ) {
138
+ skipInternalf (1 , "Only supported when %v feature is enabled" , localStorageCapacityIsolation )
136
139
}
137
140
}
138
141
0 commit comments