Hi Tom,

On 1/20/26 4:47 PM, Tom Rini wrote:
From: Dmitrii Sharshakov <[email protected]>

Check if elftools package is available before running DecodeElf().

This clarifies the error message and adds the required test for
coverage.

Signed-off-by: Dmitrii Sharshakov <[email protected]>
Signed-off-by: Andrew Soknacki <[email protected]>
Reviewed-by: Tom Rini <[email protected]>
[trini: Add the test provided by Andrew on IRC, to fix coverage]
Signed-off-by: Tom Rini <[email protected]>
---
Changes in v3:
- Add the test, provided by Andrew, to address the coverage failure in
   CI
---
  tools/binman/elf.py      |  2 ++
  tools/binman/elf_test.py | 12 ++++++++++++
  2 files changed, 14 insertions(+)

diff --git a/tools/binman/elf.py b/tools/binman/elf.py
index 6ac960e04196..899c84ad36d6 100644
--- a/tools/binman/elf.py
+++ b/tools/binman/elf.py
@@ -570,6 +570,8 @@ def is_valid(data):
      Returns:
          bool: True if a valid Elf file, False if not
      """
+    if not ELF_TOOLS:
+        raise ValueError("Python: No module named 'elftools'")
      try:
          DecodeElf(data, 0)
          return True
diff --git a/tools/binman/elf_test.py b/tools/binman/elf_test.py
index 5b1733928986..3ad0bf4c4b09 100644
--- a/tools/binman/elf_test.py
+++ b/tools/binman/elf_test.py
@@ -373,6 +373,18 @@ class TestElf(unittest.TestCase):
          self.assertEqual(True, elf.is_valid(data))
          self.assertEqual(False, elf.is_valid(data[4:]))
+ def test_is_valid_fail(self):
+        """Test calling is_valid() without elftools"""
+        old_val = elf.ELF_TOOLS
+        try:
+            elf.ELF_TOOLS = False
+            with self.assertRaises(ValueError) as e:
+                elf.is_valid(b'')
+            self.assertIn("Python: No module named 'elftools'",
+                      str(e.exception))
+        finally:
+            elf.ELF_TOOLS = old_val
+

I'm not sure this is a good idea. The issue is that you're modifying the variable from a python module. The check may be run by other tests at the same time in different threads and I think the tests will unnecessarily be skipped or fail?

I think we may want something like:

@unittest.mock.patch('binman.elf.ELF_TOOLS', False)
def test_is_valid_fail(self):
?

Cheers,
Quentin

Reply via email to