7
7
8
8
from kubernetes import config , client
9
9
from kubernetes .client import CoreV1Api , ExtensionsV1beta1Api , RbacAuthorizationV1beta1Api , CustomObjectsApi , \
10
- ApiextensionsV1beta1Api
10
+ ApiextensionsV1beta1Api , AppsV1Api
11
11
12
12
from suite .custom_resources_utils import create_crds_from_yaml , delete_crd , create_virtual_server_from_yaml , \
13
13
delete_virtual_server
@@ -36,11 +36,13 @@ class KubeApis:
36
36
"""
37
37
def __init__ (self , v1 : CoreV1Api ,
38
38
extensions_v1_beta1 : ExtensionsV1beta1Api ,
39
+ apps_v1_api : AppsV1Api ,
39
40
rbac_v1_beta1 : RbacAuthorizationV1beta1Api ,
40
41
api_extensions_v1_beta1 : ApiextensionsV1beta1Api ,
41
42
custom_objects : CustomObjectsApi ):
42
43
self .v1 = v1
43
44
self .extensions_v1_beta1 = extensions_v1_beta1
45
+ self .apps_v1_api = apps_v1_api
44
46
self .rbac_v1_beta1 = rbac_v1_beta1
45
47
self .api_extensions_v1_beta1 = api_extensions_v1_beta1
46
48
self .custom_objects = custom_objects
@@ -117,11 +119,11 @@ def ingress_controller(cli_arguments, kube_apis, ingress_controller_prerequisite
117
119
"""
118
120
namespace = ingress_controller_prerequisites .namespace
119
121
print ("------------------------- Create IC -----------------------------------" )
120
- name = create_ingress_controller (kube_apis .v1 , kube_apis .extensions_v1_beta1 , cli_arguments , namespace )
122
+ name = create_ingress_controller (kube_apis .v1 , kube_apis .apps_v1_api , cli_arguments , namespace )
121
123
122
124
def fin ():
123
125
print ("Delete IC:" )
124
- delete_ingress_controller (kube_apis .extensions_v1_beta1 , name , cli_arguments ['deployment-type' ], namespace )
126
+ delete_ingress_controller (kube_apis .apps_v1_api , name , cli_arguments ['deployment-type' ], namespace )
125
127
126
128
request .addfinalizer (fin )
127
129
@@ -193,10 +195,11 @@ def kube_apis(cli_arguments) -> KubeApis:
193
195
config .load_kube_config (config_file = kubeconfig , context = context_name , persist_config = False )
194
196
v1 = client .CoreV1Api ()
195
197
extensions_v1_beta1 = client .ExtensionsV1beta1Api ()
198
+ apps_v1_api = client .AppsV1Api ()
196
199
rbac_v1_beta1 = client .RbacAuthorizationV1beta1Api ()
197
200
api_extensions_v1_beta1 = client .ApiextensionsV1beta1Api ()
198
201
custom_objects = client .CustomObjectsApi ()
199
- return KubeApis (v1 , extensions_v1_beta1 , rbac_v1_beta1 , api_extensions_v1_beta1 , custom_objects )
202
+ return KubeApis (v1 , extensions_v1_beta1 , apps_v1_api , rbac_v1_beta1 , api_extensions_v1_beta1 , custom_objects )
200
203
201
204
202
205
@pytest .fixture (scope = "session" , autouse = True )
@@ -271,7 +274,7 @@ def crd_ingress_controller(cli_arguments, kube_apis, ingress_controller_prerequi
271
274
crd_names = create_crds_from_yaml (kube_apis .api_extensions_v1_beta1 ,
272
275
f"{ DEPLOYMENTS } /common/custom-resource-definitions.yaml" )
273
276
print ("------------------------- Create IC -----------------------------------" )
274
- name = create_ingress_controller (kube_apis .v1 , kube_apis .extensions_v1_beta1 , cli_arguments , namespace ,
277
+ name = create_ingress_controller (kube_apis .v1 , kube_apis .apps_v1_api , cli_arguments , namespace ,
275
278
request .param .get ('extra_args' , None ))
276
279
ensure_connection_to_public_endpoint (ingress_controller_endpoint .public_ip ,
277
280
ingress_controller_endpoint .port ,
@@ -282,7 +285,7 @@ def fin():
282
285
print ("Remove the CRD:" )
283
286
delete_crd (kube_apis .api_extensions_v1_beta1 , crd_name )
284
287
print ("Remove the IC:" )
285
- delete_ingress_controller (kube_apis .extensions_v1_beta1 , name , cli_arguments ['deployment-type' ], namespace )
288
+ delete_ingress_controller (kube_apis .apps_v1_api , name , cli_arguments ['deployment-type' ], namespace )
286
289
print ("Restore the ClusterRole:" )
287
290
patch_rbac (kube_apis .rbac_v1_beta1 , f"{ DEPLOYMENTS } /rbac/rbac.yaml" )
288
291
@@ -339,7 +342,7 @@ def virtual_server_setup(request, kube_apis, crd_ingress_controller, ingress_con
339
342
def fin ():
340
343
print ("Clean up Virtual Server Example:" )
341
344
delete_virtual_server (kube_apis .custom_objects , vs_name , test_namespace )
342
- delete_common_app (kube_apis .v1 , kube_apis .extensions_v1_beta1 , common_app , test_namespace )
345
+ delete_common_app (kube_apis .v1 , kube_apis .apps_v1_api , common_app , test_namespace )
343
346
344
347
request .addfinalizer (fin )
345
348
0 commit comments