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

canonical · Jan 30, 2024 · 60a56f2 · 60a56f2
1 parent 0baebf9
commit 60a56f2
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 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