From ad428d16ea749243772e01da477459a93f2995af Mon Sep 17 00:00:00 2001
From: Chad Smith <chad.smith@canonical.com>
Date: Thu, 7 May 2026 18:43:53 -0600
Subject: [PATCH] tests: expect intermittent warning on azure pro images. avoid
 client.destroy

Avoid client.destroy within the cloud_session.launch context manager
as it already performs cloud.destroy on context exit which calls
instance.destroy. This avoids the following error seen by pycloudlib
on Azure instances:

  <pycloudlib.azure.instance.AzureInstance object at ....

  @property
  def id(self):
      """Return instance id."""
      return self._instance["vm"].id
               ^^^^^^^^^^^^^^^^^^^^
      TypeError: 'NoneType' object is not subscriptable
---
 tests/integration_tests/modules/test_ubuntu_pro.py | 11 ++++++++---
 1 file changed, 8 insertions(+), 3 deletions(-)

diff --git a/tests/integration_tests/modules/test_ubuntu_pro.py b/tests/integration_tests/modules/test_ubuntu_pro.py
index f955e6eb60f..a0ebcac1281 100644
--- a/tests/integration_tests/modules/test_ubuntu_pro.py
+++ b/tests/integration_tests/modules/test_ubuntu_pro.py
@@ -231,7 +231,6 @@ def maybe_install_cloud_init(session_cloud: IntegrationCloud):
 
         client.install_new_cloud_init(source)
         session_cloud.snapshot_id = client.snapshot()
-        client.destroy()
 
     return {"image_id": session_cloud.snapshot_id}
 
@@ -252,8 +251,14 @@ def test_custom_services(self, session_cloud: IntegrationCloud):
             launch_kwargs=launch_kwargs,
         ) as client:
             log = client.read_from_file("/var/log/cloud-init.log")
-            verify_clean_log(log)
-            verify_clean_boot(client)
+            ignore_warnings = []
+            if "azure" == PLATFORM:
+                # Pro images sometimes hit a single 404 in early provisioning
+                ignore_warnings.append(
+                    "Polling IMDS failed attempt 1 with exception:"
+                    " UrlError('404"
+                )
+            verify_clean_boot(client, ignore_warnings=ignore_warnings)
             assert_ua_service_noop(client)
             assert is_attached(client)
             services_status = get_services_status(client)