tests: make it slightly easier to set a custom ppa install source

canonical · Feb 2, 2024 · d592bdd · d592bdd
1 parent 5c0192f
commit d592bdd
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/features/environment.py b/features/environment.py
@@ -256,7 +256,13 @@ def from_environ(cls, config) -> "UAClientBehaveConfig":
                 kwargs[key] = bool_value
 
         if "install_from" in kwargs:
-            kwargs["install_from"] = InstallationSource(kwargs["install_from"])
+            if str(kwargs["install_from"]).startswith("ppa:"):
+                kwargs["custom_ppa"] = kwargs["install_from"]
+                kwargs["install_from"] = InstallationSource.CUSTOM
+            else:
+                kwargs["install_from"] = InstallationSource(
+                    kwargs["install_from"]
+                )
 
         return cls(**kwargs)  # type: ignore