Exec on instance

canonical · Dec 20, 2024 · eee28a6 · eee28a6
1 parent 30d3f3b
commit eee28a6
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/tests/integration/tests/test_util/harness/lxd.py b/tests/integration/tests/test_util/harness/lxd.py
@@ -131,8 +131,9 @@ def new_instance(self, network_type: str = "IPv4") -> Instance:
         except subprocess.CalledProcessError as e:
             raise HarnessError(f"Failed to create LXD container {instance_id}") from e
 
-        stubbornly(retries=3, delay_s=5).exec(["snap", "wait", "system", "seed.loaded"])
-        return Instance(self, instance_id)
+        instance = Instance(self, instance_id)
+        stubbornly(retries=3, delay_s=5).on(instance).exec(["snap", "wait", "system", "seed.loaded"])
+        return instance
 
     def _configure_profile(self, profile_name: str, profile_config: str):
         LOG.debug("Checking for LXD profile %s", profile_name)