Use a mock test to check rpmdb corrupted vs package not installed

Author: philpep

test/test_modules.py

@@ -13,6 +13,7 @@
 import crypt
 import datetime
 import os
+from unittest import mock
 import re
 import time
 from ipaddress import IPv4Address, IPv6Address, ip_address
@@ -59,15 +60,18 @@ def test_held_package(host):
     assert python.version.startswith("3.11.")
 
 @pytest.mark.testinfra_hosts("docker://rockylinux9")
-def test_rpm_package_not_installed(host):
-    # Pass an invalid value as package name, it returns exit code 1
-    pkg_name = "-3"
-    with pytest.raises(RuntimeError) as excinfo:
-        host.package(pkg_name).is_installed
-    assert (
-        f"Could not check if RPM package '{pkg_name}' is installed. "
-        f"rpm: {pkg_name}: unknown option"
-    ) in str(excinfo.value)
+def test_rpmdb_corrupted(host):
+    host.check_output("mv /var/lib/rpm/rpmdb.sqlite /var/lib/rpm/rpmdb.sqlite.bck")
+    try:
+        host.check_output("dd if=/dev/zero of=/var/lib/rpm/rpmdb.sqlite bs=1024 count=1")
+        with pytest.raises(RuntimeError) as excinfo:
+            host.package("zsh").is_installed
+        assert (
+            f"Could not check if RPM package 'zsh' is installed. "
+            "error: sqlite failure:"
+        ) in str(excinfo.value)
+    finally:
+        host.check_output("mv /var/lib/rpm/rpmdb.sqlite.bck /var/lib/rpm/rpmdb.sqlite")
 
 @pytest.mark.testinfra_hosts("docker://rockylinux9")
 def test_non_default_package_tool(host):

testinfra/modules/package.py

@@ -165,7 +165,7 @@ def version(self):
 class RpmPackage(Package):
     @property
     def is_installed(self):
-        result = self.run("rpm -q --quiet %s 2>&1", self.name)
+        result = self.run_test("rpm -q --quiet %s 2>&1", self.name)
         if result.succeeded:
             return True
         elif result.failed and result.stdout == '':