tests/run-tests.py: Set name of injected test module to '__main__'.

dpgeorge · dpgeorge · commit cdb3f8220171 · 2024-12-18T11:57:45.000+11:00
Running unittest-based tests with --via-mpy is currently broken, because
the unittest test needs the module to be named `__main__`, whereas it's
actually called `__injected_test`.

Fix this by changing the name when the file is opened.

Signed-off-by: Damien George &lt;damien@micropython.org&gt;
diff --git a/tests/run-tests.py b/tests/run-tests.py
@@ -59,10 +59,13 @@ def base_path(*p):
 os.environ["PYTHONIOENCODING"] = "utf-8"
 
 # Code to allow a target MicroPython to import an .mpy from RAM
+# Note: the module is named `__injected_test` but it needs to have `__name__` set to
+# `__main__` so that the test sees itself as the main module, eg so unittest works.
 injected_import_hook_code = """\
 import sys, os, io, vfs
 class __File(io.IOBase):
   def __init__(self):
+    sys.modules['__injected_test'].__name__ = '__main__'
     self.off = 0
   def ioctl(self, request, arg):
     return 0
