Fix test_emulate_energy_full_not_avail (#1163)

The value may come from two different files, must mock both.
giampaolo · Oct 29, 2017 · fe53d1a · fe53d1a
1 parent 35bd6fd
commit fe53d1a
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/psutil/tests/test_linux.py b/psutil/tests/test_linux.py
@@ -1326,7 +1326,9 @@ def test_emulate_energy_full_not_avail(self):
         # Emulate a case where energy_full file does not exist.
         # Expected fallback on /capacity.
         def open_mock(name, *args, **kwargs):
-            if name.startswith("/sys/class/power_supply/BAT0/energy_full"):
+            energy_full = "/sys/class/power_supply/BAT0/energy_full"
+            charge_full = "/sys/class/power_supply/BAT0/charge_full"
+            if name.startswith(energy_full) or name.startswith(charge_full):
                 raise IOError(errno.ENOENT, "")
             elif name.startswith("/sys/class/power_supply/BAT0/capacity"):
                 return io.BytesIO(b"88")