Merge pull request #2881 from DataDog/baptiste.foy/FA/keep-env

upgrade(onboarding): Add option to keep VMs
DataDog · Aug 14, 2024 · 68a6376 · 68a6376
2 parents b7164df + 7166c02
commit 68a6376
Showing 1 changed file with 20 additions and 15 deletions.
diff --git a/utils/virtual_machine/aws_provider.py b/utils/virtual_machine/aws_provider.py
@@ -102,21 +102,26 @@ def _start_vm(self, vm):
         self.install_provision(vm, ec2_server, server_connection, create_cache=ami_id is None)
 
     def stack_destroy(self):
-        logger.info(f"Destroying VMs: {self.vms}")
-        try:
-            self.stack.destroy(on_output=logger.info, debug=True)
-            self.datadog_event_sender.sendEventToDatadog(
-                f"[E2E] Stack {self.stack_name}  : success on Pulumi stack destroy",
-                "",
-                ["operation:destroy", "result:ok", f"stack:{self.stack_name}"],
-            )
-        except Exception as pulumi_exception:
-            logger.error("Exception destroying aws provision infraestructure")
-            logger.exception(pulumi_exception)
-            self.datadog_event_sender.sendEventToDatadog(
-                f"[E2E] Stack {self.stack_name}  : error on Pulumi stack destroy",
-                repr(pulumi_exception),
-                ["operation:destroy", "result:fail", f"stack:{self.stack_name}"],
+        if os.getenv("ONBOARDING_KEEP_VMS") is None:
+            logger.info(f"Destroying VMs: {self.vms}")
+            try:
+                self.stack.destroy(on_output=logger.info, debug=True)
+                self.datadog_event_sender.sendEventToDatadog(
+                    f"[E2E] Stack {self.stack_name}  : success on Pulumi stack destroy",
+                    "",
+                    ["operation:destroy", "result:ok", f"stack:{self.stack_name}"],
+                )
+            except Exception as pulumi_exception:
+                logger.error("Exception destroying aws provision infraestructure")
+                logger.exception(pulumi_exception)
+                self.datadog_event_sender.sendEventToDatadog(
+                    f"[E2E] Stack {self.stack_name}  : error on Pulumi stack destroy",
+                    repr(pulumi_exception),
+                    ["operation:destroy", "result:fail", f"stack:{self.stack_name}"],
+                )
+        else:
+            logger.info(
+                f"Did not destroy VMs as ONBOARDING_KEEP_VMS is set. To destroy them, re-run the test without this env var."
             )
 
     def _get_cached_ami(self, vm):