Ensure async calls are performed
[osm/N2VC.git] / n2vc / tests / unit / test_k8s_helm_conn.py
1 ##
2 # Licensed under the Apache License, Version 2.0 (the "License"); you may
3 # not use this file except in compliance with the License. You may obtain
4 # a copy of the License at
5 #
6 # http://www.apache.org/licenses/LICENSE-2.0
7 #
8 # Unless required by applicable law or agreed to in writing, software
9 # distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
10 # WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
11 # License for the specific language governing permissions and limitations
12 # under the License.
13 #
14 # For those usages not covered by the Apache License, Version 2.0 please
15 # contact: alfonso.tiernosepulveda@telefonica.com
16 ##
17
18 import asynctest
19 import logging
20
21 from asynctest.mock import Mock
22 from osm_common.dbmemory import DbMemory
23 from osm_common.fslocal import FsLocal
24 from n2vc.k8s_helm_conn import K8sHelmConnector
25
26 __author__ = "Isabel Lloret <illoret@indra.es>"
27
28
29 class TestK8sHelmConn(asynctest.TestCase):
30 logging.basicConfig(level=logging.DEBUG)
31 logger = logging.getLogger(__name__)
32 logger.setLevel(logging.DEBUG)
33
34 async def setUp(self):
35 self.db = Mock(DbMemory())
36 self.fs = asynctest.Mock(FsLocal())
37 self.fs.path = "./tmp/"
38 self.namespace = "testk8s"
39 self.service_account = "osm"
40 self.cluster_id = "helm_cluster_id"
41 self.cluster_uuid = self.cluster_id
42 # pass fake kubectl and helm commands to make sure it does not call actual commands
43 K8sHelmConnector._check_file_exists = asynctest.Mock(return_value=True)
44 K8sHelmConnector._local_async_exec = asynctest.CoroutineMock(
45 return_value=(0, "")
46 )
47 cluster_dir = self.fs.path + self.cluster_id
48 self.kube_config = self.fs.path + self.cluster_id + "/.kube/config"
49 self.helm_home = self.fs.path + self.cluster_id + "/.helm"
50 self.env = {
51 "HELM_HOME": "{}/.helm".format(cluster_dir),
52 "KUBECONFIG": "{}/.kube/config".format(cluster_dir),
53 }
54 self.helm_conn = K8sHelmConnector(self.fs, self.db, log=self.logger)
55 self.logger.debug("Set up executed")
56
57 @asynctest.fail_on(active_handles=True)
58 async def test_init_env(self):
59 # TODO
60 pass
61
62 @asynctest.fail_on(active_handles=True)
63 async def test_repo_add(self):
64 repo_name = "bitnami"
65 repo_url = "https://charts.bitnami.com/bitnami"
66 self.helm_conn._local_async_exec = asynctest.CoroutineMock(return_value=("", 0))
67
68 await self.helm_conn.repo_add(self.cluster_uuid, repo_name, repo_url)
69
70 self.helm_conn.fs.sync.assert_called_once_with(from_path=self.cluster_id)
71 self.helm_conn.fs.reverse_sync.assert_called_once_with(
72 from_path=self.cluster_id
73 )
74 self.assertEqual(
75 self.helm_conn._local_async_exec.call_count,
76 2,
77 "local_async_exec expected 2 calls, called {}".format(
78 self.helm_conn._local_async_exec.call_count
79 ),
80 )
81
82 repo_update_command = (
83 "env KUBECONFIG=./tmp/helm_cluster_id/.kube/config /usr/bin/helm repo update {}"
84 ).format(repo_name)
85 repo_add_command = (
86 "env KUBECONFIG=./tmp/helm_cluster_id/.kube/config /usr/bin/helm repo add {} {}"
87 ).format(repo_name, repo_url)
88 calls = self.helm_conn._local_async_exec.call_args_list
89 call0_kargs = calls[0][1]
90 self.assertEqual(
91 call0_kargs.get("command"),
92 repo_add_command,
93 "Invalid repo add command: {}".format(call0_kargs.get("command")),
94 )
95 self.assertEqual(
96 call0_kargs.get("env"),
97 self.env,
98 "Invalid env for add command: {}".format(call0_kargs.get("env")),
99 )
100 call1_kargs = calls[1][1]
101 self.assertEqual(
102 call1_kargs.get("command"),
103 repo_update_command,
104 "Invalid repo update command: {}".format(call1_kargs.get("command")),
105 )
106 self.assertEqual(
107 call1_kargs.get("env"),
108 self.env,
109 "Invalid env for update command: {}".format(call1_kargs.get("env")),
110 )
111
112 @asynctest.fail_on(active_handles=True)
113 async def test_repo_list(self):
114 self.helm_conn._local_async_exec = asynctest.CoroutineMock(return_value=("", 0))
115
116 await self.helm_conn.repo_list(self.cluster_uuid)
117
118 self.helm_conn.fs.sync.assert_called_once_with(from_path=self.cluster_id)
119 self.helm_conn.fs.reverse_sync.assert_called_once_with(
120 from_path=self.cluster_id
121 )
122 command = "env KUBECONFIG=./tmp/helm_cluster_id/.kube/config /usr/bin/helm repo list --output yaml"
123 self.helm_conn._local_async_exec.assert_called_with(
124 command=command, env=self.env, raise_exception_on_error=False
125 )
126
127 @asynctest.fail_on(active_handles=True)
128 async def test_repo_remove(self):
129 self.helm_conn._local_async_exec = asynctest.CoroutineMock(return_value=("", 0))
130 repo_name = "bitnami"
131 await self.helm_conn.repo_remove(self.cluster_uuid, repo_name)
132
133 self.helm_conn.fs.sync.assert_called_once_with(from_path=self.cluster_id)
134 self.helm_conn.fs.reverse_sync.assert_called_once_with(
135 from_path=self.cluster_id
136 )
137 command = "env KUBECONFIG=./tmp/helm_cluster_id/.kube/config /usr/bin/helm repo remove {}".format(
138 repo_name
139 )
140 self.helm_conn._local_async_exec.assert_called_once_with(
141 command=command, env=self.env, raise_exception_on_error=True
142 )
143
144 @asynctest.fail_on(active_handles=True)
145 async def test_install(self):
146 kdu_model = "stable/openldap:1.2.2"
147 kdu_instance = "stable-openldap-0005399828"
148 db_dict = {}
149 self.helm_conn._local_async_exec = asynctest.CoroutineMock(return_value=("", 0))
150 self.helm_conn._status_kdu = asynctest.CoroutineMock(return_value=None)
151 self.helm_conn._store_status = asynctest.CoroutineMock()
152 self.helm_conn.generate_kdu_instance_name = Mock(return_value=kdu_instance)
153
154 await self.helm_conn.install(
155 self.cluster_uuid,
156 kdu_model,
157 kdu_instance,
158 atomic=True,
159 namespace=self.namespace,
160 db_dict=db_dict,
161 )
162
163 self.helm_conn.fs.sync.assert_has_calls(
164 [
165 asynctest.call(from_path=self.cluster_id),
166 asynctest.call(from_path=self.cluster_id),
167 ]
168 )
169 self.helm_conn.fs.reverse_sync.assert_has_calls(
170 [
171 asynctest.call(from_path=self.cluster_id),
172 asynctest.call(from_path=self.cluster_id),
173 ]
174 )
175 self.helm_conn._store_status.assert_called_with(
176 cluster_id=self.cluster_id,
177 kdu_instance=kdu_instance,
178 namespace=self.namespace,
179 db_dict=db_dict,
180 operation="install",
181 )
182 command = (
183 "env KUBECONFIG=./tmp/helm_cluster_id/.kube/config /usr/bin/helm install "
184 "--atomic --output yaml --timeout 300 "
185 "--name=stable-openldap-0005399828 --namespace testk8s stable/openldap "
186 "--version 1.2.2"
187 )
188 self.helm_conn._local_async_exec.assert_called_with(
189 command=command, env=self.env, raise_exception_on_error=False
190 )
191
192 @asynctest.fail_on(active_handles=True)
193 async def test_upgrade(self):
194 kdu_model = "stable/openldap:1.2.3"
195 kdu_instance = "stable-openldap-0005399828"
196 db_dict = {}
197 instance_info = {
198 "chart": "openldap-1.2.2",
199 "name": kdu_instance,
200 "namespace": self.namespace,
201 "revision": 1,
202 "status": "DEPLOYED",
203 }
204 self.helm_conn._local_async_exec = asynctest.CoroutineMock(return_value=("", 0))
205 self.helm_conn._store_status = asynctest.CoroutineMock()
206 self.helm_conn.get_instance_info = asynctest.CoroutineMock(
207 return_value=instance_info
208 )
209
210 await self.helm_conn.upgrade(
211 self.cluster_uuid, kdu_instance, kdu_model, atomic=True, db_dict=db_dict
212 )
213 self.helm_conn.fs.sync.assert_called_with(from_path=self.cluster_id)
214 self.helm_conn.fs.reverse_sync.assert_has_calls(
215 [
216 asynctest.call(from_path=self.cluster_id),
217 asynctest.call(from_path=self.cluster_id),
218 ]
219 )
220 self.helm_conn._store_status.assert_called_with(
221 cluster_id=self.cluster_id,
222 kdu_instance=kdu_instance,
223 namespace=self.namespace,
224 db_dict=db_dict,
225 operation="upgrade",
226 )
227 command = (
228 "env KUBECONFIG=./tmp/helm_cluster_id/.kube/config /usr/bin/helm upgrade "
229 "--atomic --output yaml --timeout 300 stable-openldap-0005399828 stable/openldap --version 1.2.3"
230 )
231 self.helm_conn._local_async_exec.assert_called_with(
232 command=command, env=self.env, raise_exception_on_error=False
233 )
234
235 @asynctest.fail_on(active_handles=True)
236 async def test_scale(self):
237 kdu_model = "stable/openldap:1.2.3"
238 kdu_instance = "stable-openldap-0005399828"
239 db_dict = {}
240 instance_info = {
241 "chart": "openldap-1.2.3",
242 "name": kdu_instance,
243 "namespace": self.namespace,
244 "revision": 1,
245 "status": "DEPLOYED",
246 }
247 repo_list = [
248 {
249 "name": "stable",
250 "url": "https://kubernetes-charts.storage.googleapis.com/",
251 }
252 ]
253 kdu_values = """
254 # Default values for openldap.
255 # This is a YAML-formatted file.
256 # Declare variables to be passed into your templates.
257
258 replicaCount: 1
259 dummy-app:
260 replicas: 2
261 """
262
263 self.helm_conn.repo_list = asynctest.CoroutineMock(return_value=repo_list)
264 self.helm_conn.values_kdu = asynctest.CoroutineMock(return_value=kdu_values)
265 self.helm_conn._local_async_exec = asynctest.CoroutineMock(return_value=("", 0))
266 self.helm_conn._store_status = asynctest.CoroutineMock()
267 self.helm_conn.get_instance_info = asynctest.CoroutineMock(
268 return_value=instance_info
269 )
270
271 # TEST-1
272 await self.helm_conn.scale(
273 kdu_instance,
274 2,
275 "",
276 kdu_model=kdu_model,
277 cluster_uuid=self.cluster_uuid,
278 atomic=True,
279 db_dict=db_dict,
280 )
281 command = (
282 "env KUBECONFIG=./tmp/helm_cluster_id/.kube/config "
283 "/usr/bin/helm upgrade --atomic --output yaml --set replicaCount=2 "
284 "--timeout 1800s stable-openldap-0005399828 stable/openldap "
285 "--version 1.2.3"
286 )
287 self.helm_conn._local_async_exec.assert_called_once_with(
288 command=command, env=self.env, raise_exception_on_error=False
289 )
290
291 # TEST-2
292 await self.helm_conn.scale(
293 kdu_instance,
294 3,
295 "dummy-app",
296 kdu_model=kdu_model,
297 cluster_uuid=self.cluster_uuid,
298 atomic=True,
299 db_dict=db_dict,
300 )
301 command = (
302 "env KUBECONFIG=./tmp/helm_cluster_id/.kube/config "
303 "/usr/bin/helm upgrade --atomic --output yaml --set dummy-app.replicas=3 "
304 "--timeout 1800s stable-openldap-0005399828 stable/openldap "
305 "--version 1.2.3"
306 )
307 self.helm_conn._local_async_exec.assert_called_with(
308 command=command, env=self.env, raise_exception_on_error=False
309 )
310 self.helm_conn.fs.reverse_sync.assert_called_with(from_path=self.cluster_id)
311 self.helm_conn._store_status.assert_called_with(
312 cluster_id=self.cluster_id,
313 kdu_instance=kdu_instance,
314 namespace=self.namespace,
315 db_dict=db_dict,
316 operation="scale",
317 )
318
319 @asynctest.fail_on(active_handles=True)
320 async def test_rollback(self):
321 kdu_instance = "stable-openldap-0005399828"
322 db_dict = {}
323 instance_info = {
324 "chart": "openldap-1.2.3",
325 "name": kdu_instance,
326 "namespace": self.namespace,
327 "revision": 2,
328 "status": "DEPLOYED",
329 }
330 self.helm_conn._local_async_exec = asynctest.CoroutineMock(return_value=("", 0))
331 self.helm_conn._store_status = asynctest.CoroutineMock()
332 self.helm_conn.get_instance_info = asynctest.CoroutineMock(
333 return_value=instance_info
334 )
335
336 await self.helm_conn.rollback(
337 self.cluster_uuid, kdu_instance=kdu_instance, revision=1, db_dict=db_dict
338 )
339 self.helm_conn.fs.sync.assert_called_with(from_path=self.cluster_id)
340 self.helm_conn.fs.reverse_sync.assert_called_once_with(
341 from_path=self.cluster_id
342 )
343 self.helm_conn._store_status.assert_called_with(
344 cluster_id=self.cluster_id,
345 kdu_instance=kdu_instance,
346 namespace=self.namespace,
347 db_dict=db_dict,
348 operation="rollback",
349 )
350 command = (
351 "env KUBECONFIG=./tmp/helm_cluster_id/.kube/config "
352 "/usr/bin/helm rollback stable-openldap-0005399828 1 --wait"
353 )
354 self.helm_conn._local_async_exec.assert_called_once_with(
355 command=command, env=self.env, raise_exception_on_error=False
356 )
357
358 @asynctest.fail_on(active_handles=True)
359 async def test_uninstall(self):
360 kdu_instance = "stable-openldap-0005399828"
361 instance_info = {
362 "chart": "openldap-1.2.2",
363 "name": kdu_instance,
364 "namespace": self.namespace,
365 "revision": 3,
366 "status": "DEPLOYED",
367 }
368 self.helm_conn._local_async_exec = asynctest.CoroutineMock(return_value=("", 0))
369 self.helm_conn._store_status = asynctest.CoroutineMock()
370 self.helm_conn.get_instance_info = asynctest.CoroutineMock(
371 return_value=instance_info
372 )
373
374 await self.helm_conn.uninstall(self.cluster_uuid, kdu_instance)
375 self.helm_conn.fs.sync.assert_called_with(from_path=self.cluster_id)
376 self.helm_conn.fs.reverse_sync.assert_called_once_with(
377 from_path=self.cluster_id
378 )
379 command = "env KUBECONFIG=./tmp/helm_cluster_id/.kube/config /usr/bin/helm delete --purge {}".format(
380 kdu_instance
381 )
382 self.helm_conn._local_async_exec.assert_called_once_with(
383 command=command, env=self.env, raise_exception_on_error=True
384 )
385
386 @asynctest.fail_on(active_handles=True)
387 async def test_get_services(self):
388 kdu_instance = "test_services_1"
389 service = {"name": "testservice", "type": "LoadBalancer"}
390 self.helm_conn._local_async_exec_pipe = asynctest.CoroutineMock(
391 return_value=("", 0)
392 )
393 self.helm_conn._parse_services = Mock(return_value=["testservice"])
394 self.helm_conn._get_service = asynctest.CoroutineMock(return_value=service)
395
396 services = await self.helm_conn.get_services(
397 self.cluster_uuid, kdu_instance, self.namespace
398 )
399 self.helm_conn.fs.sync.assert_called_once_with(from_path=self.cluster_id)
400 self.helm_conn.fs.reverse_sync.assert_called_once_with(
401 from_path=self.cluster_id
402 )
403 self.helm_conn._parse_services.assert_called_once()
404 command1 = "env KUBECONFIG=./tmp/helm_cluster_id/.kube/config /usr/bin/helm get manifest {} ".format(
405 kdu_instance
406 )
407 command2 = "/usr/bin/kubectl get --namespace={} -f -".format(self.namespace)
408 self.helm_conn._local_async_exec_pipe.assert_called_once_with(
409 command1, command2, env=self.env, raise_exception_on_error=True
410 )
411 self.assertEqual(
412 services, [service], "Invalid service returned from get_service"
413 )
414
415 @asynctest.fail_on(active_handles=True)
416 async def test_get_service(self):
417 service_name = "service1"
418
419 self.helm_conn._local_async_exec = asynctest.CoroutineMock(return_value=("", 0))
420 await self.helm_conn.get_service(
421 self.cluster_uuid, service_name, self.namespace
422 )
423
424 self.helm_conn.fs.sync.assert_called_once_with(from_path=self.cluster_id)
425 self.helm_conn.fs.reverse_sync.assert_called_once_with(
426 from_path=self.cluster_id
427 )
428 command = (
429 "/usr/bin/kubectl --kubeconfig=./tmp/helm_cluster_id/.kube/config "
430 "--namespace=testk8s get service service1 -o=yaml"
431 )
432 self.helm_conn._local_async_exec.assert_called_once_with(
433 command=command, env=self.env, raise_exception_on_error=True
434 )
435
436 @asynctest.fail_on(active_handles=True)
437 async def test_inspect_kdu(self):
438 self.helm_conn._local_async_exec = asynctest.CoroutineMock(return_value=("", 0))
439
440 kdu_model = "stable/openldap:1.2.4"
441 repo_url = "https://kubernetes-charts.storage.googleapis.com/"
442 await self.helm_conn.inspect_kdu(kdu_model, repo_url)
443
444 command = (
445 "/usr/bin/helm inspect openldap --repo "
446 "https://kubernetes-charts.storage.googleapis.com/ "
447 "--version 1.2.4"
448 )
449 self.helm_conn._local_async_exec.assert_called_with(command=command)
450
451 @asynctest.fail_on(active_handles=True)
452 async def test_help_kdu(self):
453 self.helm_conn._local_async_exec = asynctest.CoroutineMock(return_value=("", 0))
454
455 kdu_model = "stable/openldap:1.2.4"
456 repo_url = "https://kubernetes-charts.storage.googleapis.com/"
457 await self.helm_conn.help_kdu(kdu_model, repo_url)
458
459 command = (
460 "/usr/bin/helm inspect readme openldap --repo "
461 "https://kubernetes-charts.storage.googleapis.com/ "
462 "--version 1.2.4"
463 )
464 self.helm_conn._local_async_exec.assert_called_with(command=command)
465
466 @asynctest.fail_on(active_handles=True)
467 async def test_values_kdu(self):
468 self.helm_conn._local_async_exec = asynctest.CoroutineMock(return_value=("", 0))
469
470 kdu_model = "stable/openldap:1.2.4"
471 repo_url = "https://kubernetes-charts.storage.googleapis.com/"
472 await self.helm_conn.values_kdu(kdu_model, repo_url)
473
474 command = (
475 "/usr/bin/helm inspect values openldap --repo "
476 "https://kubernetes-charts.storage.googleapis.com/ "
477 "--version 1.2.4"
478 )
479 self.helm_conn._local_async_exec.assert_called_with(command=command)
480
481 @asynctest.fail_on(active_handles=True)
482 async def test_get_values_kdu(self):
483 self.helm_conn._local_async_exec = asynctest.CoroutineMock(return_value=("", 0))
484
485 kdu_instance = "stable-openldap-0005399828"
486 await self.helm_conn.get_values_kdu(
487 kdu_instance, self.namespace, self.env["KUBECONFIG"]
488 )
489
490 command = (
491 "env KUBECONFIG=./tmp/helm_cluster_id/.kube/config /usr/bin/helm get values "
492 "stable-openldap-0005399828 --output yaml"
493 )
494 self.helm_conn._local_async_exec.assert_called_with(command=command)
495
496 @asynctest.fail_on(active_handles=True)
497 async def test_instances_list(self):
498 self.helm_conn._local_async_exec = asynctest.CoroutineMock(return_value=("", 0))
499
500 await self.helm_conn.instances_list(self.cluster_uuid)
501 self.helm_conn.fs.sync.assert_called_once_with(from_path=self.cluster_id)
502 self.helm_conn.fs.reverse_sync.assert_called_once_with(
503 from_path=self.cluster_id
504 )
505 command = "/usr/bin/helm list --output yaml"
506 self.helm_conn._local_async_exec.assert_called_once_with(
507 command=command, env=self.env, raise_exception_on_error=True
508 )
509
510 @asynctest.fail_on(active_handles=True)
511 async def test_status_kdu(self):
512 kdu_instance = "stable-openldap-0005399828"
513 self.helm_conn._local_async_exec = asynctest.CoroutineMock(return_value=("", 0))
514
515 await self.helm_conn._status_kdu(
516 self.cluster_id, kdu_instance, self.namespace, yaml_format=True
517 )
518 command = (
519 "env KUBECONFIG=./tmp/helm_cluster_id/.kube/config /usr/bin/helm status {} --output yaml"
520 ).format(kdu_instance)
521 self.helm_conn._local_async_exec.assert_called_once_with(
522 command=command,
523 env=self.env,
524 raise_exception_on_error=True,
525 show_error_log=False,
526 )
527
528 @asynctest.fail_on(active_handles=True)
529 async def test_store_status(self):
530 kdu_instance = "stable-openldap-0005399828"
531 db_dict = {}
532 status = {
533 "info": {
534 "description": "Install complete",
535 "status": {
536 "code": "1",
537 "notes": "The openldap helm chart has been installed",
538 },
539 }
540 }
541 self.helm_conn._status_kdu = asynctest.CoroutineMock(return_value=status)
542 self.helm_conn.write_app_status_to_db = asynctest.CoroutineMock(
543 return_value=status
544 )
545
546 await self.helm_conn._store_status(
547 cluster_id=self.cluster_id,
548 kdu_instance=kdu_instance,
549 namespace=self.namespace,
550 db_dict=db_dict,
551 operation="install",
552 )
553 self.helm_conn._status_kdu.assert_called_once_with(
554 cluster_id=self.cluster_id,
555 kdu_instance=kdu_instance,
556 namespace=self.namespace,
557 yaml_format=False,
558 )
559 self.helm_conn.write_app_status_to_db.assert_called_once_with(
560 db_dict=db_dict,
561 status="Install complete",
562 detailed_status=str(status),
563 operation="install",
564 )
565
566 @asynctest.fail_on(active_handles=True)
567 async def test_reset_uninstall_false(self):
568 self.helm_conn._uninstall_sw = asynctest.CoroutineMock()
569
570 await self.helm_conn.reset(self.cluster_uuid, force=False, uninstall_sw=False)
571 self.helm_conn.fs.sync.assert_called_once_with(from_path=self.cluster_id)
572 self.helm_conn.fs.file_delete.assert_called_once_with(
573 self.cluster_id, ignore_non_exist=True
574 )
575 self.helm_conn._uninstall_sw.assert_not_called()
576
577 @asynctest.fail_on(active_handles=True)
578 async def test_reset_uninstall(self):
579 kdu_instance = "stable-openldap-0021099429"
580 instances = [
581 {
582 "app_version": "2.4.48",
583 "chart": "openldap-1.2.3",
584 "name": kdu_instance,
585 "namespace": self.namespace,
586 "revision": "1",
587 "status": "deployed",
588 "updated": "2020-10-30 11:11:20.376744191 +0000 UTC",
589 }
590 ]
591 self.helm_conn._get_namespace = Mock(return_value=self.namespace)
592 self.helm_conn._uninstall_sw = asynctest.CoroutineMock()
593 self.helm_conn.instances_list = asynctest.CoroutineMock(return_value=instances)
594 self.helm_conn.uninstall = asynctest.CoroutineMock()
595
596 await self.helm_conn.reset(self.cluster_uuid, force=True, uninstall_sw=True)
597 self.helm_conn.fs.sync.assert_called_once_with(from_path=self.cluster_id)
598 self.helm_conn.fs.file_delete.assert_called_once_with(
599 self.cluster_id, ignore_non_exist=True
600 )
601 self.helm_conn._get_namespace.assert_called_once_with(
602 cluster_uuid=self.cluster_uuid
603 )
604 self.helm_conn.instances_list.assert_called_once_with(
605 cluster_uuid=self.cluster_uuid
606 )
607 self.helm_conn.uninstall.assert_called_once_with(
608 cluster_uuid=self.cluster_uuid, kdu_instance=kdu_instance
609 )
610 self.helm_conn._uninstall_sw.assert_called_once_with(
611 cluster_id=self.cluster_id, namespace=self.namespace
612 )
613
614 @asynctest.fail_on(active_handles=True)
615 async def test_uninstall_sw_namespace(self):
616 self.helm_conn._local_async_exec = asynctest.CoroutineMock(return_value=("", 0))
617
618 await self.helm_conn._uninstall_sw(self.cluster_id, self.namespace)
619 calls = self.helm_conn._local_async_exec.call_args_list
620 self.assertEqual(
621 len(calls), 3, "To uninstall should have executed three commands"
622 )
623 call0_kargs = calls[0][1]
624 command_0 = "/usr/bin/helm --kubeconfig={} --home={} reset".format(
625 self.kube_config, self.helm_home
626 )
627 self.assertEqual(
628 call0_kargs,
629 {"command": command_0, "raise_exception_on_error": True, "env": self.env},
630 "Invalid args for first call to local_exec",
631 )
632 call1_kargs = calls[1][1]
633 command_1 = (
634 "/usr/bin/kubectl --kubeconfig={} delete "
635 "clusterrolebinding.rbac.authorization.k8s.io/osm-tiller-cluster-rule".format(
636 self.kube_config
637 )
638 )
639 self.assertEqual(
640 call1_kargs,
641 {"command": command_1, "raise_exception_on_error": False, "env": self.env},
642 "Invalid args for second call to local_exec",
643 )
644 call2_kargs = calls[2][1]
645 command_2 = (
646 "/usr/bin/kubectl --kubeconfig={} --namespace {} delete "
647 "serviceaccount/{}".format(
648 self.kube_config, self.namespace, self.service_account
649 )
650 )
651 self.assertEqual(
652 call2_kargs,
653 {"command": command_2, "raise_exception_on_error": False, "env": self.env},
654 "Invalid args for third call to local_exec",
655 )