Use RpmPackage on CentOS and Fedora and not only on CentOS

On Fedora you can also install dpkg, which will cause testinfra to think that the package manager is actually DPKG and not RPM. Thus we instead use the system_info.distribution value to prefer rpm on Fedora
pytest-dev · Nov 4, 2021 · 1133428 · 1133428
1 parent a276c11
commit 1133428
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/testinfra/modules/package.py b/testinfra/modules/package.py
@@ -70,9 +70,12 @@ def get_module_class(cls, host):
             return OpenBSDPackage
         if host.system_info.distribution in ("debian", "ubuntu"):
             return DebianPackage
-        if (
-            host.system_info.distribution
-            and host.system_info.distribution.lower() == "centos"
+        if host.system_info.distribution and (
+            host.system_info.distribution.lower()
+            in (
+                "centos",
+                "fedora",
+            )
         ):
             return RpmPackage
         if host.system_info.distribution == "arch":