@@ -17,25 +17,16 @@ limitations under the License.
17
17
package storage
18
18
19
19
import (
20
- "context"
21
20
"testing"
22
21
23
22
policyv1beta1 "k8s.io/api/policy/v1beta1"
24
- "k8s.io/apimachinery/pkg/api/apitesting"
25
23
apierrors "k8s.io/apimachinery/pkg/api/errors"
26
24
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
27
25
"k8s.io/apimachinery/pkg/runtime"
28
- "k8s.io/apimachinery/pkg/runtime/serializer"
29
- examplev1 "k8s.io/apiserver/pkg/apis/example/v1"
30
26
genericapirequest "k8s.io/apiserver/pkg/endpoints/request"
31
- "k8s.io/apiserver/pkg/registry/generic"
32
- genericregistry "k8s.io/apiserver/pkg/registry/generic/registry"
33
- "k8s.io/apiserver/pkg/storage"
34
- etcd3testing "k8s.io/apiserver/pkg/storage/etcd3/testing"
35
27
"k8s.io/client-go/kubernetes/fake"
36
28
api "k8s.io/kubernetes/pkg/apis/core"
37
29
"k8s.io/kubernetes/pkg/apis/policy"
38
- "k8s.io/kubernetes/pkg/registry/registrytest"
39
30
)
40
31
41
32
func TestEviction (t * testing.T ) {
@@ -155,41 +146,6 @@ func TestEviction(t *testing.T) {
155
146
}
156
147
}
157
148
158
- type FailDeleteUpdateStorage struct {
159
- storage.Interface
160
- }
161
-
162
- func (f FailDeleteUpdateStorage ) Delete (ctx context.Context , key string , out runtime.Object , precondition * storage.Preconditions , validateDeletion storage.ValidateObjectFunc ) error {
163
- return storage .NewKeyNotFoundError (key , 0 )
164
- }
165
-
166
- func (f FailDeleteUpdateStorage ) GuaranteedUpdate (ctx context.Context , key string , ptrToType runtime.Object , ignoreNotFound bool ,
167
- preconditions * storage.Preconditions , tryUpdate storage.UpdateFunc , suggestion ... runtime.Object ) error {
168
- return storage .NewKeyNotFoundError (key , 0 )
169
- }
170
-
171
- var scheme = runtime .NewScheme ()
172
- var codecs = serializer .NewCodecFactory (scheme )
173
-
174
- func newFailDeleteUpdateStorage (t * testing.T ) (* REST , * etcd3testing.EtcdTestServer ) {
175
- etcdStorage , server := registrytest .NewEtcdStorage (t , "" )
176
- restOptions := generic.RESTOptions {
177
- StorageConfig : etcdStorage ,
178
- Decorator : generic .UndecoratedStorage ,
179
- DeleteCollectionWorkers : 3 ,
180
- ResourcePrefix : "pods" ,
181
- }
182
- storage , err := NewStorage (restOptions , nil , nil , nil )
183
- if err != nil {
184
- t .Fatalf ("unexpected error from REST storage: %v" , err )
185
- }
186
- storage .Pod .Store .Storage = genericregistry.DryRunnableStorage {
187
- Storage : FailDeleteUpdateStorage {storage .Pod .Store .Storage .Storage },
188
- Codec : apitesting .TestStorageCodec (codecs , examplev1 .SchemeGroupVersion ),
189
- }
190
- return storage .Pod , server
191
- }
192
-
193
149
func TestEvictionDryRun (t * testing.T ) {
194
150
testcases := []struct {
195
151
name string
@@ -227,7 +183,7 @@ func TestEvictionDryRun(t *testing.T) {
227
183
for _ , tc := range testcases {
228
184
t .Run (tc .name , func (t * testing.T ) {
229
185
testContext := genericapirequest .WithNamespace (genericapirequest .NewContext (), metav1 .NamespaceDefault )
230
- storage , server := newFailDeleteUpdateStorage (t )
186
+ storage , _ , _ , server := newStorage (t )
231
187
defer server .Terminate (t )
232
188
defer storage .Store .DestroyFunc ()
233
189
0 commit comments