Introduced flag for keeping VMs alive (#110)

* Introduced flag for keeping VMs alive

* Update opennebula.py

Author: Kwaizer

alts/worker/runners/base.py

@@ -209,6 +209,7 @@ def __init__(
         package_channel: Optional[str] = None,
         test_configuration: Optional[dict] = None,
         test_flavor: Optional[Dict[str, str]] = None,
+        vm_alive: bool = False,
         verbose: bool = False,
     ):
         # Environment ID and working directory preparation
@@ -279,6 +280,7 @@ def __init__(
         self._stats = {}
         self._verbose = verbose
         self.package_channel = package_channel
+        self.vm_alive = vm_alive
 
     @property
     def artifacts(self):
@@ -1634,6 +1636,7 @@ def __init__(
         package_channel: Optional[str] = None,
         test_configuration: Optional[dict] = None,
         test_flavor: Optional[Dict[str, str]] = None,
+        vm_alive: bool = False,
         verbose: bool = False,
     ):
         super().__init__(
@@ -1652,6 +1655,7 @@ def __init__(
         self._tests_dir = CONFIG.tests_base_dir
         self._ssh_client: Optional[Union[AsyncSSHClient, LongRunSSHClient]] = None
         self._vm_ip = None
+        self._vm_alive = vm_alive
 
     def _wait_for_ssh(self, retries=60):
         ansible = local[self.ansible_binary]
@@ -1746,12 +1750,13 @@ def setup(self, skip_provision: bool = False):
         self._ssh_client = LongRunSSHClient(**params)
 
     def teardown(self, publish_artifacts: bool = True):
-        if self._ssh_client:
-            try:
-                self._ssh_client.close()
-            except:
-                pass
-        super().teardown(publish_artifacts=publish_artifacts)
+        if not self._vm_alive:
+            if self._ssh_client:
+                try:
+                    self._ssh_client.close()
+                except:
+                    pass
+            super().teardown(publish_artifacts=publish_artifacts)
 
     def exec_command(self, *args, **kwargs) -> Tuple[int, str, str]:
         command = ' '.join(args)

alts/worker/runners/docker.py

@@ -68,6 +68,7 @@ def __init__(
         dist_arch: str = 'x86_64',
         test_configuration: Optional[dict] = None,
         test_flavor: Optional[Dict[str, str]] = None,
+        vm_alive: bool = False,
         artifacts_uploader: Optional[BaseLogsUploader] = None,
         package_channel: Optional[str] = None,
         verbose: bool = False,
@@ -97,6 +98,7 @@ def __init__(
             dist_arch=dist_arch,
             test_configuration=test_configuration,
             test_flavor=test_flavor,
+            vm_alive=vm_alive,
             artifacts_uploader=artifacts_uploader,
             package_channel=package_channel,
             verbose=verbose,

alts/worker/runners/opennebula.py

@@ -49,6 +49,7 @@ def __init__(
         package_channel: Optional[str] = None,
         test_configuration: Optional[dict] = None,
         test_flavor: Optional[Dict[str, str]] = None,
+        vm_alive: bool = False,
         verbose: bool = False,
     ):
         super().__init__(
@@ -62,6 +63,7 @@ def __init__(
             package_channel=package_channel,
             test_configuration=test_configuration,
             test_flavor=test_flavor,
+            vm_alive=vm_alive,
             verbose=verbose,
         )
         user = CONFIG.opennebula_config.username

alts/worker/tasks.py

@@ -160,7 +160,8 @@ def set_artifacts_when_stage_has_unexpected_exception(
         'dist_arch': task_params.get('dist_arch', 'x86_64'),
         'package_channel': task_params.get('package_channel', 'beta'),
         'test_configuration': task_params.get('test_configuration', {}),
-        'test_flavor': task_params.get('test_flavor', {})
+        'test_flavor': task_params.get('test_flavor', {}),
+        'vm_alive': task_params.get('vm_alive')
     }
 
     runner_class = RUNNER_MAPPING[task_params['runner_type']]